/*
 * Copyright (C) 2024 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.ondevicepersonalization.services.serviceflow;

import android.adservices.ondevicepersonalization.DownloadCompletedOutputParcel;
import android.adservices.ondevicepersonalization.EventOutputParcel;
import android.adservices.ondevicepersonalization.ExecuteOptionsParcel;
import android.adservices.ondevicepersonalization.RequestLogRecord;
import android.adservices.ondevicepersonalization.aidl.IExecuteCallback;
import android.adservices.ondevicepersonalization.aidl.IRegisterMeasurementEventCallback;
import android.adservices.ondevicepersonalization.aidl.IRequestSurfacePackageCallback;
import android.content.ComponentName;
import android.content.Context;
import android.os.Bundle;
import android.os.IBinder;

import com.android.internal.annotations.VisibleForTesting;
import com.android.ondevicepersonalization.services.data.events.EventUrlPayload;
import com.android.ondevicepersonalization.services.display.WebViewFlow;
import com.android.ondevicepersonalization.services.download.DownloadFlow;
import com.android.ondevicepersonalization.services.request.AppRequestFlow;
import com.android.ondevicepersonalization.services.request.RenderFlow;
import com.android.ondevicepersonalization.services.webtrigger.WebTriggerFlow;

import com.google.common.util.concurrent.FutureCallback;

/** Factory for service flow instances. */
public class ServiceFlowFactory {

    /** Create a service flow instance give the type. */
    public static ServiceFlow createInstance(ServiceFlowType serviceFlowType, Object... args) {
        return switch (serviceFlowType) {
            case APP_REQUEST_FLOW ->
                    new AppRequestFlow(
                            (String) args[0],
                            (ComponentName) args[1],
                            (Bundle) args[2],
                            (IExecuteCallback) args[3],
                            (Context) args[4],
                            (long) args[5],
                            (long) args[6],
                            (ExecuteOptionsParcel) args[7]);
            case RENDER_FLOW ->
                    new RenderFlow(
                            (String) args[0],
                            (IBinder) args[1],
                            (int) args[2],
                            (int) args[3],
                            (int) args[4],
                            (IRequestSurfacePackageCallback) args[5],
                            (Context) args[6],
                            (long) args[7],
                            (long) args[8]);
            case WEB_TRIGGER_FLOW ->
                    new WebTriggerFlow(
                            (Bundle) args[0],
                            (Context) args[1],
                            (IRegisterMeasurementEventCallback) args[2],
                            (long) args[3],
                            (long) args[4]);
            case WEB_VIEW_FLOW ->
                    new WebViewFlow(
                            (Context) args[0],
                            (ComponentName) args[1],
                            (long) args[2],
                            (RequestLogRecord) args[3],
                            (FutureCallback<EventOutputParcel>) args[4],
                            (EventUrlPayload) args[5]);
            case DOWNLOAD_FLOW ->
                    new DownloadFlow(
                            (String) args[0],
                            (Context) args[1],
                            (FutureCallback<DownloadCompletedOutputParcel>) args[2]);
            default ->
                    throw new IllegalArgumentException(
                            "Invalid service flow type: " + serviceFlowType);
        };
    }

    /** Create a service flow instance give the type for testing only. */
    @VisibleForTesting
    public static ServiceFlow createInstanceForTest(
            ServiceFlowType serviceFlowType, Object... args) {
        // TODO(b/354265327): only support injector in app request flow. Need update constructor for
        // testing in other flows.
        return switch (serviceFlowType) {
            case APP_REQUEST_FLOW ->
                    new AppRequestFlow(
                            (String) args[0],
                            (ComponentName) args[1],
                            (Bundle) args[2],
                            (IExecuteCallback) args[3],
                            (Context) args[4],
                            (long) args[5],
                            (long) args[6],
                            (ExecuteOptionsParcel) args[7],
                            (AppRequestFlow.Injector) args[8]);
            case RENDER_FLOW ->
                    new RenderFlow(
                            (String) args[0],
                            (IBinder) args[1],
                            (int) args[2],
                            (int) args[3],
                            (int) args[4],
                            (IRequestSurfacePackageCallback) args[5],
                            (Context) args[6],
                            (long) args[7],
                            (long) args[8]);
            case WEB_TRIGGER_FLOW ->
                    new WebTriggerFlow(
                            (Bundle) args[0],
                            (Context) args[1],
                            (IRegisterMeasurementEventCallback) args[2],
                            (long) args[3],
                            (long) args[4]);
            case WEB_VIEW_FLOW ->
                    new WebViewFlow(
                            (Context) args[0],
                            (ComponentName) args[1],
                            (long) args[2],
                            (RequestLogRecord) args[3],
                            (FutureCallback<EventOutputParcel>) args[4],
                            (EventUrlPayload) args[5]);
            case DOWNLOAD_FLOW ->
                    new DownloadFlow(
                            (String) args[0],
                            (Context) args[1],
                            (FutureCallback<DownloadCompletedOutputParcel>) args[2]);
            default ->
                    throw new IllegalArgumentException(
                            "Invalid service flow type: " + serviceFlowType);
        };
    }
}