Lint include guards copyright [AP-243]

IncludeGuards.cmake

@@ -0,0 +1,51 @@
+#
+# Copyright (C) 2021 Swift Navigation Inc.
+# Contact: Swift Navigation <dev@swift-nav.com>
+#
+# This source is subject to the license found in the file 'LICENSE' which must
+# be be distributed together with this source. All other rights reserved.
+#
+# THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND,
+# EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED
+# WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A PARTICULAR PURPOSE.
+#
+
+#
+# A module to check header files for:
+# - Include guards
+# - Copyright notice
+#
+# When included will create 2 new targets:
+# - fix-include-guards - Check all header files in the repository for the
+#   above conditions
+# - fix-include-guards-check - Depends on fix-include-guards then exits with
+#   an error code if any changes were applied. Useful for enforcing checks
+#   during CI runs
+#
+# Include guards are formatted according to the header file's location in the
+# source tree. Copyright notices are boiler plate block comments
+#
+# Changes are applied in place
+#
+
+add_custom_target(fix-include-guards-${PROJECT_NAME}
+  COMMAND
+    ${CMAKE_CURRENT_LIST_DIR}/scripts/fix_include_guards.py `git ls-files '*.h'`
+  WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}
+)
+add_custom_target(fix-include-guards-check-${PROJECT_NAME}
+  COMMAND git diff --exit-code
+  DEPENDS fix-include-guards-${PROJECT_NAME}
+  WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}
+)
+
+if(${PROJECT_NAME} STREQUAL ${CMAKE_PROJECT_NAME})
+  add_custom_target(
+    fix-include-guards
+    DEPENDS fix-include-guards-${PROJECT_NAME}
+  )
+  add_custom_target(
+    fix-include-guards-check
+    DEPENDS fix-include-guards-check-${PROJECT_NAME}
+  )
+endif()

scripts/fix_include_guards.py

@@ -0,0 +1,143 @@
+#!/usr/bin/env python3
+
+#
+# Copyright (C) 2023 Swift Navigation Inc.
+# Contact: Swift Navigation <dev@swift-nav.com>
+#
+# This source is subject to the license found in the file 'LICENSE' which must
+# be be distributed together with this source. All other rights reserved.
+#
+# THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND,
+# EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED
+# WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A PARTICULAR PURPOSE.
+#
+
+import sys
+import re
+from pathlib import Path
+from datetime import datetime
+
+"""
+Don't check for header guards if there are error suppression
+comments somewhere in this file.
+
+Because this is silencing a warning for a nonexistent line, we
+only support the very specific NOLINT(build/header_guard) syntax,
+and not the general NOLINT or NOLINT(*) syntax.
+"""
+def can_ignore_file(lines):
+    if any("NOLINT(build/header_guard)" in line for line in lines) or any(
+        "#pragma once" in line for line in lines
+    ):
+        return True
+    return False
+
+
+def get_expected_guard(filename):
+    file_path = Path(filename)
+
+    # Check if "include" is a direct part of the path
+    if "include" in file_path.parts:
+        include_index = file_path.parts.index("include")
+        expected_guard_parts = file_path.parts[include_index + 1 :]
+    else:
+        # For files outside "include" directory
+        if file_path.parent.name == "src":
+            expected_guard_parts = file_path.parts[-3:]
+        else:
+            expected_guard_parts = file_path.parts[-2:]
+
+    expected_guard = "_".join(expected_guard_parts)
+    expected_guard = expected_guard.replace("++", "cpp")
+    expected_guard = re.sub(r"[/\.-]", "_", expected_guard).upper()
+
+    return expected_guard
+
+
+def find_ifndef_define(lines):
+    ifndef, ifndef_linenum = None, -1
+    define, define_linenum = None, -1
+
+    for linenum, line in enumerate(lines):
+        if line.startswith("#ifndef ") and ifndef is None:
+            _, ifndef = line.split(maxsplit=1)
+            ifndef_linenum = linenum
+
+        if line.startswith("#define ") and define is None:
+            _, define = line.split(maxsplit=1)
+            define_linenum = linenum
+
+        # Exit loop if both ifndef and define are found
+        if ifndef and define:
+            break
+
+    return ifndef, ifndef_linenum, define, define_linenum
+
+
+def fix_header_guard(filename):
+    try:
+        with open(filename, "r") as target_file:
+            lines = target_file.read().split("\n")
+
+        for linenum in range(len(lines) - 1):
+            if lines[linenum].endswith("\r"):
+                lines[linenum] = lines[linenum].rstrip("\r")
+    except IOError:
+        sys.stderr.write("Error opening {}\n".format(filename))
+        return
+
+    if can_ignore_file(lines):
+        return
+
+    expected_guard = get_expected_guard(filename)
+    ifndef, ifndef_linenum, define, define_linenum = find_ifndef_define(lines)
+
+    if (
+        ifndef_linenum == -1
+        or ifndef != expected_guard
+        or define_linenum == -1
+        or define != expected_guard
+    ):
+        if ifndef_linenum == -1:
+            # file doesn't have an include guard so generate one
+            lines.insert(
+                0, "#ifndef %s\n#define %s\n" % (expected_guard, expected_guard)
+            )
+            lines.append("\n#endif")
+        else:
+            # need to fix
+            lines[ifndef_linenum] = "#ifndef " + expected_guard
+            lines[define_linenum] = "#define " + expected_guard
+
+    # Check for copyright notice
+    default_copyright = f"""/**
+ * Copyright (C) {datetime.now().year} Swift Navigation Inc.
+ * Contact: Swift Navigation <dev@swiftnav.com>
+ *
+ * This source is subject to the license found in the file 'LICENSE' which must
+ * be be distributed together with this source. All other rights reserved.
+ *
+ * THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND,
+ * EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A PARTICULAR PURPOSE.
+ */"""
+
+    # We'll say it should occur by line 10. Don't forget there's a
+    # placeholder line at the front.
+    for line in range(1, min(len(lines), 11)):
+        if re.search(r"Copyright", lines[line], re.I):
+            break
+    else:  # means no copyright line was found
+        lines.insert(0, default_copyright)
+
+    with open(filename, "w") as output_file:
+        output_file.write("\n".join(lines))
+
+
+def main():
+    for filename in sys.argv[1:]:
+        fix_header_guard(filename)
+
+
+if __name__ == "__main__":
+    main()