#!/usr/bin/env bash
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you 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.

# print usage
usage() {
    echo "
Description:
        This script is used to restore the tablets from trash. It supports single mode
    and batch mode.
        In single mode, it will restore just one tablet.
        In batch mode, it will restore all the tablets specified in file. The content 
    of the file is comma-split tablet id and schema hash, like the following:
            12345,11111
            12346,11111
            12347,11111

Usage: $0 <options>
    Optional options:
        -h | --help         print help info
        -b | --backend      backend http service, default: http://127.0.0.1/8040
        -t | --tablet_id    tablet id to restore
        -s | --schema_hash  tablet related schema hash
        -f | --file         file with lines containing comma-split tablet id and schema hash

Examples:
    batch mode:
        bash restore_tablet_tool.sh -b http://127.0.0.1:8040 -f tablets.txt
        bash restore_tablet_tool.sh --backend http://127.0.0.1:8040 --file tablets.txt

    single mode:
        bash restore_tablet_tool.sh -b http://127.0.0.1:8040 -t 12345 -s 11111
        bash restore_tablet_tool.sh --backend http://127.0.0.1:8040 --tablet_id 12345 --schema_hash 11111
    "
    exit 1
}

if ! OPTS=$(getopt \
    -n "$0" \
    -o 'b:t:s:f:' \
    -l 'server:,tablet_id:,schema_hash:,file:,help' \
    -- "$@"); then
    usage
fi

eval set -- "${OPTS}"

SERVER='http://127.0.0.1/8040'
TABLET_ID=''
SCHEMA_HASH=''
FILENAME=''
BATCH_MODE=false

while true; do
    case "$1" in
    -b | --backend)
        SERVER="$2"
        shift 2
        ;;
    -f | --file)
        FILENAME="$2"
        BATCH_MODE=true
        shift 2
        ;;
    -t | --tablet_id)
        TABLET_ID="$2"
        shift 2
        ;;
    -s | --schema_hash)
        SCHEMA_HASH="$2"
        shift 2
        ;;
    -h | --help)
        usage
        shift
        ;;
    --)
        shift
        break
        ;;
    *)
        echo "Internal error!"
        exit 1
        ;;
    esac
done

restore_tablet() {
    echo "start to restore tablet id:$2, schema hash:$3"
    curl -X POST "$1/api/restore_tablet?tablet_id=$2&schema_hash=$3"
    echo -e "\n"
}

if ${BATCH_MODE}; then
    while read -r line; do
        # split the comma-split line
        # format: tablet_id,schema_hash
        read -r -a fields <<<"${line/,/ }"
        TABLET_ID="${fields[0]}"
        SCHEMA_HASH="${fields[1]}"
        restore_tablet "${SERVER}" "${TABLET_ID}" "${SCHEMA_HASH}"
    done <"${FILENAME}"
else
    restore_tablet "${SERVER}" "${TABLET_ID}" "${SCHEMA_HASH}"
fi