#!/bin/bash
# SPDX-License-Identifier: GPL-2.0
# Manipulate options in a .config file from the command line

myname=${0##*/}

# If no prefix forced, use the default CONFIG_
CONFIG_="${CONFIG_-CONFIG_}"

usage() {
        cat >&2 <<EOL
Manipulate options in a .config file from the command line.
Usage:
$myname options command ...
commands:
        --enable|-e option   Enable option
        --disable|-d option  Disable option
        --module|-m option   Turn option into a module
        --set-str option string
                             Set option to "string"
        --set-val option value
                             Set option to value
        --undefine|-u option Undefine option
        --state|-s option    Print state of option (n,y,m,undef)

        --enable-after|-E beforeopt option
                             Enable option directly after other option
        --disable-after|-D beforeopt option
                             Disable option directly after other option
        --module-after|-M beforeopt option
                             Turn option into module directly after other option

        commands can be repeated multiple times

options:
        --file config-file   .config file to change (default .config)
        --keep-case|-k       Keep next symbols' case (dont' upper-case it)

$myname doesn't check the validity of the .config file. This is done at next
make time.

By default, $myname will upper-case the given symbol. Use --keep-case to keep
the case of all following symbols unchanged.

$myname uses 'CONFIG_' as the default symbol prefix. Set the environment
variable CONFIG_ to the prefix to use. Eg.: CONFIG_="FOO_" $myname ...
EOL
        exit 1
}

checkarg() {
        ARG="$1"
        if [ "$ARG" = "" ] ; then
                usage
        fi
        case "$ARG" in
        ${CONFIG_}*)
                ARG="${ARG/${CONFIG_}/}"
                ;;
        esac
        if [ "$MUNGE_CASE" = "yes" ] ; then
                ARG="`echo $ARG | tr a-z A-Z`"
        fi
}

txt_append() {
        local anchor="$1"
        local insert="$2"
        local infile="$3"
        local tmpfile="$infile.swp"

        # sed append cmd: 'a\' + newline + text + newline
        cmd="$(printf "a\\%b$insert" "\n")"

        sed -e "/$anchor/$cmd" "$infile" >"$tmpfile"
        # replace original file with the edited one
        mv "$tmpfile" "$infile"
}

txt_subst() {
        local before="$1"
        local after="$2"
        local infile="$3"
        local tmpfile="$infile.swp"

        sed -e "s:$before:$after:" "$infile" >"$tmpfile"
        # replace original file with the edited one
        mv "$tmpfile" "$infile"
}

txt_delete() {
        local text="$1"
        local infile="$2"
        local tmpfile="$infile.swp"

        sed -e "/$text/d" "$infile" >"$tmpfile"
        # replace original file with the edited one
        mv "$tmpfile" "$infile"
}

set_var() {
        local name=$1 new=$2 before=$3

        name_re="^($name=|# $name is not set)"
        before_re="^($before=|# $before is not set)"
        if test -n "$before" && grep -Eq "$before_re" "$FN"; then
                txt_append "^$before=" "$new" "$FN"
                txt_append "^# $before is not set" "$new" "$FN"
        elif grep -Eq "$name_re" "$FN"; then
                txt_subst "^$name=.*" "$new" "$FN"
                txt_subst "^# $name is not set" "$new" "$FN"
        else
                echo "$new" >>"$FN"
        fi
}

undef_var() {
        local name=$1

        txt_delete "^$name=" "$FN"
        txt_delete "^# $name is not set" "$FN"
}

if [ "$1" = "--file" ]; then
        FN="$2"
        if [ "$FN" = "" ] ; then
                usage
        fi
        shift 2
else
        FN=.config
fi

if [ "$1" = "" ] ; then
        usage
fi

MUNGE_CASE=yes
while [ "$1" != "" ] ; do
        CMD="$1"
        shift
        case "$CMD" in
        --keep-case|-k)
                MUNGE_CASE=no
                continue
                ;;
        --refresh)
                ;;
        --*-after|-E|-D|-M)
                checkarg "$1"
                A=$ARG
                checkarg "$2"
                B=$ARG
                shift 2
                ;;
        -*)
                checkarg "$1"
                shift
                ;;
        esac
        case "$CMD" in
        --enable|-e)
                set_var "${CONFIG_}$ARG" "${CONFIG_}$ARG=y"
                ;;

        --disable|-d)
                set_var "${CONFIG_}$ARG" "# ${CONFIG_}$ARG is not set"
                ;;

        --module|-m)
                set_var "${CONFIG_}$ARG" "${CONFIG_}$ARG=m"
                ;;

        --set-str)
                # sed swallows one level of escaping, so we need double-escaping
                set_var "${CONFIG_}$ARG" "${CONFIG_}$ARG=\"${1//\"/\\\\\"}\""
                shift
                ;;

        --set-val)
                set_var "${CONFIG_}$ARG" "${CONFIG_}$ARG=$1"
                shift
                ;;
        --undefine|-u)
                undef_var "${CONFIG_}$ARG"
                ;;

        --state|-s)
                if grep -q "# ${CONFIG_}$ARG is not set" $FN ; then
                        echo n
                else
                        V="$(grep "^${CONFIG_}$ARG=" $FN)"
                        if [ $? != 0 ] ; then
                                echo undef
                        else
                                V="${V/#${CONFIG_}$ARG=/}"
                                V="${V/#\"/}"
                                V="${V/%\"/}"
                                V="${V//\\\"/\"}"
                                echo "${V}"
                        fi
                fi
                ;;

        --enable-after|-E)
                set_var "${CONFIG_}$B" "${CONFIG_}$B=y" "${CONFIG_}$A"
                ;;

        --disable-after|-D)
                set_var "${CONFIG_}$B" "# ${CONFIG_}$B is not set" "${CONFIG_}$A"
                ;;

        --module-after|-M)
                set_var "${CONFIG_}$B" "${CONFIG_}$B=m" "${CONFIG_}$A"
                ;;

        # undocumented because it ignores --file (fixme)
        --refresh)
                yes "" | make oldconfig
                ;;

        *)
                usage
                ;;
        esac
done