gen.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. package dartgen
  2. import (
  3. "errors"
  4. "fmt"
  5. "strings"
  6. "github.com/spf13/cobra"
  7. "github.com/zeromicro/go-zero/core/logx"
  8. "github.com/zeromicro/go-zero/tools/goctl/api/parser"
  9. )
  10. var (
  11. // VarStringDir describes the directory.
  12. VarStringDir string
  13. // VarStringAPI defines the API.
  14. VarStringAPI string
  15. // VarStringLegacy describes whether legacy.
  16. VarStringLegacy bool
  17. // VarStringHostname defines the hostname.
  18. VarStringHostname string
  19. )
  20. // DartCommand create dart network request code
  21. func DartCommand(_ *cobra.Command, _ []string) error {
  22. apiFile := VarStringAPI
  23. dir := VarStringDir
  24. isLegacy := VarStringLegacy
  25. hostname := VarStringHostname
  26. if len(apiFile) == 0 {
  27. return errors.New("missing -api")
  28. }
  29. if len(dir) == 0 {
  30. return errors.New("missing -dir")
  31. }
  32. if len(hostname) == 0 {
  33. fmt.Println("you could use '-hostname' flag to specify your server hostname")
  34. hostname = "go-zero.dev"
  35. }
  36. api, err := parser.Parse(apiFile)
  37. if err != nil {
  38. return err
  39. }
  40. if err := api.Validate(); err != nil {
  41. return err
  42. }
  43. api.Service = api.Service.JoinPrefix()
  44. if !strings.HasSuffix(dir, "/") {
  45. dir = dir + "/"
  46. }
  47. api.Info.Title = strings.Replace(apiFile, ".api", "", -1)
  48. logx.Must(genData(dir+"data/", api, isLegacy))
  49. logx.Must(genApi(dir+"api/", api, isLegacy))
  50. logx.Must(genVars(dir+"vars/", isLegacy, hostname))
  51. if err := formatDir(dir); err != nil {
  52. logx.Errorf("failed to format, %v", err)
  53. }
  54. return nil
  55. }