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

import android.content.Context;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.work.ListenableWorker;
import androidx.work.WorkerFactory;
import androidx.work.WorkerParameters;

import com.android.devicelockcontroller.util.LogUtil;

import com.google.common.util.concurrent.ListeningExecutorService;
import com.google.common.util.concurrent.MoreExecutors;

import java.lang.reflect.InvocationTargetException;
import java.util.concurrent.Executors;

/** A factory which produces {@link ListenableWorker}s with parameters. */
public final class DeviceLockControllerWorkerFactory extends WorkerFactory {
    private static final String TAG = "DeviceLockControllerWorkerFactory";

    private static final ListeningExecutorService sExecutorService =
            MoreExecutors.listeningDecorator(Executors.newCachedThreadPool());

    @Nullable
    @Override
    public ListenableWorker createWorker(
            @NonNull Context context,
            @NonNull String workerClassName,
            @NonNull WorkerParameters workerParameters) {
        ListenableWorker worker = null;
        Class<?> clazz = null;
        try {
            clazz = Class.forName(workerClassName);
        } catch (ClassNotFoundException e) {
            LogUtil.e(TAG, "Can not find class for name: " + workerClassName, e);
        }

        if (clazz != null) {
            try {
                worker = (ListenableWorker) clazz.getConstructor(
                                Context.class,
                                WorkerParameters.class,
                                ListeningExecutorService.class)
                        .newInstance(context, workerParameters, sExecutorService);
            } catch (InstantiationException | IllegalAccessException
                     | InvocationTargetException | NoSuchMethodException e) {
                // Unable to create the instance by this WorkerFactory
                LogUtil.i(TAG, "Delegating to default WorkerFactory to create: " + workerClassName,
                        e);
            }
        }
        return worker;
    }
}