Skip to content

Latest commit

 

History

History
913 lines (609 loc) · 22.1 KB

stdlib.md

File metadata and controls

913 lines (609 loc) · 22.1 KB

Shac runtime standard library

Shac uses the starlark language. Starlark is a python derivative. https://bazel.build/rules/language is a great resource if the language is new to you, just ignore the bazel references. The starlark language formal specification is documented at https://github.com/google/starlark-go/blob/HEAD/doc/spec.md.

While all starlark-go's built-in constants and functions are available, a few are explicitly documented here to highlight them.

These experimental options are enabled:

  • Set: "set" built-in is enabled.
  • While: while statements are allowed. This allows potentially unbounded runtime.
  • Recursion: recursive function calls are allowed.

Table of contents

shac

shac is the global available at runtime when loading your starlark code.

Fields:

  • check
  • commit_hash
  • register_check
  • version

shac.commit_hash

The git hash of shac.git where shac was built.

shac.version

The semver version number of shac.

shac.check

Constructs a shac check object.

Example

def cb(ctx):
    fail("implement me")

fail_often = shac.check(cb, name="fail_often")

shac.register_check(fail_often)

Arguments

  • impl: Starlark function that is called back to implement the check. The callback must accept one ctx(...) argument and return None.
  • name: (optional) Name of the check. Defaults to the callback function name.
  • formatter: (optional) Whether the check is a formatter. If set to True, the formatter will be run and have its results written to disk by shac fmt.

shac.register_check

Registers a shac check.

It must be called at least once for the starlark file to be a valid check file. Each callback will be run in parallel. Each check must have a different name.

Example

def cb(ctx):
    fail("implement me")

fail_often = shac.check(cb, name="fail_often")

shac.register_check(fail_often)

register_check also accepts a bare function for convenience when registering simple checks. The callback function name will be used as the check name.

def fail_often(ctx):
    fail("implement me")

shac.register_check(cb, fail_often)

Arguments

  • check: shac.check() object or Starlark function that is called back to implement the check.

check

check is the object returned by shac.check().

Fields:

  • with_args
  • with_name

check.with_args

Create a copy of the check with keyword arguments overridden.

Example

def cb(ctx, level = "warning"):
    ctx.emit.finding(
        level = level,
        message = "Found an issue",
    )

warning_check = shac.check(cb)
error_check = warning_check.with_args(level = "error")

Arguments

  • **kwargs: Overridden keyword arguments.

check.with_name

Create a copy of the check with name overridden.

Useful for changing the name of checks provided by shared Starlark libraries.

Example

Shared library code:

def _check_with_bad_name(ctx):
  pass

check_with_bad_name = shac.check(_check_with_bad_name)

Downstream code:

load("@library", "check_with_bad_name")

shac.register_check(check_with_bad_name.with_name("better_name"))

Arguments

  • name: The new name of the check.

ctx

ctx is the object passed to shac.register_check(...) callback.

Fields:

  • emit
  • io
  • os
  • platform
  • re
  • scm
  • vars

ctx.emit

ctx.emit is the object that exposes the API to emit results for checks.

Fields:

  • finding
  • artifact

ctx.emit.finding

Emits a finding from the current check.

Example

A check level finding:

def cb(ctx):
    ctx.emit.finding(
        level="warning",
        message="Do not change anything",
    )

shac.register_check(cb)

A finding associated with a specific file:

def cb(ctx):
    for path, _ in ctx.scm.affected_files().items():
        ctx.emit.finding(
            level="notice",
            message="great code",
            filepath=path,
        )

shac.register_check(cb)

A finding associated with a specific line within a file:

def cb(ctx):
    for path, meta in ctx.scm.affected_files().items():
        for num, line in meta.new_lines():
            ctx.emit.finding(
                level="error",
                message="This line is superfluous",
                filepath=path,
                line=num,
                # Suggesting an empty string as a replacement results in the line
                # being deleted when applying the replacement.
                replacements=[""],
            )

shac.register_check(cb)

A finding associated with a line and column within a file:

def cb(ctx):
    for path, meta in ctx.scm.affected_files().items():
        for num, line in meta.new_lines():
            idx = str.find("bad_word")
            if idx < 0:
                continue
            ctx.emit.finding(
                level="error",
                message="Do not use bad_word",
                filepath=path,
                line=num,
                col=idx+1,
                end_col=idx+1+len("bad_word"),
                replacements=["best_word", "good_word"],
            )

shac.register_check(cb)

Arguments

  • level: One of "notice", "warning" or "error".
  • message: Message of the finding. May be omitted for checks marked with formatter=True, as long as filepath is specified, level is "error", and replacements has a length of 1.
  • filepath: (optional) Path to the source file to annotate.
  • line: (optional) Line where the finding should start. 1 based.
  • col: (optional) Column where the finding should start. 1 based.
  • end_line: (optional) Line where the finding should end if it represents a span, inclusive. 1 based.
  • end_col: (optional) Column where the finding should end if it represents a span, exclusive. 1 based.
  • replacements: (optional) A sequence of str, representing possible replacement suggestions. The sequence can be a list or a tuple. The replacements apply to the entire file if no span is specified for the finding. Replacements that apply to entire lines should include trailing newlines, unless the line should be removed.

ctx.emit.artifact

Emits an artifact from the current check.

Example

def cb(ctx):
    ctx.emit.artifact("result.txt", "fake data")

shac.register_check(cb)

Arguments

  • filepath: File name of the artifact. The path must be relative and in POSIX format, using / separator.
  • content: (optional) Content. If content is omitted, the content of the file at filepath will be saved as an artifact.

ctx.io

ctx.io is the object that exposes the API to interact with the file system.

Fields:

  • read_file
  • tempdir
  • tempfile

ctx.io.read_file

Returns the content of a file.

Example

def cb(ctx):
    # Read at most 4Kib of "path/to/file.txt".
    content = str(ctx.io_read_file("path/to/file.txt", size=4096))
    # Usually run a regexp via ctx.re.match(), or other simple text
    # processing.
    print(content)

shac.register_check(cb)

Arguments

  • filepath: Path of the file to read. The file must be within the workspace. The path must be relative and in POSIX format, using / separator.
  • size: (optional) Limits the maximum number of bytes to return. The file is silently truncated to this value. The whole file is buffered in memory. Defaults to 128Mib on 32 bits runtime, 4Gib on 64 bits runtime.

Returns

Content of the file as bytes.

ctx.io.tempdir

Returns a new temporary directory.

ctx.io.tempfile

Returns a new temporary file.

Arguments

  • content: String or bytes to populate the file with.
  • name: (optional) The basename to give the file. May contain path separators, in which case the file will be nested accordingly. Will be chosen randomly if not specified.

Returns

Absolute path to the created file.

ctx.os

ctx.os is the object that exposes the API to interact with the operating system.

Fields:

  • exec

ctx.os.exec

Runs a command as a subprocess.

Subprocesses are denied network access by default on Linux. Use allow_network = True to grant the subprocess network access.

Example

def cb(ctx):
    res = ctx.os.exec(["echo", "hello world"], cwd=".").wait()
    print(res.stdout)  # "hello world"

shac.register_check(cb)

Use raise_on_failure = False to prevent a non-zero retcode from automatically failing the check:

def cb(ctx):
    res = ctx.os.exec(
        ["cat", "does-not-exist.txt"],
        raise_on_failure = False,
    ).wait()
    print(res.retcode)  # 1
    print(res.stderr)   # cat: does-not-exist.txt: No such file or directory

shac.register_check(cb)

Use env to pass environment variables:

def cb(ctx):
    ctx.os.exec(["foo"], env = {"FOO_CONFIG": "foo.config"}).wait()

shac.register_check(cb)

Arguments

  • cmd: Subprocess command line.
  • cwd: (optional) Relative path to cwd for the subprocess. Defaults to the directory containing shac.star.
  • env: (optional) Dictionary of environment variables to set for the subprocess.
  • stdin: (optional) str or bytes to pass to the subprocess as standard input.
  • allow_network: (optional) Allow network access. Defaults to false.
  • ok_retcodes: (optional) List of exit codes that should be considered successes. Any other exit code will immediately fail the check. The effective default is [0].
  • raise_on_failure: (optional) Whether the running check should automatically fail if the subcommand returns a non-zero exit code. Defaults to true. Cannot be false if ok_retcodes is also set.

Returns

A subprocess object with a wait() method. wait() returns a struct(retcode=..., stdout="...", stderr="...")

ctx.platform

ctx.platform exposes data about the underlying platform.

Fields:

  • os
  • arch

ctx.platform.os

ctx.platform.os contains the OS as described by GOOS. Frequent values are "windows", "linux" and "darwin". The full exact list can be retrieved with the command "go tool dist list | cut -f 1 -d / | uniq"

ctx.platform.arch

ctx.platform.arch contains the CPU architecture as described by GOARCH. Frequent values are "amd64" and "arm64". The full exact list can be retrieved with the command "go tool dist list | cut -f 2 -d / | sort | uniq"

ctx.re

ctx.re is the object that exposes the API to run regular expressions on starlark strings.

Fields:

  • allmatches
  • match

ctx.re.allmatches

Returns all the matches of the regexp pattern onto content.

Example

def cb(ctx):
    content = str(ctx.io_read_file("path/to/file.txt"))
    for match in ctx.re.allmatches("TODO\(([^)]+)\).*", content):
        print(match)

shac.register_check(cb)

Arguments

Returns

list(struct(offset=bytes_offset, groups=list(matches)))

ctx.re.match

Returns the first match of the regexp pattern onto content.

Example

def cb(ctx):
    content = str(ctx.io_read_file("path/to/file.txt"))
    # Only print the first match, if any.
    match = ctx.re.match("TODO\(([^)]+)\).*", "content/true")
    print(match)

shac.register_check(cb)

Arguments

Returns

struct(offset=bytes_offset, groups=list(matches))

ctx.scm

ctx.scm is the object exposes the API to query the source control management (e.g. git).

Fields:

  • root
  • affected_files
  • all_files

ctx.scm.root

ctx.scm.root is the absolute path to the project root.

ctx.scm.affected_files

Returns affected files as determined by the SCM.

If shac detected that the tree is managed by a source control management system, e.g. git, it will detect the upstream branch and return only the files currently modified.

If the current directory is not controlled by a SCM, the result is equivalent to ctx.scm.all_files().

If shac is run with the --all options, all files are considered "added" to do a full run on all files.

Example

def new_todos(ctx):
    # Prints only the TODO that were added compared to upstream.
    for path, meta in ctx.scm.affected_files().items():
        for num, line in meta.new_lines():
            m = ctx.re.match("TODO\(([^)]+)\).*", line)
            print(path + "(" + str(num) + "): " + m.groups[0])

shac.register_check(new_todos)

Arguments

  • glob: (optional) TODO: Will later accept a glob.
  • include_deleted: (optional) Whether to include deleted files. By default deleted files are excluded.

Returns

A map of {path: struct()} where the struct has a string field action and a function new_lines().

The action field's value is a single letter corresponding to the --diff-filter representation of the action per https://git-scm.com/docs/git-diff#Documentation/git-diff.txt---diff-filterACDMRTUXB82308203

ctx.scm.all_files

Returns all files found in the current workspace.

All files are considered "added" or "deleted".

Example

def all_todos(ctx):
    for path, meta in ctx.scm.all_files().items():
        for num, line in meta.new_lines():
            m = ctx.re.match("TODO\(([^)]+)\).*", line)
            print(path + "(" + str(num) + "): " + m.groups[0])

shac.register_check(all_todos)

Arguments

  • glob: (optional) TODO: Will later accept a glob.
  • include_deleted: (optional) Whether to include deleted files. By default deleted files are excluded.

Returns

A map of {path: struct()} where the struct has a string field action and a function new_lines().

The action field's value is a single letter corresponding to the --diff-filter representation of the action per https://git-scm.com/docs/git-diff#Documentation/git-diff.txt---diff-filterACDMRTUXB82308203

If a file was not modified relative to the upstream commit, the action field will be an empty string.

ctx.vars

ctx.vars provides access to runtime-configurable variables.

Fields:

  • get

ctx.vars.get

Returns the value of a runtime-configurable variable.

The value may be specified at runtime by using the --var name=value flag when running shac. In order to be set at runtime, a variable must be registered in shac.textproto.

If not set via the command line, the value will be the default declared in shac.textproto.

Raises an error if the requested variable is not registered in the project's shac.textproto config file.

Example

def cb(ctx):
    build_dir = ctx.vars.get("build_directory")
    ctx.os.exec([build_dir + "/compiled_tool"]).wait()

shac.register_check(cb)

Arguments

  • name: The name of the variable.

Returns

A string corresponding to the current value of the variable.

dir

Starlark builtin that returns all the attributes of an object.

Primarily used to explore and debug a starlark file.

See the official documentation at https://github.com/google/starlark-go/blob/HEAD/doc/spec.md#dir.

Example

def print_attributes(name, obj):
    for attrname in dir(obj):
      attrval = getattr(obj, attrname)
      attrtype = type(attrval)
      fullname = name + "." + attrname
      if attrtype in ("builtin_function_or_method", "function"):
          print(fullname + "()")
      elif attrtype == "struct":
          print_attributes(fullname, attrval)
      else:
          print(fullname + "=" + repr(attrval))

def cb(ctx):
    print_attributes("ctx", ctx)
    print_attributes("str", "")
    print_attributes("dict", {})
    print_attributes("set", set())
    print_attributes("struct", struct(foo = "bar", p = print_attributes))

shac.register_check(cb)

Arguments

  • x: Object that has its properties enumerated.

Returns

List of x object properties as strings. You can use getattr() to retrieve each attributes in a loop.

fail

Starlark builtin that fails immediately the execution.

This function will abort execution. When called in the first phase, outside a check callback, it will prevent execution of checks.

When called within a check, it stops the check execution and annotates it with an abnormal failure. It also prevents checks that were not yet started from running.

See the official documentation at https://github.com/google/starlark-go/blob/HEAD/doc/spec.md#fail.

Example

The code after the fail() call will not be executed:

fail("implement me")

The check will be annotated with the abnormal failure:

def cb1(ctx):
    fail("implement me")

def cb2(ctx):
    # This check may or may not run, depending on concurrency.
    pass

shac.register_check(cb1)
shac.register_check(cb2)

Arguments

  • *args: Arguments to print out.
  • sep: (optional) Separator between the items in args. Defaults to " ".

json

json is a global module that exposes json functions.

The documentation here is listed as a struct instead of a module. The two are functionally equivalent.

The implementation matches the official bazel's documentation at https://bazel.build/rules/lib/json except that encode_indent is not implemented.

Fields:

  • decode
  • encode
  • indent

json.decode

Decodes a JSON encoded string into the Starlark value that the string denotes.

Supported types include null, bool, int, float, str, dict and list.

See the full documentation at https://bazel.build/rules/lib/json#decode.

Example

data = json.decode('{"foo":"bar}')
print(data["foo"])

def cb(ctx):
    # Load a configuration from a json file in the tree, containing a
    # dict with a "version" key.
    decoded = ctx.io.read_file("config.json")
    print(decoded["version"])

shac.register_check(cb)

Arguments

  • x: String or bytes of JSON encoded data to convert back to starlark.

json.encode

Encodes the starlark value into a JSON encoded string.

Supported types include null, bool, int, float, str, dict, list and struct.

See the full documentation at https://bazel.build/rules/lib/json#encode.

Example

config = struct(
    foo = "bar",
)
print(json.encode(config))

Arguments

  • x: Starlark value to encode to a JSON encoded string.

json.indent

Returns the indented form of a valid JSON-encoded string.

See the full documentation at https://bazel.build/rules/lib/json#indent.

Example

config = struct(
    foo = "bar",
)
d = json.encode(config)
print(json.indent(d))

Arguments

  • s: String or bytes of JSON encoded data to reformat.
  • prefix: (optional) Prefix for each new line. Defaults to "".
  • indent: (optional) Indent for nested fields. Defaults to " ".

load

Starlark builtin that loads an additional shac starlark package and make symbols (var, struct, functions) from this file accessible.

See the official documentation at https://github.com/google/starlark-go/blob/HEAD/doc/spec.md#name-binding-and-variables and at https://github.com/google/starlark-go/blob/HEAD/doc/spec.md#load-statements.

After a starlark module is loaded, its values are frozen as described at https://github.com/google/starlark-go/blob/HEAD/doc/spec.md#freezing-a-value.

Example

There are 5 forms for the module to load:

  1. Loads a file in the same directory as the calling code. In this case, ".." is allowed as long as it stays within the root directory of the current project:
load("../common/go.star", "gosec")

register_check(gosec)
  1. Loads a file relative to the root directory:
load("//common/go.star", "gosec")

register_check(gosec)
  1. Loading an external package via a package alias and default api.star. This uses the aliases defined in shac.textproto and loads api.star in this dependency:
# Implicitly loads api.star
load("@static-checks", "go")

register_check(go.gosec)
  1. Loading an external package via a package alias and a specific file. This uses the aliases defined in shac.textproto and loads the specified file in this dependency:
load("@static-checks//go.star", "go")

register_check(go.gosec)
  1. and 6. Loading a specific file in external package, using a fully qualified URI instead of an alias. The delimitation is still "//" between the resource and the path within the package:
# Implicitly loads api.star
load("@go.fuchsia.dev/shac-project/static-checks", "gosec")
# or
load("@go.fuchsia.dev/shac-project/static-checks//go.star", "gosec")

register_check(gosec)

Arguments

  • module: Path to a module to load. Three forms are accepted. The default is a relative path relative to the current module path. Second, it can be relative to the root of the project by using the "//" prefix. Third it can be an external package by using the "@" prefix. External references can either be fully qualified or an alias. Either way, it has to be specified in shac.textproto. An optional part, delimited by "//" can be appended, which specifies a file within the package. If omitted, the file api.star is loaded. When loading an external dependency and a path is specified, the path cannot contain ".." or "internal". Unlike bazel, the ":" form is not allowed.
  • *symbols: Symbols to load from the module.
  • **kwsymbols: Symbols to load from the module that will be accessible under a new name.

print

Starlark builtin that prints a debug log.

This function should only be used while debugging the starlark code. See the official documentation at https://github.com/google/starlark-go/blob/HEAD/doc/spec.md#print.

Example

print("shac", "is", "great")

Arguments

  • *args: Arguments to print out.
  • sep: (optional) Separator between the items in args. Defaults to " ".

struct

Creates and return a structure instance.

Create an "object" that has immutable properties. It is similar to a dictionary is usage. It is intentionally not as powerful as a python class instance.

Example

def _do():
    print("it works")

obj = struct(
    value = "a value",
    do = _do,
)

print(obj.value)
obj.do()

Arguments

  • **kwargs: structure's fields. The argument name becomes the property name, and the argument value becomes the property value.