/*
 * Copyright (C) 2017 The Android Open Source Project
 *
 * Licensed 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.
 */

package com.android.server.wifi.aware;

import android.os.Binder;
import android.text.TextUtils;
import android.util.Log;

import com.android.modules.utils.BasicShellCommandHandler;

import java.io.PrintWriter;
import java.util.HashMap;
import java.util.Map;

/**
 * Interprets and executes 'adb shell cmd wifiaware [args]'.
 */
public class WifiAwareShellCommand extends BasicShellCommandHandler {
    private static final String TAG = "WifiAwareShellCommand";

    private Map<String, DelegatedShellCommand> mDelegatedCommands = new HashMap<>();

    /**
     * Register an delegated command interpreter for the specified 'command'. Each class can
     * interpret and execute their own commands.
     */
    public void register(String command, DelegatedShellCommand shellCommand) {
        if (mDelegatedCommands.containsKey(command)) {
            Log.e(TAG, "register: overwriting existing command -- '" + command + "'");
        }

        mDelegatedCommands.put(command, shellCommand);
    }

    @Override
    public int onCommand(String cmd) {
        checkRootPermission();

        final PrintWriter pw = getErrPrintWriter();
        try {
            if ("reset".equals(cmd)) {
                for (DelegatedShellCommand dsc: mDelegatedCommands.values()) {
                    dsc.onReset();
                }
                return 0;
            } else {
                DelegatedShellCommand delegatedCmd = null;
                if (!TextUtils.isEmpty(cmd)) {
                    delegatedCmd = mDelegatedCommands.get(cmd);
                }

                if (delegatedCmd != null) {
                    return delegatedCmd.onCommand(this);
                } else {
                    return handleDefaultCommands(cmd);
                }
            }
        } catch (Exception e) {
            pw.println("Exception: " + e);
        }
        return -1;
    }

    private void checkRootPermission() {
        final int uid = Binder.getCallingUid();
        if (uid == 0) {
            // Root can do anything.
            return;
        }
        throw new SecurityException("Uid " + uid + " does not have access to wifiaware commands");
    }

    @Override
    public void onHelp() {
        final PrintWriter pw = getOutPrintWriter();

        pw.println("Wi-Fi Aware (wifiaware) commands:");
        pw.println("  help");
        pw.println("    Print this help text.");
        pw.println("  reset");
        pw.println("    Reset parameters to default values.");
        for (Map.Entry<String, DelegatedShellCommand> sce: mDelegatedCommands.entrySet()) {
            sce.getValue().onHelp(sce.getKey(), this);
        }
        pw.println();
    }

    /**
     * Interface that delegated command targets must implement. They are passed the parent shell
     * command (the real command interpreter) from which they can obtain arguments.
     */
    public interface DelegatedShellCommand {
        /**
         * Execute the specified command. Use the parent shell to obtain arguments. Note that the
         * first argument (which specified the delegated shell) has already been extracted.
         */
        int onCommand(BasicShellCommandHandler parentShell);

        /**
         * Reset all parameters to their default values.
         */
        void onReset();

        /**
         * Print out help for the delegated command. The name of the delegated command is passed
         * as a first argument as an assist (prevents hard-coding of that string in multiple
         * places).
         */
        void onHelp(String command, BasicShellCommandHandler parentShell);

    }
}