/*
 * 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.server.devicelock;

import android.annotation.NonNull;
import android.content.BroadcastReceiver;
import android.content.Context;
import android.content.Intent;
import android.content.IntentFilter;
import android.os.Process;
import android.os.UserHandle;
import android.os.UserManager;
import android.util.Slog;

import com.android.server.SystemService;

import java.util.Objects;

/**
 * Service implementing DeviceLock functionality. Delegates actual interface
 * implementation to DeviceLockServiceImpl.
 */
public final class DeviceLockService extends SystemService {

    private static final String TAG = "DeviceLockService";

    private final DeviceLockServiceImpl mImpl;

    public DeviceLockService(Context context) {
        super(context);
        Slog.d(TAG, "DeviceLockService constructor");
        mImpl = new DeviceLockServiceImpl(context);

        final IntentFilter userFilter = new IntentFilter();
        userFilter.addAction(Intent.ACTION_USER_ADDED);
        context.registerReceiver(mUserReceiver, userFilter);
    }

    @Override
    public void onStart() {
        Slog.i(TAG, "Registering " + Context.DEVICE_LOCK_SERVICE);
        publishBinderService(Context.DEVICE_LOCK_SERVICE, mImpl);
    }

    @Override
    public void onBootPhase(int phase) {
        Slog.d(TAG, "onBootPhase: " + phase);
    }

    @NonNull
    private static Context getUserContext(@NonNull Context context, @NonNull UserHandle user) {
        if (Process.myUserHandle().equals(user)) {
            return context;
        } else {
            return context.createContextAsUser(user, 0 /* flags */);
        }
    }

    @Override
    public boolean isUserSupported(@NonNull TargetUser user) {
        return isUserSupported(user.getUserHandle());
    }

    private boolean isUserSupported(UserHandle userHandle) {
        final UserManager userManager =
                getUserContext(getContext(), userHandle).getSystemService(UserManager.class);
        return !userManager.isProfile();
    }

    @Override
    public void onUserSwitching(@NonNull TargetUser from, @NonNull TargetUser to) {
        Objects.requireNonNull(to);
        Slog.d(TAG, "onUserSwitching from: " + from + " to: " + to);
        final UserHandle userHandle = to.getUserHandle();
        mImpl.onUserSwitching(userHandle);
    }

    @Override
    public void onUserUnlocking(@NonNull TargetUser user) {
        Slog.d(TAG, "onUserUnlocking: " + user);
    }

    @Override
    public void onUserUnlocked(@NonNull TargetUser user) {
        Slog.d(TAG, "onUserUnlocked: " + user);
        final UserHandle userHandle = user.getUserHandle();
        final Context userContext = getUserContext(getContext(), userHandle);
        mImpl.onUserUnlocked(userContext, userHandle);
    }

    @Override
    public void onUserStopping(@NonNull TargetUser user) {
        Slog.d(TAG, "onUserStopping: " + user);
    }

    private BroadcastReceiver mUserReceiver = new BroadcastReceiver() {
        @Override
        public void onReceive(Context context, Intent intent) {
            if (Intent.ACTION_USER_ADDED.equals(intent.getAction())) {
                final int userId = intent.getIntExtra(Intent.EXTRA_USER_HANDLE, -1);
                UserHandle userHandle = UserHandle.of(userId);
                if (!isUserSupported(userHandle)) {
                    return;
                }
                Slog.d(TAG, "onUserAdded: " + userHandle);
                mImpl.onUserAdded(userHandle);
            }
        }
    };
}