11package ratio
22
3- // original: https://github.com/hhatto/gocloc/blob/b2dad3847df87ab84c56bb8d27c91ca041e69c16 /language.go
3+ // original: https://github.com/hhatto/gocloc/blob/7b24285f3e4368e0b3df5cd16b0969f3c9be03cb /language.go
44import (
55 "bufio"
66 "bytes"
7- "log"
87 "os"
98 "path/filepath"
109 "regexp"
@@ -27,9 +26,9 @@ var shebang2ext = map[string]string{
2726 "escript" : "erl" ,
2827}
2928
30- func shebang (line string ) (shebangLang string , ok bool ) {
29+ func getShebang (line string ) (shebangLang string , ok bool ) {
3130 ret := reShebangEnv .FindAllStringSubmatch (line , - 1 )
32- if len ( ret ) != 0 && len (ret [0 ]) == 3 {
31+ if ret != nil && len (ret [0 ]) == 3 {
3332 shebangLang = ret [0 ][2 ]
3433 if sl , ok := shebang2ext [shebangLang ]; ok {
3534 return sl , ok
@@ -38,7 +37,7 @@ func shebang(line string) (shebangLang string, ok bool) {
3837 }
3938
4039 ret = reShebangLang .FindAllStringSubmatch (line , - 1 )
41- if len ( ret ) != 0 && len (ret [0 ]) >= 2 {
40+ if ret != nil && len (ret [0 ]) >= 2 {
4241 shebangLang = ret [0 ][1 ]
4342 if sl , ok := shebang2ext [shebangLang ]; ok {
4443 return sl , ok
@@ -49,18 +48,47 @@ func shebang(line string) (shebangLang string, ok bool) {
4948 return "" , false
5049}
5150
52- func fileType (path string ) (ext string , ok bool ) {
51+ func getFileTypeByShebang (path string ) (shebangLang string , ok bool ) {
52+ f , err := os .Open (path )
53+ if err != nil {
54+ return // ignore error
55+ }
56+ defer f .Close ()
57+
58+ reader := bufio .NewReader (f )
59+ line , err := reader .ReadBytes ('\n' )
60+ if err != nil {
61+ return
62+ }
63+ line = bytes .TrimLeftFunc (line , unicode .IsSpace )
64+
65+ if len (line ) > 2 && line [0 ] == '#' && line [1 ] == '!' {
66+ return getShebang (string (line ))
67+ }
68+ return
69+ }
70+
71+ func getFileType (path string ) (ext string , ok bool ) {
5372 ext = filepath .Ext (path )
5473 base := filepath .Base (path )
5574
5675 switch ext {
5776 case ".m" , ".v" , ".fs" , ".r" , ".ts" :
58- content , err := os .ReadFile (filepath .Clean (path ))
77+ content , err := os .ReadFile (path )
78+ if err != nil {
79+ return "" , false
80+ }
81+ lang := enry .GetLanguage (path , content )
82+ return lang , true
83+ case ".mo" :
84+ content , err := os .ReadFile (path )
5985 if err != nil {
6086 return "" , false
6187 }
6288 lang := enry .GetLanguage (path , content )
63- log .Printf ("path=%v, lang=%v\n " , path , lang )
89+ if lang != "" {
90+ return "Motoko" , true
91+ }
6492 return lang , true
6593 }
6694
@@ -88,7 +116,7 @@ func fileType(path string) (ext string, ok bool) {
88116 return "" , false
89117 }
90118
91- shebangLang , ok := fileTypeByShebang (path )
119+ shebangLang , ok := getFileTypeByShebang (path )
92120 if ok {
93121 return shebangLang , true
94122 }
@@ -98,25 +126,3 @@ func fileType(path string) (ext string, ok bool) {
98126 }
99127 return ext , ok
100128}
101-
102- func fileTypeByShebang (path string ) (shebangLang string , ok bool ) {
103- f , err := os .Open (filepath .Clean (path ))
104- if err != nil {
105- return // ignore error
106- }
107- reader := bufio .NewReader (f )
108- line , err := reader .ReadBytes ('\n' )
109- if err != nil {
110- _ = f .Close () //nostyle:handlerrors
111- return
112- }
113- line = bytes .TrimLeftFunc (line , unicode .IsSpace )
114-
115- if len (line ) > 2 && line [0 ] == '#' && line [1 ] == '!' {
116- _ = f .Close () //nostyle:handlerrors
117- return shebang (string (line ))
118- }
119- _ = f .Close () //nostyle:handlerrors
120-
121- return
122- }
0 commit comments