Merge pull request #45 from fashing/master

Support Spanner schema and Skylark files.

main.go

@@ -219,13 +219,13 @@ func licenseHeader(path string, tmpl *template.Template, data *copyrightData) ([
 		lic, err = prefix(tmpl, data, "/**", " * ", " */")
 	case ".cc", ".cpp", ".cs", ".go", ".hh", ".hpp", ".java", ".m", ".mm", ".proto", ".rs", ".scala", ".swift", ".dart", ".groovy", ".kt", ".kts", ".v", ".sv":
 		lic, err = prefix(tmpl, data, "", "// ", "")
-	case ".py", ".sh", ".yaml", ".yml", ".dockerfile", "dockerfile", ".rb", "gemfile", ".tcl":
+	case ".py", ".sh", ".yaml", ".yml", ".dockerfile", "dockerfile", ".rb", "gemfile", ".tcl", ".bzl":
 		lic, err = prefix(tmpl, data, "", "# ", "")
 	case ".el", ".lisp":
 		lic, err = prefix(tmpl, data, "", ";; ", "")
 	case ".erl":
 		lic, err = prefix(tmpl, data, "", "% ", "")
-	case ".hs", ".sql":
+	case ".hs", ".sql", ".sdl":
 		lic, err = prefix(tmpl, data, "", "-- ", "")
 	case ".html", ".xml", ".vue":
 		lic, err = prefix(tmpl, data, "<!--", " ", "-->")

testdata/expected/file.bzl

@@ -0,0 +1,19 @@
+# Copyright 2018 Google LLC
+#
+# 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.
+
+# Say hello
+def hello():
+  print("Hello world!")
+
+hello()

testdata/expected/file.sdl

@@ -0,0 +1,22 @@
+-- Copyright 2018 Google LLC
+--
+-- 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.
+
+-- This is a fake table.
+CREATE TABLE Foo (
+  -- This is a fake column.
+  Bar INT64 NOT NULL,
+
+  -- Another fake column.
+  Baz STRING(MAX) NOT NULL,
+) PRIMARY KEY (Bar);

testdata/initial/file.bzl

@@ -0,0 +1,5 @@
+# Say hello
+def hello():
+  print("Hello world!")
+
+hello()

testdata/initial/file.sdl

@@ -0,0 +1,8 @@
+-- This is a fake table.
+CREATE TABLE Foo (
+  -- This is a fake column.
+  Bar INT64 NOT NULL,
+
+  -- Another fake column.
+  Baz STRING(MAX) NOT NULL,
+) PRIMARY KEY (Bar);