Visão Geral de Geração de Código

Vamos construir um gerador de código que cria métodos String() para tipos semelhantes a enums! Embora não possamos executar o go generate real neste ambiente, você simulará o que as ferramentas de geração de código fazem — produzindo código-fonte Go programaticamente com base em definições de tipo.

Você organizará seu código em dois arquivos:

• generator.go: Crie sua lógica de geração de código.

  Construa uma struct EnumType que represente uma definição de tipo semelhante a um enum com um Name (string) e Values (slice de strings representando os nomes das constantes).

  Crie uma função GenerateStringMethod(e EnumType) string que produza código-fonte Go para um método String(). O código gerado deve seguir este padrão:

  func (e TypeName) String() string {
      switch e {
      case Value1:
          return "Value1"
      case Value2:
          return "Value2"
      default:
          return "Unknown"
      }
  }

  Crie também uma função GenerateConstants(e EnumType) string que produza as declarações de constantes usando iota:

  const (
      Value1 TypeName = iota
      Value2
      Value3
  )

  Por fim, crie GenerateFullCode(e EnumType) string que combine ambas as saídas com um comentário de cabeçalho indicando que este é um código gerado:

  // Code generated by enum generator. DO NOT EDIT.
  
  [constants]
  
  [string method]
• main.go: Use seu gerador para produzir código para diferentes tipos de enum.

  Leia o nome do tipo enum, depois leia uma contagem de valores seguida por cada nome de valor. Passe essas informações para o seu gerador e imprima o código gerado completo.

As seguintes entradas serão fornecidas:

• Linha 1: Nome do tipo enum
• Linha 2: Número de valores
• Linhas seguintes: Cada nome de valor

Por exemplo, dado:

Status
3
Pending
Active
Completed

Sua saída deve ser:

// Code generated by enum generator. DO NOT EDIT.

const (
    Pending Status = iota
    Active
    Completed
)

func (e Status) String() string {
    switch e {
    case Pending:
        return "Pending"
    case Active:
        return "Active"
    case Completed:
        return "Completed"
    default:
        return "Unknown"
    }
}

E dado:

Priority
2
Low
High

Sua saída deve ser:

// Code generated by enum generator. DO NOT EDIT.

const (
    Low Priority = iota
    High
)

func (e Priority) String() string {
    switch e {
    case Low:
        return "Low"
    case High:
        return "High"
    default:
        return "Unknown"
    }
}

Este exercício demonstra o conceito central por trás de ferramentas como stringer — gerar código repetitivo, mas com segurança de tipos em tempo de compilação, em vez de usar reflexão em tempo de execução. Seu gerador produz código Go válido que poderia ser salvo em um arquivo e compilado com o restante de um projeto.