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

import static android.companion.datatransfer.SystemDataTransferRequest.DATA_TYPE_PERMISSION_SYNC;
import static android.view.WindowManager.LayoutParams.SYSTEM_FLAG_HIDE_NON_SYSTEM_OVERLAY_WINDOWS;

import static com.android.companiondevicemanager.Utils.getHtmlFromResources;

import static java.util.Objects.requireNonNull;

import android.annotation.SuppressLint;
import android.app.Activity;
import android.companion.datatransfer.PermissionSyncRequest;
import android.companion.datatransfer.SystemDataTransferRequest;
import android.content.Intent;
import android.os.Build;
import android.os.Bundle;
import android.os.ResultReceiver;
import android.util.Log;
import android.widget.Button;
import android.widget.TextView;

/**
 * This activity manages the UI of companion device data transfer.
 */
@SuppressLint("LongLogTag")
public class CompanionDeviceDataTransferActivity extends Activity {

    private static final String LOG_TAG = "CDM_CompanionDeviceDataTransferActivity";

    // Intent data keys from SystemDataTransferProcessor
    private static final String EXTRA_PERMISSION_SYNC_REQUEST = "permission_sync_request";
    private static final String EXTRA_COMPANION_DEVICE_NAME = "companion_device_name";
    private static final String EXTRA_SYSTEM_DATA_TRANSFER_RESULT_RECEIVER =
            "system_data_transfer_result_receiver";

    // Intent data keys to SystemDataTransferProcessor
    private static final int RESULT_CODE_SYSTEM_DATA_TRANSFER_ALLOWED = 0;
    private static final int RESULT_CODE_SYSTEM_DATA_TRANSFER_DISALLOWED = 1;

    private SystemDataTransferRequest mRequest;
    private CharSequence mCompanionDeviceName;
    private ResultReceiver mCdmServiceReceiver;

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

        Log.i(LOG_TAG, "Creating UI for data transfer confirmation.");

        getWindow().addSystemFlags(SYSTEM_FLAG_HIDE_NON_SYSTEM_OVERLAY_WINDOWS);

        setContentView(R.layout.data_transfer_confirmation);

        TextView titleView = findViewById(R.id.title);
        TextView summaryView = findViewById(R.id.summary);
        Button allowButton = findViewById(R.id.btn_positive);
        Button disallowButton = findViewById(R.id.btn_negative);

        final Intent intent = getIntent();
        mRequest = intent.getParcelableExtra(EXTRA_PERMISSION_SYNC_REQUEST,
                PermissionSyncRequest.class);
        mCompanionDeviceName = intent.getCharSequenceExtra(EXTRA_COMPANION_DEVICE_NAME);
        mCdmServiceReceiver = intent.getParcelableExtra(EXTRA_SYSTEM_DATA_TRANSFER_RESULT_RECEIVER,
                ResultReceiver.class);

        requireNonNull(mRequest);
        requireNonNull(mCdmServiceReceiver);

        final String primaryDeviceName = Build.MODEL;

        if (mRequest.getDataType() == DATA_TYPE_PERMISSION_SYNC) {
            titleView.setText(getHtmlFromResources(this,
                    R.string.permission_sync_confirmation_title, mCompanionDeviceName,
                    primaryDeviceName));
            summaryView.setText(getHtmlFromResources(this, R.string.permission_sync_summary,
                    mCompanionDeviceName));
            allowButton.setOnClickListener(v -> allow());
            disallowButton.setOnClickListener(v -> disallow());
        }
    }

    private void allow() {
        Log.i(LOG_TAG, "allow()");

        sendDataToReceiver(RESULT_CODE_SYSTEM_DATA_TRANSFER_ALLOWED);

        setResultAndFinish(RESULT_CODE_SYSTEM_DATA_TRANSFER_ALLOWED);
    }

    private void disallow() {
        Log.i(LOG_TAG, "disallow()");

        sendDataToReceiver(RESULT_CODE_SYSTEM_DATA_TRANSFER_DISALLOWED);

        setResultAndFinish(RESULT_CODE_SYSTEM_DATA_TRANSFER_DISALLOWED);
    }

    private void sendDataToReceiver(int cdmResultCode) {
        Bundle data = new Bundle();
        if (mRequest instanceof PermissionSyncRequest) {
            data.putParcelable(EXTRA_PERMISSION_SYNC_REQUEST, (PermissionSyncRequest) mRequest);
        }
        mCdmServiceReceiver.send(cdmResultCode, data);
    }

    private void setResultAndFinish(int cdmResultCode) {
        setResult(cdmResultCode == RESULT_CODE_SYSTEM_DATA_TRANSFER_ALLOWED
                ? RESULT_OK : RESULT_CANCELED);
        finish();
    }
}