11package main
22
33import (
4- "bytes"
54 "flag"
65 "fmt"
6+ "github.com/daixiang0/gci/pkg/gci"
77 "go/scanner"
8- "io"
9- "io/ioutil"
108 "os"
11- "path/filepath"
12- "strings"
139)
1410
1511
1612
1713var (
18- write = flag .Bool ("w" , false , "write result to (source) file instead of stdout" )
19- doDiff = flag .Bool ("d" , false , "display diffs instead of rewriting files" )
14+ doWrite = flag .Bool ("w" , false , "doWrite result to (source) file instead of stdout" )
15+ doDiff = flag .Bool ("d" , false , "display diffs instead of rewriting files" )
2016
2117 localFlag string
2218
2521
2622func report (err error ) {
2723 scanner .PrintError (os .Stderr , err )
28- exitCode = 2
29- }
30-
31- func isGoFile (f os.FileInfo ) bool {
32- // ignore non-Go files
33- name := f .Name ()
34- return ! f .IsDir () && ! strings .HasPrefix (name , "." ) && strings .HasSuffix (name , ".go" )
24+ exitCode = 1
3525}
3626
3727func parseFlags () []string {
@@ -42,86 +32,30 @@ func parseFlags() []string {
4232}
4333
4434func usage () {
45- fmt .Fprintf (os .Stderr , "usage: gci [flags] [path ...]\n " )
35+ _ , _ = fmt .Fprintf (os .Stderr , "usage: gci [flags] [path ...]\n " )
4636 flag .PrintDefaults ()
4737 os .Exit (2 )
4838}
4939
50- func processFile (filename string , out io.Writer ) error {
51- var err error
52-
53- f , err := os .Open (filename )
54- if err != nil {
55- return err
56- }
57- defer f .Close ()
58-
59- src , err := ioutil .ReadAll (f )
60- if err != nil {
61- return err
62- }
63-
64- ori := make ([]byte , len (src ))
65- copy (ori , src )
66- start := bytes .Index (src , importStartFlag )
67- // in case no importStartFlag or importStartFlag exist in the commentFlag
68- if start < 0 {
69- fmt .Printf ("skip file %s since no import\n " , filename )
70- return nil
71- }
72- end := bytes .Index (src [start :], importEndFlag ) + start
73-
74- ret := bytes .Split (src [start + len (importStartFlag ):end ], []byte (linebreak ))
75-
76- p := newPkg (ret , localFlag )
77-
78- res := append (src [:start + len (importStartFlag )], append (p .fmt (), src [end + 1 :]... )... )
79-
80- if ! bytes .Equal (ori , res ) {
81- exitCode = 1
82-
83- if * write {
84- // On Windows, we need to re-set the permissions from the file. See golang/go#38225.
85- var perms os.FileMode
86- if fi , err := os .Stat (filename ); err == nil {
87- perms = fi .Mode () & os .ModePerm
88- }
89- err = ioutil .WriteFile (filename , res , perms )
90- if err != nil {
91- return err
92- }
93- }
94- if * doDiff {
95- data , err := diff (ori , res , filename )
96- if err != nil {
97- return fmt .Errorf ("failed to diff: %v" , err )
98- }
99- fmt .Printf ("diff -u %s %s\n " , filepath .ToSlash (filename + ".orig" ), filepath .ToSlash (filename ))
100- if _ , err := out .Write (data ); err != nil {
101- return fmt .Errorf ("failed to write: %v" , err )
102- }
103- }
104- }
105- if ! * write && ! * doDiff {
106- if _ , err = out .Write (res ); err != nil {
107- return fmt .Errorf ("failed to write: %v" , err )
108- }
109- }
110-
111- return err
112- }
11340
11441func main () {
11542 flag .Usage = usage
11643 paths := parseFlags ()
44+
45+ flagSet := & gci.FlagSet {
46+ LocalFlag : localFlag ,
47+ DoWrite : doWrite ,
48+ DoDiff : doDiff ,
49+ }
50+
11751 for _ , path := range paths {
11852 switch dir , err := os .Stat (path ); {
11953 case err != nil :
12054 report (err )
12155 case dir .IsDir ():
122- walkDir ( path )
56+ report ( gci . WalkDir ( path , flagSet ) )
12357 default :
124- if err := processFile (path , os .Stdout ); err != nil {
58+ if err := gci . ProcessFile (path , os .Stdout , flagSet ); err != nil {
12559 report (err )
12660 }
12761 }
0 commit comments