apicodegen/translate/got/middleware_update_relation.go

package got

import (
	"fmt"
	"text/template"

	api "git.eugeniocarvalho.dev/eugeniucarvalho/apicodegen/api"
	. "git.eugeniocarvalho.dev/eugeniucarvalho/apicodegen/common"
	// "github.com/davecgh/go-spew/spew"
)

var (
	updateRelationStmtsTmpl *template.Template
	updateRelationStmtsErr  error
)

func init() {
	updateRelationStmtsTmpl, updateRelationStmtsErr = ParseTemplate(`
	func UpdateRelation{{.entity}}(options *api.Filter) {
		var (
			filter *api.Filter
			err    *errs.Error
			entity = &models.{{.entity}}Reference{}
			wg     = sync.WaitGroup{}
		)
		
		options.Entity = entity
		
		if _, err = models.Api.FindOne(options); err != nil {
			return
		}
		
		entity.Id = entity.Id.(primitive.ObjectID).Hex()
		{{range .relations}}
		wg.Add(1)
		go func(){
			defer wg.Done()
			
			filter = &api.Filter{
				Collection: "{{.Collection}}",
				DB:         "{{.DB}}",
				Patchs:     &bson.A{bson.M{"$set": bson.M{"{{.path}}": entity}}},
				Query:      &bson.M{"{{.attr}}._id": entity.Id},
			}
			
			if _, err = models.Api.PatchMany(filter); err != nil {
				fmt.Println(err.Error())
			}
		}()
		{{end}}

		wg.Wait()
	}`)

	if updateRelationStmtsErr != nil {
		panic(updateRelationStmtsErr)
	}
}

func createUpdateRelationMethod(method *Method, context map[string]interface{}) (out string, err error) {
	var (
		indexMap = map[bool]string{
			true:  ".$",
			false: "",
		}
		hasUpdateRelation              = SR.Has(method.Entity)
		relations                      = []map[string]interface{}{}
		updRelation                    = UPDATE_RELATION + method.Entity
		_, updateRelationMethodDefined = relationFunctionDefined[updRelation]
	)

	context["hasUpdateRelation"] = hasUpdateRelation
	context["relations"] = &relations

	if hasUpdateRelation {
		if updateRelationMethodDefined {
			err = fmt.Errorf("function defined")
			return
		}
		relationFunctionDefined[updRelation] = true

		fmt.Printf("Generate update callback for entity '%s'\n", method.Entity)

		for _, relation := range SR.Get(method.Entity) {

			if relation.DB == "" || relation.Collection == "" {
				api.LogWarning(0, fmt.Sprintf("DB and Collection are required! %s -> %s", relation.Target, relation.Source))
				continue
			}
			relations = append(relations, map[string]interface{}{
				"DB":         relation.DB,
				"Collection": relation.Collection,
				"attr":       relation.Attr,
				"path":       fmt.Sprintf("%s%s", relation.Attr, indexMap[relation.IsArray]),
			})
		}
		out, err = TemplateToString(updateRelationStmtsTmpl, context)
	}
	return
}

// var (
// 	GenUpdateStmts = &Middleware{
// 		Id:   "implement",
// 		Type: "method",
// 		Fn: func(ctx *MiddlewareContext) error {
// 			var (
// 				project          = ctx.Project
// 				method           = ctx.Method
// 				dbName           = project.GetEntityDB(method.Entity)
// 				collectionName   = project.GetCollection(method.Entity)
// 				dependenceMethod = BASE_HAS_DEPENDE + method.Entity
// 				updRelation      = UPDATE_RELATION + method.Entity
// 				indexMap         = map[bool]string{
// 					true:  ".$",
// 					false: "",
// 				}
// 				beforeSend                     = method.Hook("beforeSend")
// 				beforePersist                  = method.Hook("beforePersist")
// 				hasUpdateRelation              = SR.Has(method.Entity)
// 				_, updateRelationMethodDefined = relationFunctionDefined[updRelation]
// 				relations                      = []map[string]interface{}{}
// 				context                        = map[string]interface{}{
// 					"dbName":            dbName,
// 					"collectionName":    collectionName,
// 					"entity":            method.Entity,
// 					"dependenceMethod":  dependenceMethod,
// 					"beforeSend":        beforeSend,
// 					"beforePersist":     beforePersist,
// 					"hasUpdateRelation": hasUpdateRelation,
// 					"relations":         &relations,
// 				}
// 			)

// 			if beforePersist {
// 				generateHookCall(project, method, "beforePersist")
// 			}
// 			if beforeSend {
// 				generateHookCall(project, method, "beforeSend")
// 			}

// 			// Nome do metodo que verifica se a entidade tem dependencias
// 			out, _ := TemplateToString(updateStmtsTmpl, context)
// 			afterMethod := ctx.Statement.Block(G.Id(out)).Line()

// 			return nil
// 		},
// 	}
// )

// j = G.Id("filter").Op("=").Op("&").Qual(API_URL, "Filter").Values(G.Dict{
// 	// Especifica a collection
// 	G.Id("DB"):         G.Lit(relation.DB),
// 	G.Id("Collection"): G.Lit(relation.Collection),

// 	// Especifica a query
// 	G.Id("Query"): G.Op("&").Qual(BSON, "M").Values(G.Dict{
// 		G.Lit(attrId + "._id"): G.Id("f").Dot("Id").Dot("Hex").Call(),
// 	}),

// 	// G.Id("Entity"): G.Qual(BSON, "M").Values(G.Dict{
// 	// 	G.Lit("$set"): G.Qual(BSON, "M").Values(G.Dict{
// 	// 		G.Lit(attrId + index): G.Id("f").Dot("Entity"),
// 	// 	}),
// 	// }),
// 	G.Id("Patchs"): G.Op("&").Qual(BSON, "A").Values(
// 		G.Qual(BSON, "M").Values(G.Dict{
// 			G.Lit("$set"): G.Qual(BSON, "M").Values(G.Dict{
// 				G.Lit(attrId + index): G.Id("f").Dot("Entity"),
// 			}),
// 		}),
// 	),
// }).Line().If(
// G.List(G.Id("_"), G.Id("err")).Op("=").Id("Models").Dot("PatchMany").Call(
// G.Id("filter"),
// ),
// G.Id("err").Op("!=").Id("nil"),
// ).Block(
// G.Qual("fmt", "Println").Call(G.Id("err").Dot("Error").Call()),
// ).Line().Line()

// k.Add(j)
// .Do(func(s *G.Statement) {

// 	_, found := relationFunctionDefined[updRelation]
// 	if !createUpdateRelation || found {
// 		return
// 	}

// 	s.Comment(updRelation + " atualiza as referências da entidade " + method.Entity).Line()

// 	s.Func().Id(updRelation).Params(
// 		G.Id("f").Op("*").Qual(API_URL, "Filter"),
// 	).Block(
// 		G.Var().Defs(
// 			G.Id("filter").Op("*").Qual(API_URL, "Filter"),
// 			G.Id("err").Op("*").Qual(API_ERROR, "Error"),
// 		),
// 		// atualiza o filtro para a entidade de referencia
// 		G.Id("f").Dot("Entity").Op("=").Op("&").Id(method.Entity+"Reference").Values(),

// 		// Carrega os dados do banco para atualizar nas entidades relacionadas
// 		G.If(
// 			G.List(G.Id("_"), G.Id("err")).Op("=").Id("Models").Dot("FindOne").Call(
// 				G.Id("f"),
// 			),
// 			G.Id("err").Op("!=").Id("nil"),
// 		).Block(
// 			G.Return(),
// 		),
// 		G.Do(func(k *G.Statement) { // Adiciona as regras de atualizacao
// 			var (
// 				j     *G.Statement
// 				index string
// 			)

// 			for _, relation := range SR.Get(method.Entity) {
// 				if relation.DB == "" || relation.Collection == "" {
// 					api.LogWarning(0, fmt.Sprintf("Relation DB or Collection can't be empty! %s -> %s", relation.Target, relation.Source))
// 					continue
// 				}

// 				// spew.Dump(relation)
// 				// panic("sss")
// 				// e := project.GetSchema(relation)
// 				attrId := relation.Attr
// 				if relation.IsArray {
// 					index = ".$"
// 				} else {
// 					index = ""
// 				}

// 				j = G.Id("filter").Op("=").Op("&").Qual(API_URL, "Filter").Values(G.Dict{
// 					// Especifica a collection
// 					G.Id("DB"):         G.Lit(relation.DB),
// 					G.Id("Collection"): G.Lit(relation.Collection),

// 					// Especifica a query
// 					G.Id("Query"): G.Op("&").Qual(BSON, "M").Values(G.Dict{
// 						G.Lit(attrId + "._id"): G.Id("f").Dot("Id").Dot("Hex").Call(),
// 					}),

// 					// G.Id("Entity"): G.Qual(BSON, "M").Values(G.Dict{
// 					// 	G.Lit("$set"): G.Qual(BSON, "M").Values(G.Dict{
// 					// 		G.Lit(attrId + index): G.Id("f").Dot("Entity"),
// 					// 	}),
// 					// }),
// 					G.Id("Patchs"): G.Op("&").Qual(BSON, "A").Values(
// 						G.Qual(BSON, "M").Values(G.Dict{
// 							G.Lit("$set"): G.Qual(BSON, "M").Values(G.Dict{
// 								G.Lit(attrId + index): G.Id("f").Dot("Entity"),
// 							}),
// 						}),
// 					),
// 				}).Line().If(
// 					G.List(G.Id("_"), G.Id("err")).Op("=").Id("Models").Dot("PatchMany").Call(
// 						G.Id("filter"),
// 					),
// 					G.Id("err").Op("!=").Id("nil"),
// 				).Block(
// 					G.Qual("fmt", "Println").Call(G.Id("err").Dot("Error").Call()),
// 				).Line().Line()

// 				k.Add(j)
// 			}
// 		}),
// 	)
// })

// var (
// 	// UpdateRelationRotine *G.Statement
// 	method               = ctx.Method
// 	project              = ctx.Project
// 	updRelation          = UPDATE_RELATION + method.Entity
// 	createUpdateRelation = false
// 	entity               = method.Request
// 	responseEntity       = GenericPart.ReplaceAllString(method.Response, "")
// )
// if entity == "" {
// 	panic("Modelo de request não definido em UpdateStmt!")
// }
// if responseEntity == "" {
// 	panic("Modelo de response não definido!")
// }

// ctx.Statement.Block(
// 	// Declaracao das variaveis
// 	G.Var().Defs(
// 		// G.Id("err").Op("*").Qual(API_ERROR, "Error"),
// 		G.Id("entity").Op("=").Op("&").Id(entity).Values(),
// 	).Line(),
// 	// Fazendo o parse do body
// 	G.If(
// 		G.Id("err").Op("=").Qual(API_URL, "ReadJson").Call(G.Id("ctx"), G.Id("entity")),
// 		G.Id("err").Op("!=").Id("nil"),
// 	).Block(
// 		// G.Id("api").Dot("Falha").Call(
// 		// 	G.Id("ctx"),
// 		// 	G.Id("api").Dot("ErrGeneral"),
// 		// 	G.Id("err").Dot("Error").Call(),
// 		// 	G.Nil(),
// 		// ),
// 		// G.Return(G.Id("err")),
// 		G.Return(),
// 	),
// 	G.Do(func(s *G.Statement) {
// 		entity := ctx.Project.EntityDesc(method.Entity)
// 		if entity != nil && entity.HasMode {
// 			s.Add(G.Id(`entity.SetMode("create")`))
// 		}
// 	}),
// 	G.If(
// 		G.Id("err").Op("=").Qual(API_URL, "Validate").Call(G.Id("entity")),
// 		G.Id("err").Op("!=").Id("nil"),
// 	).Block(
// 		// G.Return(G.Id("err")),
// 		G.Return(),
// 	).Line(),
// 	// Captura a base de values
// 	G.Id("values").Op(":=").Id("ctx").Dot("Values").Call(),

// 	// Gera as atribuicoes de variaveis que seram atualizadas (usuario e ultimo update quando existir)
// 	G.Do(func(part *G.Statement) {
// 		entity := ctx.Project.GetSchema(method.Entity)
// 		user := false
// 		for _, prop := range entity.Properties {
// 			if def, ok := prop.Autogenerate["update"]; ok {
// 				switch def.Type {
// 				case "user":
// 					if !user {
// 						part.Add(G.Id(`user := values.Get("$user.ref").(*UserReference)`).Line())
// 						user = true
// 					}

// 					part.Add(G.Id("entity").Dot(strings.Title(prop.ID)).Op("=").Id("user").Line())
// 				case "now":
// 					part.Add(G.Id("entity").Dot(strings.Title(prop.ID)).Op("=").Qual("time", "Now").Call().Id(".Unix()").Line())
// 				}
// 			}
// 		}
// 	}).Line(),

// 	// Carrega o filtro do contexto para adicionar a entidade
// 	G.Id("filter").Op(":=").Id("values").Dot("Get").Call(
// 		G.Lit("$filter"),
// 	).Assert(G.Op("*").Qual(API_URL, "Filter")).Line(),

// 	// Adiciona a entidade
// 	G.Id("filter").Dot("DB").Op("=").Lit(ctx.Project.GetEntityDB(method.Entity)),
// 	G.Id("filter").Dot("Collection").Op("=").Lit(ctx.Project.GetCollection(method.Entity)),
// 	G.Id("filter").Dot("Entity").Op("=").Id("entity").Line(),
// 	generateHookCall(project, method, "beforePersist"),
// 	// Inseri a entidade na coleção e verifica se a operação ocorreu com exito
// 	G.If(
// 		G.List(G.Id("_"), G.Id("err")).Op("=").Id("Models").Dot("UpdateOne").Call(
// 			// G.Lit(ctx.Project.GetCollection(method.Entity)),
// 			G.Id("filter"),
// 		),
// 		G.Id("err").Op("!=").Id("nil"),
// 	).Block(
// 		// G.Id("api").Dot("Falha").Call(
// 		// 	G.Id("ctx"),
// 		// 	G.Id("api").Dot("ErrGeneral"),
// 		// 	G.Id("err").Dot("Error").Call(),
// 		// 	G.Nil(),
// 		// ),
// 		// bson.IsObjectIdHex(m.Id.Hex())
// 		// G.Return(G.Id("err")),
// 		G.Return(),
// 	),
// 	// Cria a rotina de atualização de relacionamentos
// 	G.Do(func(s *G.Statement) {

// 		if SR.Has(method.Entity) {
// 			createUpdateRelation = true

// 			relationFunctionDefined[updRelation] = true
// 			s.Comment("Cria uma thread que executa a atualizaca das referências.").Line().Go().Func().Params().Block(
// 				G.Id(updRelation).Call(G.Id("filter")),
// 			).Call()

// 		}
// 	}),

// 	// Envia a resposta pro usuario em caso de sucesso
// 	// G.Line().Id("resp").Op("=").Id(responseEntity).Values(G.Dict{
// 	// 	G.Id("Entity"): G.Id("entity"),
// 	// }),
// 	G.Line().Id("resp").Op("=").Id("entity"),
// 	generateHookCall(project, method, "beforeSend"),
// 	// G.Return(G.Nil()),
// 	G.Return(),
// ).Line()