Thanks to visit codestin.com
Credit goes to github.com

Skip to content

entc cli command to remove generated code #386

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
paulvollmer wants to merge 7 commits into from
Closed

entc cli command to remove generated code #386

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
4a91026
entc/clean: added command to remove generated go code
paulvollmer Mar 13, 2020
ca22509
entc/clean: fix lint error
paulvollmer Mar 13, 2020
c81e8a5
entc/clean: walk files and find files related to schemas
paulvollmer Mar 13, 2020
458cf91
entc/clean: fixed golangci-lint errors
paulvollmer Mar 13, 2020
6c62ea5
entc/clean implement file walk
paulvollmer Apr 20, 2020
3728254
Merge branch 'master' into feat/cli-clean
paulvollmer Sep 28, 2020
f392c38
Merge branch 'master' into feat/cli-clean
paulvollmer Nov 24, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions cmd/entc/entc.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ func main() {
base.InitCmd(),
base.DescribeCmd(),
base.GenerateCmd(migrate),
base.CleanCmd(),
)
_ = cmd.Execute()
}
Expand Down
60 changes: 60 additions & 0 deletions cmd/internal/base/base.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,8 @@ import (
"bytes"
"errors"
"fmt"
"io"
"io/ioutil"
"log"
"os"
"path/filepath"
Expand Down Expand Up @@ -175,6 +177,64 @@ func GenerateCmd(postRun ...func(*gen.Config)) *cobra.Command {
return cmd
}

func CleanCmd() *cobra.Command {
return &cobra.Command{
Use: "clean [flags] path",
Short: "delete generated go code",
Example: examples(
"entc clean ./ent",
),
Args: cobra.ExactArgs(1),
Run: func(cmd *cobra.Command, path []string) {
// delete .go files
err := filepath.Walk(path[0], func(fpath string, info os.FileInfo, err error) error {
if !info.IsDir() {
src, err := ioutil.ReadFile(fpath)
if err != nil {
return err
}

if strings.Contains(string(src), "// Code generated by entc, DO NOT EDIT.") {
err = os.RemoveAll(fpath)
if err != nil {
return err
}
}
}
return nil
})
if err != nil {
log.Fatalln(err)
}

// delete empty directories
err = filepath.Walk(path[0], func(fpath string, info os.FileInfo, err error) error {
if info.IsDir() {
if path[0] != fpath {
f, err := os.Open(fpath)
if err != nil {
return err
}
defer f.Close()

_, err = f.Readdir(1)
if err == io.EOF {
err = os.RemoveAll(fpath)
if err != nil {
return err
}
}
}
}
return nil
})
if err != nil {
log.Fatalln(err)
}
},
}
}

// initEnv initialize an environment for ent codegen.
func initEnv(target string, names []string) error {
if err := createDir(target); err != nil {
Expand Down