/*
 * Copyright (C) 2015 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.afwsamples.testdpc;

import android.accounts.Account;
import android.accounts.AccountManager;
import android.app.Activity;
import android.content.pm.ApplicationInfo;
import android.content.pm.PackageManager;
import android.os.Bundle;
import android.util.Log;
import android.view.View;
import android.widget.Button;
import android.widget.ImageView;
import android.widget.TextView;

import com.afwsamples.testdpc.common.LaunchIntentUtil;
import com.afwsamples.testdpc.common.Util;
import com.afwsamples.testdpc.provision.ProvisioningUtil;
import com.android.setupwizardlib.GlifLayout;

public class FinalizeActivity extends Activity {
    private Button mFinishButton;
    private GlifLayout mSetupWizardLayout;

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

        if (savedInstanceState == null) {
            if (Util.isManagedProfileOwner(this)) {
                ProvisioningUtil.enableProfile(this);
            }
        }
        setContentView(R.layout.finalize_activity);
        mSetupWizardLayout = findViewById(R.id.setup_wizard_layout);
        mFinishButton = mSetupWizardLayout.findViewById(R.id.suw_navbar_next);
        mFinishButton.setText(R.string.finish_button);
        mFinishButton.setOnClickListener(this::onNavigateNext);


        // This is just a user friendly shortcut to the policy management screen of this app.
        ImageView appIcon = findViewById(R.id.app_icon);
        TextView appLabel = findViewById(R.id.app_label);
        try {
            PackageManager packageManager = getPackageManager();
            ApplicationInfo applicationInfo = packageManager.getApplicationInfo(
                    getPackageName(), 0 /* Default flags */);
            appIcon.setImageDrawable(packageManager.getApplicationIcon(applicationInfo));
            appLabel.setText(packageManager.getApplicationLabel(applicationInfo));
        } catch (PackageManager.NameNotFoundException e) {
            Log.w("TestDPC", "Couldn't look up our own package?!?!", e);
        }

        // Show the user which account now has management, if specified.
        String addedAccount = getIntent().getStringExtra(LaunchIntentUtil.EXTRA_ACCOUNT_NAME);
        if (addedAccount != null) {
            View accountMigrationStatusLayout;
            if (isAccountMigrated(addedAccount)) {
                accountMigrationStatusLayout = findViewById(R.id.account_migration_success);
            } else {
                accountMigrationStatusLayout = findViewById(R.id.account_migration_fail);
            }
            accountMigrationStatusLayout.setVisibility(View.VISIBLE);
            TextView managedAccountName = (TextView) accountMigrationStatusLayout.findViewById(
                    R.id.managed_account_name);
            managedAccountName.setText(addedAccount);
        }

        ((TextView) findViewById(R.id.explanation)).setText(Util.isDeviceOwner(this)
                ? R.string.all_done_explanation_device_owner
                : R.string.all_done_explanation_profile_owner);
    }

    private boolean isAccountMigrated(String addedAccount) {
        Account[] accounts = AccountManager.get(this).getAccounts();
        for (Account account : accounts) {
            if (addedAccount.equalsIgnoreCase(account.name)) {
                return true;
            }
        }
        return false;
    }

    public void onNavigateNext(View nextButton) {
        finish();
    }
}

@@ -4073,24 +4073,54 @@
 xtButton) %7B%0A
+        setResult(RESULT_OK);%0A
         fini

Return RESULT_OK from the policy compliance screen.