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 {
|
func addLicense(path string, fmode os.FileMode, tmpl *template.Template, data *copyrightData) error {
|
||||||
var lic []byte
|
var lic []byte
|
||||||
var err error
|
var err error
|
||||||
switch filepath.Ext(path) {
|
switch fileExtension(path) {
|
||||||
default:
|
default:
|
||||||
return nil
|
return nil
|
||||||
case ".c", ".h":
|
case ".c", ".h":
|
||||||
lic, err = prefix(tmpl, data, "/*", " * ", " */")
|
lic, err = prefix(tmpl, data, "/*", " * ", " */")
|
||||||
case ".js", ".jsx", ".tsx", ".css", ".tf":
|
case ".js", ".jsx", ".tsx", ".css", ".tf":
|
||||||
lic, err = prefix(tmpl, data, "/**", " * ", " */")
|
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, "", "// ", "")
|
lic, err = prefix(tmpl, data, "", "// ", "")
|
||||||
case ".py", ".sh", ".yaml", ".yml":
|
case ".py", ".sh", ".yaml", ".yml", ".dockerfile", "dockerfile", ".rb", "gemfile":
|
||||||
lic, err = prefix(tmpl, data, "", "# ", "")
|
lic, err = prefix(tmpl, data, "", "# ", "")
|
||||||
case ".el", ".lisp":
|
case ".el", ".lisp":
|
||||||
lic, err = prefix(tmpl, data, "", ";; ", "")
|
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)
|
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{
|
var head = []string{
|
||||||
"#!", // shell script
|
"#!", // shell script
|
||||||
"<?xml", // XML declaratioon
|
"<?xml", // XML declaratioon
|
||||||
"<!doctype", // HTML doctype
|
"<!doctype", // HTML doctype
|
||||||
|
"# encoding:", // Ruby encoding
|
||||||
|
"# frozen_string_literal:", // Ruby interpreter instruction
|
||||||
}
|
}
|
||||||
|
|
||||||
func hashBang(b []byte) []byte {
|
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