/*
 * Copyright (C) 2018 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.slice.cts;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.timeout;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import android.app.Instrumentation;
import android.app.PendingIntent;
import android.app.slice.Slice;
import android.app.slice.SliceItem;
import android.app.slice.SliceManager;
import android.content.Context;
import android.content.Intent;
import android.net.Uri;
import android.os.ParcelFileDescriptor;
import android.support.test.InstrumentationRegistry;
import android.support.test.runner.AndroidJUnit4;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.stubbing.Answer;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

@RunWith(AndroidJUnit4.class)
public class SliceManagerTest {

    private static final Uri BASE_URI = Uri.parse("content://android.slice.cts.local/main");
    private final Context mContext = InstrumentationRegistry.getContext();
    private final SliceManager mSliceManager = mContext.getSystemService(SliceManager.class);

    @Before
    public void setup() {
        LocalSliceProvider.sProxy = mock(SliceProvider.class);
        try {
            mSliceManager.unpinSlice(BASE_URI);
        } catch (Exception e) {
        }
    }

    @After
    public void teardown() throws Exception {
        try {
            mSliceManager.unpinSlice(BASE_URI);
        } catch (Exception e) {
        }
    }

    @Test
    public void testPinSlice() throws Exception {
        mSliceManager.pinSlice(BASE_URI, Collections.emptySet());

        verify(LocalSliceProvider.sProxy, timeout(2000)).onSlicePinned(eq(BASE_URI));
    }

    @Test
    public void testUnpinSlice() throws Exception {
        mSliceManager.pinSlice(BASE_URI, Collections.emptySet());

        verify(LocalSliceProvider.sProxy, timeout(2000)).onSlicePinned(eq(BASE_URI));

        mSliceManager.unpinSlice(BASE_URI);

        verify(LocalSliceProvider.sProxy, timeout(2000)).onSliceUnpinned(eq(BASE_URI));
    }

    @Test
    public void testPinList() {
        Uri uri = BASE_URI;
        Uri longerUri = uri.buildUpon().appendPath("something").build();
        try {
            mSliceManager.pinSlice(uri, Collections.emptySet());
            mSliceManager.pinSlice(longerUri, Collections.emptySet());
            verify(LocalSliceProvider.sProxy, timeout(2000)).onSlicePinned(eq(longerUri));

            List<Uri> uris = mSliceManager.getPinnedSlices();
            assertEquals(2, uris.size());
            assertTrue(uris.contains(uri));
            assertTrue(uris.contains(longerUri));
        } finally {
            mSliceManager.unpinSlice(uri);
            mSliceManager.unpinSlice(longerUri);
        }
    }

    @Test
    public void testMapIntentToUri() {
        Intent intent = new Intent("android.slice.cts.action.TEST_ACTION");
        intent.setPackage("android.slice.cts");
        intent.putExtra("path", "intent");

        when(LocalSliceProvider.sProxy.onMapIntentToUri(any())).then(
                (Answer<Uri>) invocation -> BASE_URI.buildUpon().path(
                        ((Intent) invocation.getArguments()[0]).getStringExtra("path")).build());

        Uri uri = mSliceManager.mapIntentToUri(intent);

        assertEquals(BASE_URI.buildUpon().path("intent").build(), uri);
        verify(LocalSliceProvider.sProxy).onMapIntentToUri(eq(intent));
    }

    @Test
    public void testOnCreatePermissionSlice() {
        LocalSliceProvider.sAnswer = invocation -> {
            throw new SecurityException("No slices allowed");
        };
        try {
            Uri uri = BASE_URI.buildUpon().path("permission").build();
            PendingIntent intent = PendingIntent.getBroadcast(mContext, 0, new Intent(""), 0);

            when(LocalSliceProvider.sProxy.onCreatePermissionRequest(any())).thenReturn(intent);

            Slice s = mSliceManager.bindSlice(uri, Collections.emptySet());

            /// Make sure we get a callback for creating the intent.
            verify(LocalSliceProvider.sProxy).onCreatePermissionRequest(eq(uri));
            // Verify the intent we get back is the one we returned.
            PendingIntent i = findAction(s);
            assertTrue(intent == i);
        } finally {
            LocalSliceProvider.sAnswer = null;
        }
    }

    private PendingIntent findAction(Slice s) {
        for (SliceItem item : s.getItems()) {
            if (SliceItem.FORMAT_SLICE.equals(item.getFormat())) {
                return findAction(item.getSlice());
            }
            if (SliceItem.FORMAT_ACTION.equals(item.getFormat())) {
                return item.getAction();
            }
        }
        return null;
    }

    public static String getDefaultLauncher() throws Exception {
        final String PREFIX = "Launcher: ComponentInfo{";
        final String POSTFIX = "}";
        for (String s : runShellCommand("cmd shortcut get-default-launcher")) {
            if (s.startsWith(PREFIX) && s.endsWith(POSTFIX)) {
                return s.substring(PREFIX.length(), s.length() - POSTFIX.length());
            }
        }
        throw new Exception("Default launcher not found");
    }

    public static ArrayList<String> runShellCommand(String command) throws Exception {
        ParcelFileDescriptor pfd = getInstrumentation().getUiAutomation()
                .executeShellCommand(command);

        ArrayList<String> ret = new ArrayList<>();
        // Read the input stream fully.
        try (BufferedReader r = new BufferedReader(
                new InputStreamReader(new ParcelFileDescriptor.AutoCloseInputStream(pfd)))) {
            String line;
            while ((line = r.readLine()) != null) {
                ret.add(line);
            }
        }
        return ret;
    }

    public static Instrumentation getInstrumentation() {
        return InstrumentationRegistry.getInstrumentation();
    }

}