Tools/Addlicense
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #12 from google/xml-html

Browse Source 
Better support for XML and HTML
This commit is contained in:
alex
2018-07-21 22:40:47 +02:00
committed by GitHub
parent 95943e310c 071b332aa8
commit f9cb8b9a1d
4 changed files with 39 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
main.go
View File

@@ -24,6 +24,7 @@ import (
"log"
"os"
"path/filepath"
"strings"
"sync"
"time"
)
@@ -155,6 +156,12 @@ func addLicense(path string, fmode os.FileMode, typ string, data *copyrightData)
return ioutil.WriteFile(path, b, fmode)
}
var head = []string{
"#!", // shell script
"<?xml", // XML declaratioon
"<!doctype", // HTML doctype
}
func hashBang(b []byte) []byte {
var line []byte
for _, c := range b {
@@ -163,8 +170,11 @@ func hashBang(b []byte) []byte {
break
}
}
if bytes.HasPrefix(line, []byte("#!")) {
return line
first := strings.ToLower(string(line))
for _, h := range head {
if strings.HasPrefix(first, h) {
return line
}
}
return nil
}

2
testdata/expected/file.html vendored
View File

@@ -1,3 +1,4 @@
<!doctype html>
<!--
Copyright 2016 Google Inc.
@@ -14,5 +15,4 @@
limitations under the License.
-->
<!doctype html>
<p>Hello World!</p>

21
testdata/expected/file2.xml vendored Normal file
View File

@@ -0,0 +1,21 @@
<?xml version="1.0" encoding="UTF-8" standalone="no" ?>
<!--
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.
-->
<root>
<one>one</one>
<two/>
</root>

5
testdata/initial/file2.xml vendored Normal file
View File

@@ -0,0 +1,5 @@
<?xml version="1.0" encoding="UTF-8" standalone="no" ?>
<root>
<one>one</one>
<two/>
</root>