// Generated package name for resources supporting the given version. func packageName(version *design.APIVersionDefinition) (pack string) { pack = TargetPackage if version.Version != "" { pack = codegen.Goify(codegen.VersionPackage(version.Version), false) } return }
// Generate the application code, implement codegen.Generator. func (g *Generator) Generate(api *design.APIDefinition) (_ []string, err error) { if api == nil { return nil, fmt.Errorf("missing API definition, make sure design.Design is properly initialized") } go utils.Catch(nil, func() { g.Cleanup() }) defer func() { if err != nil { g.Cleanup() } }() outdir := AppOutputDir() err = api.IterateVersions(func(v *design.APIVersionDefinition) error { verdir := outdir if v.Version != "" { verdir = filepath.Join(verdir, codegen.VersionPackage(v.Version)) } if err := os.MkdirAll(verdir, 0755); err != nil { return err } if err := g.generateContexts(verdir, api, v); err != nil { return err } if err := g.generateControllers(verdir, v); err != nil { return err } if err := g.generateHrefs(verdir, v); err != nil { return err } if err := g.generateMediaTypes(verdir, v); err != nil { return err } if err := g.generateUserTypes(verdir, v); err != nil { return err } return nil }) if err != nil { return nil, err } return g.genfiles, nil }
func okResp(a *design.ActionDefinition, v string) map[string]interface{} { var ok *design.ResponseDefinition for _, resp := range a.Responses { if resp.Status == 200 { ok = resp break } } if ok == nil { return nil } var mt *design.MediaTypeDefinition var ok2 bool if mt, ok2 = design.Design.MediaTypes[design.CanonicalIdentifier(ok.MediaType)]; !ok2 { return nil } var pkg string if v == "" { pkg = TargetPackage } else { pkg = codegen.VersionPackage(v) } name := codegen.GoTypeRef(mt, mt.AllRequired(), 1) var pointer string if strings.HasPrefix(name, "*") { name = name[1:] pointer = "*" } typeref := fmt.Sprintf("%s%s.%s", pointer, pkg, name) if strings.HasPrefix(typeref, "*") { typeref = "&" + typeref[1:] } return map[string]interface{}{ "Name": ok.Name, "HasMultipleViews": len(mt.Views) > 1, "GoType": codegen.GoNativeType(mt), "TypeRef": typeref, } }
// Generate produces the skeleton main. func (g *Generator) Generate(api *design.APIDefinition) (_ []string, err error) { go utils.Catch(nil, func() { g.Cleanup() }) defer func() { if err != nil { g.Cleanup() } }() mainFile := filepath.Join(codegen.OutputDir, "main.go") if Force { os.Remove(mainFile) } g.genfiles = append(g.genfiles, mainFile) _, err = os.Stat(mainFile) funcs := template.FuncMap{ "tempvar": tempvar, "generateSwagger": generateSwagger, "goify": codegen.Goify, "okResp": okResp, "newControllerVersion": newControllerVersion, "versionPkg": codegen.VersionPackage, "targetPkg": func() string { return TargetPackage }, } gopath := filepath.SplitList(os.Getenv("GOPATH"))[0] if err != nil { var tmpl *template.Template tmpl, err = template.New("main").Funcs(funcs).Parse(mainTmpl) if err != nil { panic(err.Error()) // bug } gg := codegen.NewGoGenerator(mainFile) var outPkg string outPkg, err = filepath.Rel(gopath, codegen.OutputDir) if err != nil { return } outPkg = strings.TrimPrefix(filepath.ToSlash(outPkg), "src/") appPkg := path.Join(outPkg, "app") swaggerPkg := path.Join(outPkg, "swagger") imports := []*codegen.ImportSpec{ codegen.SimpleImport("github.com/raphael/goa"), codegen.SimpleImport(appPkg), codegen.SimpleImport(swaggerPkg), codegen.NewImport("log", "gopkg.in/inconshreveable/log15.v2"), } if generateSwagger() { jsonSchemaPkg := path.Join(outPkg, "schema") imports = append(imports, codegen.SimpleImport(jsonSchemaPkg)) } gg.WriteHeader("", "main", imports) data := map[string]interface{}{ "Name": AppName, "API": api, } if err = tmpl.Execute(gg, data); err != nil { return } if err = gg.FormatCode(); err != nil { return } } tmpl, err := template.New("ctrl").Funcs(funcs).Parse(ctrlTmpl) if err != nil { panic(err.Error()) // bug } imp, err := filepath.Rel(filepath.Join(gopath, "src"), codegen.OutputDir) if err != nil { return } imp = path.Join(filepath.ToSlash(imp), "app") imports := []*codegen.ImportSpec{codegen.SimpleImport(imp)} api.IterateVersions(func(v *design.APIVersionDefinition) error { if v.IsDefault() { return nil } imports = append(imports, codegen.SimpleImport(imp+"/"+codegen.VersionPackage(v.Version))) return nil }) err = api.IterateResources(func(r *design.ResourceDefinition) error { filename := filepath.Join(codegen.OutputDir, snakeCase(r.Name)+".go") if Force { if err := os.Remove(filename); err != nil { return err } } g.genfiles = append(g.genfiles, filename) if _, err := os.Stat(filename); err != nil { resGen := codegen.NewGoGenerator(filename) resGen.WriteHeader("", "main", imports) err := tmpl.Execute(resGen, r) if err != nil { return err } if err := resGen.FormatCode(); err != nil { return err } } return nil }) if err != nil { return } return g.genfiles, nil }
// Generate produces the skeleton main. func (g *Generator) Generate(api *design.APIDefinition) (_ []string, err error) { go utils.Catch(nil, func() { g.Cleanup() }) defer func() { if err != nil { g.Cleanup() } }() mainFile := filepath.Join(codegen.OutputDir, "main.go") if Force { os.Remove(mainFile) } g.genfiles = append(g.genfiles, mainFile) _, err = os.Stat(mainFile) funcs := template.FuncMap{ "tempvar": tempvar, "generateSwagger": generateSwagger, "okResp": okResp, "newControllerVersion": newControllerVersion, "targetPkg": func() string { return TargetPackage }, } if err != nil { file, err := codegen.SourceFileFor(mainFile) if err != nil { return nil, err } var outPkg string outPkg, err = codegen.PackagePath(codegen.OutputDir) if err != nil { return nil, err } outPkg = strings.TrimPrefix(filepath.ToSlash(outPkg), "src/") appPkg := path.Join(outPkg, "app") swaggerPkg := path.Join(outPkg, "swagger") imports := []*codegen.ImportSpec{ codegen.SimpleImport("github.com/raphael/goa"), codegen.SimpleImport(appPkg), codegen.SimpleImport(swaggerPkg), codegen.NewImport("log", "gopkg.in/inconshreveable/log15.v2"), } if generateSwagger() { jsonSchemaPkg := path.Join(outPkg, "schema") imports = append(imports, codegen.SimpleImport(jsonSchemaPkg)) } file.WriteHeader("", "main", imports) data := map[string]interface{}{ "Name": AppName, "API": api, } if err = file.ExecuteTemplate("main", mainT, funcs, data); err != nil { return nil, err } if err = file.FormatCode(); err != nil { return nil, err } } imp, err := codegen.PackagePath(codegen.OutputDir) if err != nil { return } imp = path.Join(filepath.ToSlash(imp), "app") imports := []*codegen.ImportSpec{codegen.SimpleImport(imp)} api.IterateVersions(func(v *design.APIVersionDefinition) error { if v.IsDefault() { return nil } imports = append(imports, codegen.SimpleImport(imp+"/"+codegen.VersionPackage(v.Version))) return nil }) err = api.IterateResources(func(r *design.ResourceDefinition) error { filename := filepath.Join(codegen.OutputDir, snakeCase(r.Name)+".go") if Force { if err := os.Remove(filename); err != nil { return err } } g.genfiles = append(g.genfiles, filename) if _, err := os.Stat(filename); err != nil { file, err := codegen.SourceFileFor(filename) if err != nil { return err } file.WriteHeader("", "main", imports) err = file.ExecuteTemplate("controller", ctrlT, funcs, r) if err != nil { return err } if err := file.FormatCode(); err != nil { return err } } return nil }) if err != nil { return } return g.genfiles, nil }