From 825081f900656b7e52fb01f4754402fb9d176ae7 Mon Sep 17 00:00:00 2001 From: tanberBro Date: Tue, 8 Nov 2022 19:04:37 +0800 Subject: [PATCH] lifted-gen.go add flag parse Signed-off-by: tanberBro --- hack/tools/lifted-gen/lifted-gen.go | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/hack/tools/lifted-gen/lifted-gen.go b/hack/tools/lifted-gen/lifted-gen.go index a1135744a..8746028ac 100644 --- a/hack/tools/lifted-gen/lifted-gen.go +++ b/hack/tools/lifted-gen/lifted-gen.go @@ -13,6 +13,7 @@ import ( "strings" "github.com/olekukonko/tablewriter" + "github.com/spf13/pflag" ) const ( @@ -30,13 +31,16 @@ const ( ) var ( - // TODO: add to flag - liftedDir = "pkg/util/lifted" - output = "pkg/util/lifted/doc.go" + liftedDir = "" + output = "" excludeFiles = []string{"doc.go"} ) func main() { + pflag.StringVar(&liftedDir, "lifted-dir", "pkg/util/lifted", "The path of lifted dir") + pflag.StringVarP(&output, "output", "o", "pkg/util/lifted/doc.go", "The path of lifted-gen output") + pflag.Parse() + a := newAnalyzer() a.collect(liftedDir) a.dump(output)