Skip to content

A collection of utility scripts to simplify common tasks in shell scripts, templates for Makefile and testing/building scripts for C++ projects.

Notifications You must be signed in to change notification settings

massenz/common-utils

Repository files navigation

common-utils Shell Utilities


License Build & Test Release

C++ OS

A collection of utility scripts to simplify common tasks in shell scripts, templates for Makefile and testing/building scripts for C++ projects.

Common Utils

Why these utilities

Remember that:

If you are writing a script that is more than 100 lines long, or that uses non-straightforward control flow logic, you should rewrite it in a more structured language now. Bear in mind that scripts grow. Rewrite your script early to avoid a more time-consuming rewrite at a later date.

Google Shell Style Guide

In order to keep the size and complexity of Bash scripts down to a minimal size, and yet retain enough expressivity within the script for some primitive functionality, I have decided to factor out all the commonality and just source it in my scripts.

To paraphrase the authors of the style guide: this repository "is more a recognition of its use rather than a suggestion that it be used for widespread deployment".

Contents

Install

The easiest way to install is to use the installer script:

export COMMON_UTILS=/path/to/common-utils
export VERSION=...
curl -s -L https://cdn.githubraw.com/massenz/common-utils/$VERSION/install.sh | zsh -s

with a recent Release for the VERSION string.

NOTE

If you are using the Bourne Shell (bash) replace zsh in the command above with bash

The initialization necessary to use the common-utils is written out to $HOME/.commonrc either copy it to your shell's initialization script (.zshrc if you are using the Z Shell; .bashrc for the Bourne Shell) or source it directly from there.

Alternatively, you can simply download the tarball from the Releases page and do the above manually; this is what is needed to properly use the utils, in your .zshrc:

export COMMON_UTILS=/path/to/common-utils
export PATH=$PATH:${COMMON_UTILS}
source ${COMMON_UTILS}/utils

Usage

The commons.cmake can then be used inside other projects to include these files, as needed.

For an example, see this project.

To add the functions defined in utils.sh use something like:

source ${COMMON_UTILS}/utils

It is recommended that you add $COMMON_UTILS to your system's PATH:

export PATH=$PATH:$COMMON_UTILS

Even better, use the Common Utilities:

source ${COMMON_UTILS}/utils && \
    addpath ${COMMON_UTILS} && \
    success "Added ${COMMON_UTILS} to PATH"

Utils

The utils script contains a collection of simple utilities for shell scripts (see utils.sh for a full description of each command).

  • general file and path handlers:
abspath [FILE | PATH]
addpath PATH
findfile [--dir DIR] FILE
  • logging facilities (emit a timestamp and a log level):
msg MSG1 MSG2 ...
errmsg MSG1 MSG2 ...
success MSG
fatal MSG
  • command wrappers:
wrap CMD [ARGS...]
wrap_no_out ERRMSG CMD [ARGS...]
  • general utilities:
killn PROC-NAME
newenv NAME
now

Command-Line Argument Parser

There is something to be said for the immediacy of using shell scripts, especially when dealing with relatively simple system operations; however, parsing command line arguments has always been rather cumbersome and usually done along the lines of painful if [[ ${1} == '--build' ]] ....

On the other hand, Python is pretty convenient for system operations (especially when using the sh module), but sometimes a bit of an overkill, or just missing the immediacy of a simple shell script; however, the argparse module is nothing short of awesome, when it comes to power and flexibility in parsing command line options.

This simple Python script tries to marry the best of both worlds, allowing with a relatively simple setup to parse arbitrary command line options, and then having their values reflected in the corresponding local environment variables.

Usage

The usage is rather straightforward: we invoke it with a list of the desired option names, followed by the actual command line arguments ($@) separated with --.

For example:

source $(./parse_args keep- take counts! mount -- $@)

The - indicates a boolean flag (its presence will set the associated variable, no value expected); the ! indicates a required argument.

The values of the arguments (if any) are then available via the ${ } operator:

if [[ -n ${keep} ]]; then
  echo "Keeping mount: ${mount}"
fi

Modifiers

Each option by default indicates a named --option argument that expects a value:

--mounts 3

The argument is optional and if it's not present the corresponding variable will be unset ([[-z ${mount} ]] would return true).

An optional trailing modifier changes the meaning of the argument:

  • ! : indicates a required argument, its absence will cause an error;
  • - : designates a boolean argument, which takes no value and whose presence will result in the corresponding variable to be set);
  • + : a positional, required, argument;
  • ? : an optional positional argument.
  • * : an array of positional arguments (possibly empty, if none is present); should be obviously last.

NOTE

"Positional" arguments are those which are not preceded by a --arg flag and whose order matters when the command line is parsed. As such, obviously, an optional positional (or an array) must be the last in the list.

It is best to avoid combining optional positionals (?) and positional arrays (*) as they may result in unexpected behavior.

For example (see the parse_example script):

source $(./parse_args keep- counts! take mount+ -- $@)

will result in:

$ ./parse_example --keep --take 3 --counts yes /var/loc/bac

Keeping mount: /var/loc/bac
Take: 3, counts: yes

$ ./parse_example.sh --keep --take 3 /mnt/media

usage: [-h] [--keep] [--take TAKE] --counts COUNTS [--mount MOUNT]
ERROR: the following arguments are required: --counts

Note how the mount "positional" argument is required and cannot be omitted:

$ ./parse_example --keep --take 3 --counts no         
usage: [-h] [--keep] [--take TAKE] --counts COUNTS mount
ERROR: the following arguments are required: mount

Implementation

The source code is available here and revolves around adding arguments to argparse.ArgumentParser dynamically:

for arg in args:
    kwargs = {}
    m = re.match(MODIFIED_PATTERN, arg)
    if m:
        # Take different action depending on the `modifier`            
        # then add to the parser.
        parser.add_argument(f"{prefix}{m.group('opt')}", **kwargs)

We have subclassed the ArgumentParser with a StderrParser so that:

  • when erroring out, we emit error messages to stderr so they don't get "swallowed" in the bash script; and
  • we need to exit with an error code, so that using set -e in our shell script will cause it to terminate, instead of executing the source command with potentially unexpected consequences.

Build/Test scripts

These are generic scripts, which rely on a common env.sh script to be sourced from the current directory:

export COMMON_UTILS=...
export PATH=$PATH:$COMMON_UTILS

build && runtests

They also expect a $BUILDDIR full path to point to the build directory, and the tests binaries to be in $BUILDDIR/tests/bin.

If your directory structure is something like this:

project
  |
  `-  env.sh
  |
  `- src/
  |
  `- build
  `- ... etc.

your env.sh could look something like (utils is sourced by the build script immediately before sourceing env.sh):

set -eu

BUILDDIR=$(abspath "./build")
CLANG=$(which clang++)

OS_NAME=$(uname -s)
msg "Build Platform: ${OS_NAME}"

... other configurations

See the libdist project for an example.

Contributions

Are warmly appreciated; please open an Issue to describe what you think is missing and you'd like to see added; or even feel free to contribute code via Pull Request.

This repository follows strictly the Google Shell Style Guide, please make sure you understand it, and follow it in your contribution.

About

A collection of utility scripts to simplify common tasks in shell scripts, templates for Makefile and testing/building scripts for C++ projects.

Resources

Stars

Watchers

Forks