# Copyright 2023 The Shac Authors
#
# 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.

"""Checks for shac itself

This file will evolve as new shac functionality is being added.
"""

load("//checks/buildifier.star", "buildifier")
load("//checks/check_doc.star", "check_docs")
load("//checks/go.star", "gofmt", "gosec", "ineffassign", "no_fork_without_lock", "shadow", "staticcheck")
load("//checks/licenses.star", "check_license_headers")

def suggest_version_bump(ctx):
    affected_files = set(ctx.scm.affected_files())
    if any([f.endswith(".go") and not f.endswith("_test.go") for f in affected_files]):
        version_file = "internal/engine/version.go"
        if "internal/engine/version.go" not in affected_files:
            ctx.emit.finding(
                level = "notice",
                message = "Consider updating the shac version when making API changes.",
                filepath = version_file,
            )

def _is_todo_valid(ctx, s):
    """Returns True if the x part of "TODO(x): y" is valid."""

    # For some project, it could be a bug number or an URL to a bug.
    return bool(ctx.re.match("^[a-z]+$", s))

def new_todos(ctx):
    """Prints the added TODOs.

    Args:
      ctx: A ctx instance.
    """
    for path, meta in ctx.scm.affected_files().items():
        for num, line in meta.new_lines():
            m = ctx.re.match("TODO\\(([^)]+)\\).*", line)
            if not m:
                continue

            # TODO(maruel): Have ctx.re.match() return the offset since it's
            # inefficient to calculate back.
            if _is_todo_valid(ctx, m.groups[1]):
                level = "notice"
                message = m.groups[0]
            else:
                level = "error"
                message = "Use a valid username in your TODO, %r is not valid" % m.groups[1]
            ctx.emit.finding(
                level = level,
                message = message,
                filepath = path,
                line = num,
                col = line.index(m.groups[0]) + 1,
                end_line = num,
                end_col = len(line) + 1,
            )

shac.register_check(buildifier)
shac.register_check(check_docs)
shac.register_check(check_license_headers)
shac.register_check(gofmt)
shac.register_check(gosec)
shac.register_check(ineffassign)
shac.register_check(new_todos)
shac.register_check(no_fork_without_lock)
shac.register_check(shadow)
shac.register_check(staticcheck)
shac.register_check(suggest_version_bump)