add copyright hook

.pre-commit-config.yaml

@@ -16,16 +16,25 @@ repos:
     -   id: detect-private-key
         files: (?!.*tar.gz)^.*$ 
     -   id: end-of-file-fixer
-        files: (src).*\.(md|py|mm|swift|java|c|cc|cxx|cpp|cu|h|hpp|hxx)$
+        files: (src).*\.(md|py|mm|swift|java|c|cc|cxx|cpp|h|hpp|hxx)$
     -   id: trailing-whitespace
-        files: (src).*\.(md|py|mm|swift|java|c|cc|cxx|cpp|cu|h|hpp|hxx)$
+        files: (src).*\.(md|py|mm|swift|java|c|cc|cxx|cpp|h|hpp|hxx)$
+
+-   repo: local
+    hooks:
+    -   id: copyright
+        name: copyright
+        entry: python ./tools/pre-commit.hooks/copyright.hook
+        language: system
+        files: (src).*\.(c|cc|cxx|cpp|h|hpp|hxx|py)$
+        exclude: (?!.*third_party)^.*$ | (?!.*book)^.*$
 
 -   repo: local
     hooks:
     -   id: clang-format
         name: clang-format
         description: Format files with ClangFormat.
-        entry: bash ./tools/pre-commit.hooks/.clang-format.hook -i
+        entry: bash ./tools/pre-commit.hooks/clang-format.hook -i
         language: system
         files: (src).*\.(c|cc|cxx|cpp|h|hpp|hxx)$
 
@@ -44,15 +53,8 @@ repos:
     -   id: cpplint
         name: cpplint
         description: Check C++ code style using cpplint.
-        entry: bash ./tools/pre-commit.hooks/.cpplint.hook
+        entry: bash ./tools/pre-commit.hooks/cpplint.hook
         language: system
-        files: (src).*\.(c|cc|cxx|cpp|cu|h|hpp|hxx)$
+        files: (src).*\.(c|cc|cxx|cpp|h|hpp|hxx)$
         exclude: (?!.*third_party)^.*$ | (?!.*book)^.*$i | *\.pb\.cpp
-#-   repo: local
-#    hooks:
-#    -   id: copyright_checker
-#        name: copyright_checker
-#        entry: python ./tools/pre-commit.hooks/.copyright.hook
-#        language: system
-#        files: (src).*\.(c|cc|cxx|cpp|cu|h|hpp|hxx|proto|py)$
-#        exclude: (?!.*third_party)^.*$ | (?!.*book)^.*$
+

tools/pre-commit.hooks/.copyright.hook → tools/pre-commit.hooks/copyright.hook

@@ -3,28 +3,22 @@ from __future__ import print_function
 from __future__ import unicode_literals
 
 import argparse
-import io, re
-import sys, os
-import subprocess
+import io
 import platform
+import re
+import subprocess
 
 COPYRIGHT = '''
-Copyright (c) 2016 Baidu, Inc. All Rights Reserved.
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserved.
+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.
 '''
 
 LANG_COMMENT_MARK = None
@@ -46,22 +40,27 @@ else:
 
 
 def generate_copyright(template, lang='C'):
+    end_line = ""
     if lang == 'Python':
-        LANG_COMMENT_MARK = '#'
+        lang_coment_mark = '# '
+        start = lang_coment_mark
+        blank = " "
     else:
-        LANG_COMMENT_MARK = "//"
-
+        lang_coment_mark = ""
+        start = "/* "
+        blank = ""
+        end_line = " */"
     lines = template.split(NEW_LINE_MARK)
-    BLANK = " "
-    ans = LANG_COMMENT_MARK + BLANK + COPYRIGHT_HEADER + NEW_LINE_MARK
+
+    ans = start + blank + COPYRIGHT_HEADER + NEW_LINE_MARK
+
     for lino, line in enumerate(lines):
-        if lino == 0 or lino == 1 or lino == len(lines) - 1: continue
-        if len(line)  == 0:
-            BLANK = ""
+        if lino == 0 or lino == 1 or lino == len(lines) - 1:
+            continue
+        if lino == (len(lines) - 2):
+            ans += lang_coment_mark + blank + line + end_line + NEW_LINE_MARK
         else:
-            BLANK = " "
-        ans += LANG_COMMENT_MARK + BLANK + line + NEW_LINE_MARK
-
+            ans += lang_coment_mark + blank + line + NEW_LINE_MARK
     return ans + "\n"
 
 
@@ -78,14 +77,6 @@ def lang_type(filename):
         return "C"
     elif filename.endswith(".cpp"):
         return "C"
-    elif filename.endswith(".cu"):
-        return "C"
-    elif filename.endswith(".cuh"):
-        return "C"
-    elif filename.endswith(".go"):
-        return "C"
-    elif filename.endswith(".proto"):
-        return "C"
     else:
         print("Unsupported filetype %s", filename)
         exit(0)
@@ -105,9 +96,10 @@ def main(argv=None):
         fd = io.open(filename, encoding="utf-8")
         first_line = fd.readline()
         second_line = fd.readline()
-        if "COPYRIGHT (C)" in first_line.upper(): continue
-        if first_line.startswith("#!") or PYTHON_ENCODE.match(
-                second_line) != None or PYTHON_ENCODE.match(first_line) != None:
+        if "COPYRIGHT (C)" in first_line.upper() or "COPYRIGHT (C)" in second_line.upper():
+            continue
+        if first_line.startswith("/*") or first_line.startswith("#!") or PYTHON_ENCODE.match(
+                second_line) is not None or PYTHON_ENCODE.match(first_line) is not None:
             continue
         original_contents = io.open(filename, encoding="utf-8").read()
         new_contents = generate_copyright(
@@ -116,9 +108,14 @@ def main(argv=None):
         retv = 1
         with io.open(filename, 'w') as output_file:
             output_file.write(new_contents)
-
     return retv
 
 
+def test_generate_copyright():
+    print(generate_copyright(COPYRIGHT))
+
+
 if __name__ == '__main__':
+    # test_generate_copyright()
     exit(main())
+