...

kitcom/goparser.go

@@ -17,11 +17,12 @@ type apiStruct struct {
 }
 
 type GoApiParser struct {
-	apiStructs []*apiStruct
 }
 
 func (g *GoApiParser) Parse(sourceFile string) (Api, error) {
 
+	var apiStructs []*apiStruct
+
 	fileSet := token.NewFileSet()
 	astFile, err := parser.ParseFile(fileSet, sourceFile, nil, parser.ParseComments|parser.SkipObjectResolution)
 	if err != nil {
@@ -57,14 +58,13 @@ func (g *GoApiParser) Parse(sourceFile string) (Api, error) {
 		}
 		_ = structType
 
-		g.apiStructs = append(g.apiStructs, &apiStruct{
+		apiStructs = append(apiStructs, &apiStruct{
 			name:    typeSpec.Name.Name,
 			pkgName: pkgName,
 		})
 	}
 
-	if len(g.apiStructs) == 0 {
+	if len(apiStructs) == 0 {
-		// todo support arbitrary order of input files
 		return Api{}, fmt.Errorf("no api struct found")
 	}
 
@@ -95,7 +95,7 @@ func (g *GoApiParser) Parse(sourceFile string) (Api, error) {
 			continue
 		}
 
-		for _, apiStrct := range g.apiStructs {
+		for _, apiStrct := range apiStructs {
 			if recvIdent.Name == apiStrct.name && pkgName == apiStrct.pkgName {
 				apiStrct.methods = append(apiStrct.methods, funcDecl)
 			}

kitcom/habr.go

@@ -1,241 +0,0 @@
-//go:build exclude
-
-package main
-
-import (
-	"bytes"
-	"fmt"
-	"go/ast"
-	"go/parser"
-	"go/printer"
-	"go/token"
-	"log"
-	"os"
-	"strings"
-	"text/template"
-
-	"golang.org/x/tools/go/ast/inspector"
-)
-
-// Шаблон, на основе которого будем генерировать
-// .EntityName, .PrimaryType — параметры,
-// в которые будут установлены данные, добытые из AST-модели
-var repositoryTemplate = template.Must(template.New("").Parse(`
-package main
-
-import (
-    "github.com/jinzhu/gorm"
-)
-
-type {{ .EntityName }}Repository struct {
-    db *gorm.DB
-}
-
-func New{{ .EntityName }}Repository(db *gorm.DB) {{ .EntityName }}Repository {
-    return {{ .EntityName }}Repository{ db: db}
-}
-
-func (r {{ .EntityName }}Repository) Get({{ .PrimaryName }} {{ .PrimaryType}}) (*{{ .EntityName }}, error) {
-    entity := new({{ .EntityName }})
-    err := r.db.Limit(1).Where("{{ .PrimarySQLName }} = ?", {{ .PrimaryName }}).Find(entity).Error()
-    return entity, err
-}
-
-
-func (r {{ .EntityName }}Repository) Create(entity *{{ .EntityName }}) error {
-    return r.db.Create(entity).Error
-}
-
-func (r {{ .EntityName }}Repository) Update(entity *{{ .EntityName }}) error {
-    return r.db.Model(entity).Update.Error
-}
-
-func (r {{ .EntityName }}Repository) Update(entity *{{ .EntityName }}) error {
-    return r.db.Model(entity).Update.Error
-}
-
-func (r {{ .EntityName }}Repository) Delete(entity *{{ .EntityName }}) error {
-    return r.db.Delete.Error
-}
-`))
-
-// Агрегатор данных для установки параметров в шаблоне
-type repositoryGenerator struct {
-	typeSpec   *ast.TypeSpec
-	structType *ast.StructType
-}
-
-// Просто helper-функция для печати замысловатого ast.Expr в обычный string
-func expr2string(expr ast.Expr) string {
-	var buf bytes.Buffer
-	err := printer.Fprint(&buf, token.NewFileSet(), expr)
-	if err != nil {
-		log.Fatalf("error print expression to string: #{err}")
-	}
-	return buf.String()
-}
-
-// Helper для извлечения поля структуры,
-// которое станет первичным ключом в таблице DB
-// Поиск поля ведётся по тегам
-// Ищем то, что мы пометили gorm:"primary_key"
-func (r repositoryGenerator) primaryField() (*ast.Field, error) {
-	for _, field := range r.structType.Fields.List {
-		if !strings.Contains(field.Tag.Value, "primary") {
-			continue
-		}
-		return field, nil
-	}
-	return nil, fmt.Errorf("has no primary field")
-}
-
-// Собственно, генератор
-// оформлен методом структуры repositoryGenerator,
-// так что параметры передавать не нужно:
-// они уже аккумулированы в ресивере метода r repositoryGenerator
-// Передаём ссылку на ast.File,
-// в котором и окажутся плоды трудов
-func (r repositoryGenerator) Generate(outFile *ast.File) error {
-	//Находим первичный ключ
-	primary, err := r.primaryField()
-	if err != nil {
-		return err
-	}
-	//Аллокация и установка параметров для template
-	params := struct {
-		EntityName     string
-		PrimaryName    string
-		PrimarySQLName string
-		PrimaryType    string
-	}{
-		//Параметры извлекаем из ресивера метода
-		EntityName:     r.typeSpec.Name.Name,
-		PrimaryName:    primary.Names[0].Name,
-		PrimarySQLName: primary.Names[0].Name,
-		PrimaryType:    expr2string(primary.Type),
-	}
-	//Аллокация буфера,
-	//куда будем заливать выполненный шаблон
-	var buf bytes.Buffer
-	//Процессинг шаблона с подготовленными параметрами
-	//в подготовленный буфер
-	err = repositoryTemplate.Execute(&buf, params)
-	if err != nil {
-		return fmt.Errorf("execute template: %v", err)
-	}
-	//Теперь сделаем парсинг обработанного шаблона,
-	//который уже стал валидным кодом Go,
-	//в дерево разбора,
-	//получаем AST этого кода
-	templateAst, err := parser.ParseFile(
-		token.NewFileSet(),
-		//Источник для парсинга лежит не в файле,
-		"",
-		//а в буфере
-		buf.Bytes(),
-		//mode парсинга, нас интересуют в основном комментарии
-		parser.ParseComments,
-	)
-	if err != nil {
-		return fmt.Errorf("parse template: %v", err)
-	}
-	//Добавляем декларации из полученного дерева
-	//в результирующий outFile *ast.File,
-	//переданный нам аргументом
-	for _, decl := range templateAst.Decls {
-		outFile.Decls = append(outFile.Decls, decl)
-	}
-	return nil
-}
-
-func main() {
-	//Цель генерации передаётся переменной окружения
-	path := os.Getenv("GOFILE")
-	if path == "" {
-		log.Fatal("GOFILE must be set")
-	}
-	//Разбираем целевой файл в AST
-	astInFile, err := parser.ParseFile(
-		token.NewFileSet(),
-		path,
-		nil,
-		//Нас интересуют комментарии
-		parser.ParseComments,
-	)
-	if err != nil {
-		log.Fatalf("parse file: %v", err)
-	}
-	//Для выбора интересных нам деклараций
-	//используем Inspector из golang.org/x/tools/go/ast/inspector
-	i := inspector.New([]*ast.File{astInFile})
-	//Подготовим фильтр для этого инспектора
-	iFilter := []ast.Node{
-		//Нас интересуют декларации
-		&ast.GenDecl{},
-	}
-	//Выделяем список заданий генерации
-	var genTasks []repositoryGenerator
-	//Запускаем инспектор с подготовленным фильтром
-	//и литералом фильтрующей функции
-	i.Nodes(iFilter, func(node ast.Node, push bool) (proceed bool) {
-		genDecl := node.(*ast.GenDecl)
-		//Код без комментариев не нужен,
-		if genDecl.Doc == nil {
-			return false
-		}
-		//интересуют спецификации типов,
-		typeSpec, ok := genDecl.Specs[0].(*ast.TypeSpec)
-		if !ok {
-			return false
-		}
-		//а конкретно структуры
-		structType, ok := typeSpec.Type.(*ast.StructType)
-		if !ok {
-			return false
-		}
-		//Из оставшегося
-		for _, comment := range genDecl.Doc.List {
-			switch comment.Text {
-			//выделяем структуры, помеченные комментарием repogen:entity,
-			case "//repogen:entity":
-				//и добавляем в список заданий генерации
-				genTasks = append(genTasks, repositoryGenerator{
-					typeSpec:   typeSpec,
-					structType: structType,
-				})
-			}
-		}
-		return false
-	})
-	//Аллокация результирующего дерева разбора
-	astOutFile := &ast.File{
-		Name: astInFile.Name,
-	}
-	//Запускаем список заданий генерации
-	for _, task := range genTasks {
-		//Для каждого задания вызываем написанный нами генератор
-		//как метод этого задания
-		//Сгенерированные декларации помещаются в результирующее дерево разбора
-		err = task.Generate(astOutFile)
-		if err != nil {
-			log.Fatalf("generate: %v", err)
-		}
-	}
-	//Подготовим файл конечного результата всей работы,
-	//назовем его созвучно файлу модели, добавим только суффикс _gen
-	outFile, err := os.Create(strings.TrimSuffix(path, ".go") + "_gen.go")
-	if err != nil {
-		log.Fatalf("create file: %v", err)
-	}
-	//Не забываем прибраться
-	defer outFile.Close()
-	//Печатаем результирующий AST в результирующий файл исходного кода
-	//«Печатаем» не следует понимать буквально,
-	//дерево разбора нельзя просто переписать в файл исходного кода,
-	//это совершенно разные форматы
-	//Мы здесь воспользуемся специализированным принтером из пакета ast/printer
-	err = printer.Fprint(outFile, token.NewFileSet(), astOutFile)
-	if err != nil {
-		log.Fatalf("print file: %v", err)
-	}
-}