#! /bin/bash

. /usr/lib/netctl/globals


usage() {
    cat << END
Usage: netctl {COMMAND} [PROFILE]
              [--help|--version]

Commands:
  list                   List available profiles
  store                  Save which profiles are active
  restore                Load saved profiles
  stop-all               Stops all profiles
  start [PROFILE]        Start a profile
  stop [PROFILE]         Stop a profile
  restart [PROFILE]      Restart a profile
  switch-to [PROFILE]    Switch to a profile
  is-active [PROFILE]    Check whether a profile is active
  status [PROFILE]       Show runtime status of a profile
  enable [PROFILE]       Enable the systemd unit for a profile
  disable [PROFILE]      Disable the systemd unit for a profile
  reenable [PROFILE]     Reenable the systemd unit for a profile
  is-enabled [PROFILE]   Check whether a profile is enabled
  edit [PROFILE]         Edit a profile
  wait-online [PROFILE]  Wait for a profile to finish connecting
END
}

list() {
    local indicators=( ' ' '+' '*' ) i
    list_profiles | while read -r Profile; do
        if sd_call "is-active --quiet" "$Profile" &> /dev/null; then
            [[ $(sd_status_text "$Profile") == "online" ]]
            (( i = 2 - $? ))
        else
            (( i = 0 ))
        fi
        printf '%s %s\n' "${indicators[i]}" "$Profile"
    done
}

store() {
    mkdir -p "$(dirname "$STATE_FILE")"
    list_profiles | while read -r Profile; do
        if sd_call "is-active --quiet" "$Profile" &> /dev/null; then
            printf "%s\n" "$Profile"
        fi
    done > "$STATE_FILE"
}

restore() {
    if [[ ! -r $STATE_FILE ]]; then
        exit_error "Could not read state file '$STATE_FILE'"
    elif [[ ! -s $STATE_FILE ]]; then
        report_debug "No profiles to restore in state file '$STATE_FILE'"
    else
        mapfile -t Profiles < "$STATE_FILE"
        do_debug sd_call start "${Profiles[@]}"
    fi
}

stop_all() {
    # We cannot pipe to mapfile, as the end of a pipe is inside a subshell
    mapfile -t Profiles < <(list_profiles)
    (( ${#Profiles[@]} )) && do_debug sd_call stop "${Profiles[@]}" 2> \
      >(grep -Fv "not loaded" >&2)
}

switch_to() {
    cd "$PROFILE_DIR"
    if [[ ! -r $1 ]]; then
        exit_error "Profile '$1' does not exist or is not readable"
    fi
    # We assume interface names are not quoted
    # Using read removes leading whitespace
    read InterfaceLine < \
      <(grep -om1 "^[[:space:]]*Interface=[[:alnum:]:._-]\+" "$1")
    if [[ -z $InterfaceLine ]]; then
        exit_error "Profile '$1' does not specify an interface"
    fi
    mapfile -t AllProfiles < <(list_profiles)
    mapfile -t Profiles < <(grep -Fl "$InterfaceLine" "${AllProfiles[@]}")
    (( ${#Profiles[@]} )) && do_debug sd_call stop "${Profiles[@]}" 2> \
      >(grep -Fv "not loaded" >&2)
    do_debug sd_call start "$1"
}

unit_enable() {
    local unit="/etc/systemd/system/netctl@$(systemd-escape "$1").service"
    if [[ -e $unit ]]; then
        report_error "A unit file for profile '$1' already exists"
        return 1
    fi
    load_profile "$1"
    do_readable touch "$unit"
    echo ".include /usr/lib/systemd/system/netctl@.service" > "$unit"
    echo -e "\n[Unit]" >> "$unit"
    [[ $Description ]] && echo "Description=$Description" >> "$unit"
    declare -p BindsToInterfaces &> /dev/null || BindsToInterfaces=$Interface
    if (( ${#BindsToInterfaces[@]} )); then
        : ${InterfaceRoot=sys/subsystem/net/devices/}
        printf "BindsTo=$(systemd-escape "$InterfaceRoot")%s.device\n" \
               $(systemd-escape "${BindsToInterfaces[@]}") >> "$unit"
        printf "After=$(systemd-escape "$InterfaceRoot")%s.device\n" \
               $(systemd-escape "${BindsToInterfaces[@]}") >> "$unit"
    fi
    if (( ${#After[@]} )); then
        printf 'After="netctl@%s.service"\n' \
               $(systemd-escape "${After[@]}") >> "$unit"
    fi
    mkdir -p "/etc/systemd/system/multi-user.target.wants"
    echo "ln -s '$unit' '${unit/system\//system/multi-user.target.wants/}'"
    ln -s "$unit" "${unit/system\//system/multi-user.target.wants/}"
}

unit_disable() {
    local unit="/etc/systemd/system/netctl@$(systemd-escape "$1").service"
    if sd_call "is-enabled --quiet" "$1" &> /dev/null; then
        sd_call disable "$1"
    fi
    if [[ ! -f $unit ]]; then
        report_error "No regular unit file found for profile '$1'"
        return 1
    fi
    do_debug rm "$unit"
}

wait_online() {
    local profile="$1"
    if sd_call "is-active --quiet" "$profile"; then
        timeout_wait "${TIMEOUT_ONLINE:-120}" \
                     '[[ $(sd_status_text "$profile") == "online" ]]'
    else
        return 1
    fi
}


case $# in
  1)
    case $1 in
      --version)
        report_notice "netctl version $NETCTL_VERSION";;
      --help)
        usage;;
      list)
        list;;
      store|restore)
        ensure_root "$(basename "$0")"
        "$1";;
      stop-all)
        stop_all;;
      *)
        exit_error "$(usage)";;
    esac;;
  2)
    case $1 in
      start|stop|restart|is-active|status|is-enabled)
        sd_call "$1" "$2";;
      switch-to)
        ensure_root "$(basename "$0")"
        switch_to "$2";;
      enable|disable)
        ensure_root "$(basename "$0")"
        "unit_$1" "$2"
        if systemd-notify --booted; then
            systemctl daemon-reload
        fi;;
      reenable)
        ensure_root "$(basename "$0")"
        unit_disable "$2"
        unit_enable "$2"
        if systemd-notify --booted; then
            systemctl daemon-reload
        fi;;
      edit)
        exec ${EDITOR:-nano} "$PROFILE_DIR/$2";;
      wait-online)
        wait_online "$2";;
      *)
        exit_error "$(usage)";;
    esac;;
  *)
    exit_error "$(usage)";;
esac


# vim: ft=sh ts=4 et sw=4:
