Add support for several other file types
Adds support for Dockerfiles, Gemfiles, Ruby files, and Groovy files. Also, in order to fully support Dockerfile and Gemfiles, a new file extension parser was created to deal with files that do not have an extension.
This commit is contained in:
21
main.go
21
main.go
@@ -135,16 +135,16 @@ func walk(ch chan<- *file, start string) {
|
||||
func addLicense(path string, fmode os.FileMode, tmpl *template.Template, data *copyrightData) error {
|
||||
var lic []byte
|
||||
var err error
|
||||
switch filepath.Ext(path) {
|
||||
switch fileExtension(path) {
|
||||
default:
|
||||
return nil
|
||||
case ".c", ".h":
|
||||
lic, err = prefix(tmpl, data, "/*", " * ", " */")
|
||||
case ".js", ".jsx", ".tsx", ".css", ".tf":
|
||||
lic, err = prefix(tmpl, data, "/**", " * ", " */")
|
||||
case ".cc", ".cpp", ".cs", ".go", ".hh", ".hpp", ".java", ".m", ".mm", ".proto", ".rs", ".scala", ".swift", ".dart":
|
||||
case ".cc", ".cpp", ".cs", ".go", ".hh", ".hpp", ".java", ".m", ".mm", ".proto", ".rs", ".scala", ".swift", ".dart", ".groovy":
|
||||
lic, err = prefix(tmpl, data, "", "// ", "")
|
||||
case ".py", ".sh", ".yaml", ".yml":
|
||||
case ".py", ".sh", ".yaml", ".yml", ".dockerfile", "dockerfile", ".rb", "gemfile":
|
||||
lic, err = prefix(tmpl, data, "", "# ", "")
|
||||
case ".el", ".lisp":
|
||||
lic, err = prefix(tmpl, data, "", ";; ", "")
|
||||
@@ -178,10 +178,19 @@ func addLicense(path string, fmode os.FileMode, tmpl *template.Template, data *c
|
||||
return ioutil.WriteFile(path, b, fmode)
|
||||
}
|
||||
|
||||
func fileExtension(name string) string {
|
||||
if v := filepath.Ext(name); v != "" {
|
||||
return strings.ToLower(v)
|
||||
}
|
||||
return strings.ToLower(filepath.Base(name))
|
||||
}
|
||||
|
||||
var head = []string{
|
||||
"#!", // shell script
|
||||
"<?xml", // XML declaratioon
|
||||
"<!doctype", // HTML doctype
|
||||
"#!", // shell script
|
||||
"<?xml", // XML declaratioon
|
||||
"<!doctype", // HTML doctype
|
||||
"# encoding:", // Ruby encoding
|
||||
"# frozen_string_literal:", // Ruby interpreter instruction
|
||||
}
|
||||
|
||||
func hashBang(b []byte) []byte {
|
||||
|
||||
15
testdata/expected/Gemfile
vendored
Normal file
15
testdata/expected/Gemfile
vendored
Normal file
@@ -0,0 +1,15 @@
|
||||
# Copyright 2016 Google Inc.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
puts "Hello world!"
|
||||
16
testdata/expected/file.Dockerfile
vendored
Normal file
16
testdata/expected/file.Dockerfile
vendored
Normal file
@@ -0,0 +1,16 @@
|
||||
# Copyright 2016 Google Inc.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
FROM scratch
|
||||
CMD ["echo", "hello world"]
|
||||
17
testdata/expected/file.groovy
vendored
Normal file
17
testdata/expected/file.groovy
vendored
Normal file
@@ -0,0 +1,17 @@
|
||||
// Copyright 2016 Google Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
def dummy_function() {
|
||||
println "Hello world"
|
||||
}
|
||||
16
testdata/expected/file.rb
vendored
Normal file
16
testdata/expected/file.rb
vendored
Normal file
@@ -0,0 +1,16 @@
|
||||
# frozen_string_literal: true
|
||||
# Copyright 2016 Google Inc.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
puts "Hello world!"
|
||||
1
testdata/initial/Gemfile
vendored
Normal file
1
testdata/initial/Gemfile
vendored
Normal file
@@ -0,0 +1 @@
|
||||
puts "Hello world!"
|
||||
2
testdata/initial/file.Dockerfile
vendored
Normal file
2
testdata/initial/file.Dockerfile
vendored
Normal file
@@ -0,0 +1,2 @@
|
||||
FROM scratch
|
||||
CMD ["echo", "hello world"]
|
||||
3
testdata/initial/file.groovy
vendored
Normal file
3
testdata/initial/file.groovy
vendored
Normal file
@@ -0,0 +1,3 @@
|
||||
def dummy_function() {
|
||||
println "Hello world"
|
||||
}
|
||||
2
testdata/initial/file.rb
vendored
Normal file
2
testdata/initial/file.rb
vendored
Normal file
@@ -0,0 +1,2 @@
|
||||
# frozen_string_literal: true
|
||||
puts "Hello world!"
|
||||
Reference in New Issue
Block a user