/*
 * 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 android.app.cts.android.app.cts.tools;

import android.app.ActivityManager;
import android.app.Instrumentation;
import android.content.Context;
import android.content.Intent;
import android.content.pm.ApplicationInfo;
import android.content.pm.PackageManager;
import android.os.IBinder;
import android.os.Parcel;
import android.os.RemoteException;

import androidx.test.InstrumentationRegistry;

import com.android.compatibility.common.util.SystemUtil;

import java.io.IOException;

/**
 * Helper for monitoring and controlling the state of a process under test.
 * Primarily currently a convenience for cleanly killing a process and waiting
 * for it to entirely disappear from the system.
 */
public final class ServiceProcessController {
    final Context mContext;
    final Instrumentation mInstrumentation;
    final String mMyPackageName;
    final Intent[] mServiceIntents;
    final String mServicePackage;
    final long mDefaultWaitTime;

    final ActivityManager mAm;
    final Parcel mData;
    final ServiceConnectionHandler[] mConnections;
    final int mUid;
    final UidImportanceListener mUidForegroundListener;
    final UidImportanceListener mUidGoneListener;
    final WatchUidRunner mUidWatcher;

    public ServiceProcessController(Context context, Instrumentation instrumentation,
            String myPackageName, Intent[] serviceIntents)
            throws IOException, PackageManager.NameNotFoundException {
        this(context, instrumentation, myPackageName, serviceIntents, 5*1000);
    }

    public ServiceProcessController(Context context, Instrumentation instrumentation,
            String myPackageName, Intent[] serviceIntents, long defaultWaitTime)
            throws IOException, PackageManager.NameNotFoundException {
        mContext = context;
        mInstrumentation = instrumentation;
        mMyPackageName = myPackageName;
        mServiceIntents = serviceIntents;
        mServicePackage = mServiceIntents[0].getComponent().getPackageName();
        mDefaultWaitTime = defaultWaitTime;

        InstrumentationRegistry.getInstrumentation().getUiAutomation().grantRuntimePermission(
                mMyPackageName, android.Manifest.permission.PACKAGE_USAGE_STATS);
        /*
        Log.d("XXXX", "Invoke: " + cmd);
        Log.d("XXXX", "Result: " + result);
        Log.d("XXXX", SystemUtil.runShellCommand(getInstrumentation(), "dumpsys package "
                + STUB_PACKAGE_NAME));
        */

        mAm = mContext.getSystemService(ActivityManager.class);
        mData = Parcel.obtain();
        mConnections = new ServiceConnectionHandler[serviceIntents.length];
        for (int i=0; i<serviceIntents.length; i++) {
            mConnections[i] = new ServiceConnectionHandler(mContext, serviceIntents[i],
                    mDefaultWaitTime);
        }

        ApplicationInfo appInfo = mContext.getPackageManager().getApplicationInfo(
                mServicePackage, 0);
        mUid = appInfo.uid;

        mUidForegroundListener = new UidImportanceListener(mContext, appInfo.uid,
                ActivityManager.RunningAppProcessInfo.IMPORTANCE_SERVICE, mDefaultWaitTime);
        mUidForegroundListener.register();
        mUidGoneListener = new UidImportanceListener(mContext, appInfo.uid,
                ActivityManager.RunningAppProcessInfo.IMPORTANCE_EMPTY, mDefaultWaitTime);
        mUidGoneListener.register();

        mUidWatcher = new WatchUidRunner(instrumentation, appInfo.uid, mDefaultWaitTime);
    }

    public void denyBackgroundOp() throws IOException {
        denyBackgroundOp(mDefaultWaitTime);
    }

    public void denyBackgroundOp(long timeout) throws IOException {
        String cmd = "appops set " + mServicePackage + " RUN_IN_BACKGROUND deny";
        String result = SystemUtil.runShellCommand(mInstrumentation, cmd);

        // This is a side-effect of the app op command.
        mUidWatcher.expect(WatchUidRunner.CMD_IDLE, null, timeout);
        mUidWatcher.expect(WatchUidRunner.CMD_PROCSTATE, "NONE", timeout);
    }

    public void allowBackgroundOp() throws IOException {
        String cmd = "appops set " + mServicePackage + " RUN_IN_BACKGROUND allow";
        String result = SystemUtil.runShellCommand(mInstrumentation, cmd);
    }

    public void makeUidIdle() throws IOException {
        String cmd = "am make-uid-idle " + mServicePackage;
        String result = SystemUtil.runShellCommand(mInstrumentation, cmd);
    }

    public void removeFromWhitelist() throws IOException {
        String cmd = "cmd deviceidle whitelist -" + mServicePackage;
        String result = SystemUtil.runShellCommand(mInstrumentation, cmd);
    }

    public void addToWhitelist() throws IOException {
        String cmd = "cmd deviceidle whitelist +" + mServicePackage;
        String result = SystemUtil.runShellCommand(mInstrumentation, cmd);
    }

    public void tempWhitelist(long duration) throws IOException {
        String cmd = "cmd deviceidle tempwhitelist -d " + duration + " " + mServicePackage;
        String result = SystemUtil.runShellCommand(mInstrumentation, cmd);
    }

    public void removeFromTempWhitelist() throws IOException {
        String cmd = "cmd deviceidle tempwhitelist -r " + mServicePackage;
        SystemUtil.runShellCommand(mInstrumentation, cmd);
    }

    public void setAppOpMode(String opStr, String mode) throws IOException {
        String cmd = "cmd appops set " + mServicePackage + " " + opStr + "  " + mode;
        SystemUtil.runShellCommand(mInstrumentation, cmd);
    }

    public void cleanup() throws IOException {
        removeFromWhitelist();
        allowBackgroundOp();
        mUidWatcher.finish();
        mUidGoneListener.unregister();
        mUidForegroundListener.unregister();
        mData.recycle();
    }

    public ServiceConnectionHandler getConnection(int index) {
        return mConnections[index];
    }

    public int getUid() {
        return mUid;
    }

    public UidImportanceListener getUidForegroundListener() {
        return mUidForegroundListener;
    }

    public UidImportanceListener getUidGoneListener() {
        return mUidGoneListener;
    }

    public WatchUidRunner getUidWatcher() {
        return mUidWatcher;
    }

    public void ensureProcessGone() {
        ensureProcessGone(mDefaultWaitTime);
    }

    public void ensureProcessGone(long timeout) {
        for (int i=0; i<mConnections.length; i++) {
            mConnections[i].bind(timeout);
        }

        for (int i=0; i<mConnections.length; i++) {
            IBinder serviceBinder = mConnections[i].getServiceIBinder();
            mConnections[i].unbind(timeout);
            try {
                serviceBinder.transact(IBinder.FIRST_CALL_TRANSACTION, mData, null, 0);
            } catch (RemoteException e) {
            }
        }

        // Wait for uid's process to go away.
        mUidGoneListener.waitForValue(ActivityManager.RunningAppProcessInfo.IMPORTANCE_GONE,
                ActivityManager.RunningAppProcessInfo.IMPORTANCE_GONE, timeout);
        int importance = mAm.getPackageImportance(mServicePackage);
        if (importance != ActivityManager.RunningAppProcessInfo.IMPORTANCE_GONE) {
            throw new IllegalStateException("Unexpected importance after killing process: "
                    + importance);
        }
        mUidWatcher.waitFor(WatchUidRunner.CMD_GONE, null, timeout);
    }
}