make_policy.go 639 B

12345678910111213141516171819202122232425262728293031
  1. package make
  2. import (
  3. "fmt"
  4. "os"
  5. "github.com/spf13/cobra"
  6. "github.com/runningwater/gohub/pkg/console"
  7. )
  8. var CmdMakePolicy = &cobra.Command{
  9. Use: "policy",
  10. Short: "Create policy file, example: make policy user",
  11. Run: runMakePolicy,
  12. Args: cobra.ExactArgs(1), // 只允许且必须传 1 个参数
  13. }
  14. func runMakePolicy(cmd *cobra.Command, args []string) {
  15. model := makeModelFromString(args[0])
  16. err := os.MkdirAll("app/policies", os.ModePerm)
  17. if err != nil {
  18. console.Error(err.Error())
  19. return
  20. }
  21. filePath := fmt.Sprintf("app/policies/%s_policy.go", model.PackageName)
  22. createFileFromTpl(filePath, "policy", model)
  23. }