Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 13 additions & 0 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,19 @@ repos:

- repo: local
hooks:
- id: script-tests
name: script-tests
language: system
entry: python3 -m unittest discover --start-directory scripts --pattern '*.py'
pass_filenames: false
files: ^scripts/.*\.py$

- id: check-namespaced-defines
name: check-namespaced-defines
language: system
entry: python3 scripts/check_namespaced_defines.py
files: ^deploy/helm/[^/]+/templates/

- id: regenerate-charts
name: regenerate-charts
language: system
Expand Down
95 changes: 95 additions & 0 deletions scripts/check_namespaced_defines.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,95 @@
"""
Fails when a Helm defined template, or a call to one, is not prefixed with the
name of the chart it lives in.

To run tests for this script:
python3 -m unittest check_namespaced_defines.py
"""

import re
import unittest
from pathlib import Path

NAME_PATTERN = r'\b(?:define|include|template)\s+"([^"]+)"'
CHART_PATTERN = r"(?:^|/)deploy/helm/(?P<chart>[^/]+)/templates/"


def chart_of(path):
"""The chart whose templates directory holds path, or None if it is outside one."""
found = re.search(CHART_PATTERN, Path(path).as_posix())
return found.group("chart") if found else None


def unprefixed(file_contents, chart_name):
"""The defined template names in file_contents, and the calls to them, lacking the chart prefix."""
names = re.findall(NAME_PATTERN, file_contents)
return sorted({name for name in names if not name.startswith(f"{chart_name}.")})


# Run these with: python3 -m unittest check_namespaced_defines.py
class TestCoreMethods(unittest.TestCase):
def test_chart_of(self):
self.assertEqual(
chart_of("deploy/helm/trino-operator/templates/x.yaml"), "trino-operator"
)
# An absolute path has to give the same answer, or a manual run invents a chart name.
self.assertEqual(
chart_of("/tmp/wt/deploy/helm/secret-operator/templates/a/b.yaml"),
"secret-operator",
)
self.assertIsNone(chart_of("deploy/helm/trino-operator/values.yaml"))
self.assertIsNone(chart_of("README.md"))

def test_prefixed_is_accepted(self):
text = '{{- define "trino-operator.labels" -}}\n{{ include "trino-operator.chart" . }}\n'
self.assertEqual(unprefixed(text, "trino-operator"), [])

def test_unprefixed_define_is_reported(self):
self.assertEqual(
unprefixed('{{- define "helper.thing" -}}', "trino-operator"),
["helper.thing"],
)

def test_unprefixed_call_is_reported(self):
# A renamed definition whose call sites did not move with it stops the chart rendering.
text = (
'{{ include "operator.fullname" . }}\n{{ template "operator.labels" . }}\n'
)
self.assertEqual(
unprefixed(text, "trino-operator"), ["operator.fullname", "operator.labels"]
)

def test_keyword_must_stand_alone(self):
# Without a word boundary any identifier ending in the keyword matches.
self.assertEqual(
unprefixed('{{ .Values.xinclude "bar.baz" }}', "foo-operator"), []
)


if __name__ == "__main__":
import sys

if not sys.argv[1:]:
print(f"usage: {sys.argv[0]} deploy/helm/<chart>/templates/*", file=sys.stderr)
sys.exit(2)

failed = False
for path in sys.argv[1:]:
chart_name = chart_of(path)
if chart_name is None:
print(
f"{sys.argv[0]}: {path} is not inside deploy/helm/<chart>/templates/",
file=sys.stderr,
)
sys.exit(2)
file_contents = Path(path).read_text(encoding="utf-8")
unprefixed_names = unprefixed(file_contents, chart_name)
if unprefixed_names:
failed = True
print(
f"{path}: defined templates and the calls to them must be prefixed with '{chart_name}.'"
)
for name in unprefixed_names:
print(f" {name}")

sys.exit(1 if failed else 0)