/*
 * Copyright (C) 2025 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.virtualdevicemanager;

import android.annotation.StringRes;
import android.app.Activity;
import android.app.AppOpsManager;
import android.app.Dialog;
import android.content.DialogInterface;
import android.content.pm.ApplicationInfo;
import android.content.pm.PackageManager;
import android.graphics.drawable.Drawable;
import android.os.Bundle;
import android.os.ResultReceiver;
import android.text.Html;
import android.text.Spanned;
import android.view.View;
import android.widget.ImageView;
import android.widget.TextView;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.fragment.app.DialogFragment;

/**
 * A VDM consent dialog for starting a computer control session.
 */
public class RequestComputerControlAccessFragment extends DialogFragment {

    private static final String ARG_AGENT_PACKAGE_NAME = "argAgentPackageName";
    private static final String ARG_RESULT_RECEIVER = "argResultReceiver";

    private String mAgentPackageName;
    private ResultReceiver mResultReceiver;

    private CharSequence mAgentAppLabel;
    private Drawable mAgentAppIcon;
    private int mAgentUid;

    static RequestComputerControlAccessFragment newInstance(
            @NonNull String agentPackageName, @NonNull ResultReceiver resultReceiver) {
        final var fragment = new RequestComputerControlAccessFragment();
        final Bundle args = new Bundle();
        args.putString(ARG_AGENT_PACKAGE_NAME, agentPackageName);
        args.putParcelable(ARG_RESULT_RECEIVER, resultReceiver);
        fragment.setArguments(args);
        return fragment;
    }

    @Override
    public void onCreate(@Nullable Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);

        mAgentPackageName = getArguments().getString(ARG_AGENT_PACKAGE_NAME);
        mResultReceiver = getArguments().getParcelable(ARG_RESULT_RECEIVER, ResultReceiver.class);

        PackageManager packageManager = requireActivity().getPackageManager();
        try {
            final ApplicationInfo appInfo = packageManager.getApplicationInfo(
                    mAgentPackageName, PackageManager.ApplicationInfoFlags.of(0));
            mAgentAppLabel = packageManager.getApplicationLabel(appInfo);
            mAgentAppIcon = packageManager.getApplicationIcon(appInfo);
            mAgentUid = appInfo.uid;
        } catch (PackageManager.NameNotFoundException e) {
            finish(Activity.RESULT_CANCELED);
        }
    }

    @Override
    @NonNull
    public Dialog onCreateDialog(@Nullable Bundle savedInstanceState) {
        Activity activity = requireActivity();
        View view = activity.getLayoutInflater()
                .inflate(R.layout.request_computer_control_access_dialog, null);

        view.findViewById(R.id.allow_button).setOnClickListener(this::onAllow);
        view.findViewById(R.id.dont_allow_button).setOnClickListener(v -> getDialog().cancel());
        view.findViewById(R.id.always_allow_button).setOnClickListener(this::onAlwaysAllow);

        TextView titleView = view.findViewById(R.id.title);
        Spanned title = getHtmlFromResources(
                R.string.request_computer_control_access_dialog_title, mAgentAppLabel);
        titleView.setText(title);
        TextView descriptionView = view.findViewById(R.id.description);
        Spanned description = getHtmlFromResources(
                R.string.request_computer_control_access_dialog_description, mAgentAppLabel);
        descriptionView.setText(description);

        ImageView iconView = view.findViewById(R.id.agent_icon);
        iconView.setImageDrawable(mAgentAppIcon);

        Dialog dialog = new Dialog(activity);
        dialog.setContentView(view);
        return dialog;
    }

    private void finish(int resultCode) {
        mResultReceiver.send(resultCode, null);
        requireActivity().finish();
    }

    @Override
    public void onCancel(@NonNull DialogInterface dialog) {
        super.onCancel(dialog);
        finish(Activity.RESULT_CANCELED);
    }

    private void onAllow(View view) {
        finish(Activity.RESULT_OK);
    }

    private void onAlwaysAllow(View view) {
        AppOpsManager appOpsManager = requireActivity().getSystemService(AppOpsManager.class);
        appOpsManager.setMode(AppOpsManager.OP_COMPUTER_CONTROL, mAgentUid, mAgentPackageName,
                AppOpsManager.MODE_ALLOWED);
        finish(Activity.RESULT_OK);
    }

    private Spanned getHtmlFromResources(@StringRes int resId, CharSequence... formatArgs) {
        final String[] escapedArgs = new String[formatArgs.length];
        for (int i = 0; i < escapedArgs.length; i++) {
            escapedArgs[i] = formatArgs[i] == null ? null : Html.escapeHtml(formatArgs[i]);
        }
        final String plain = requireActivity().getString(resId, (Object[]) escapedArgs);
        return Html.fromHtml(plain, 0);
    }
}