services/autofill/java/com/android/server/intelligence/ContentCaptureSession.java

/*
 * 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 com.android.server.intelligence;

import android.annotation.NonNull;
import android.content.ComponentName;
import android.content.Context;
import android.os.IBinder;
import android.service.intelligence.IntelligenceService;
import android.service.intelligence.InteractionContext;
import android.service.intelligence.InteractionSessionId;
import android.util.Slog;

import com.android.internal.annotations.GuardedBy;
import com.android.server.AbstractRemoteService;
import com.android.server.intelligence.RemoteIntelligenceService.RemoteIntelligenceServiceCallbacks;

import java.io.PrintWriter;

final class ContentCaptureSession implements RemoteIntelligenceServiceCallbacks {

    private static final String TAG = "ContentCaptureSession";

    private final Object mLock;
    private final IBinder mActivityToken;

    private final IntelligencePerUserService mService;
    private final RemoteIntelligenceService mRemoteService;
    private final InteractionContext mInterationContext;
    private final InteractionSessionId mId;

    ContentCaptureSession(@NonNull Context context, int userId, @NonNull Object lock,
            @NonNull IBinder activityToken, @NonNull IntelligencePerUserService service,
            @NonNull ComponentName serviceComponentName, @NonNull ComponentName appComponentName,
            int taskId, int displayId, int localSessionId, int globalSessionId, int flags,
            boolean bindInstantServiceAllowed, boolean verbose) {
        mLock = lock;
        mActivityToken = activityToken;
        mService = service;
        mRemoteService = new RemoteIntelligenceService(context,
                IntelligenceService.SERVICE_INTERFACE, serviceComponentName, userId, this,
                bindInstantServiceAllowed, verbose);
        mId = new InteractionSessionId(globalSessionId, localSessionId);
        mInterationContext = new InteractionContext(appComponentName, taskId, displayId, flags);
    }

    /**
     * Notifies the {@link IntelligenceService} that the service started.
     */
    @GuardedBy("mLock")
    public void notifySessionStartedLocked() {
        mRemoteService.onSessionLifecycleRequest(mInterationContext, mId);
    }

    /**
     * Cleans up the session and remove itself from the service.
     *
     * @param notifyRemoteService whether it should trigger a {@link
     * IntelligenceService#onDestroyInteractionSession(InteractionSessionId)}
     * request.
     */
    @GuardedBy("mLock")
    public void removeSelfLocked(boolean notifyRemoteService) {
        try {
            if (notifyRemoteService) {
                mRemoteService.onSessionLifecycleRequest(/* context= */ null, mId);
            }
        } finally {
            mService.removeSessionLocked(mInterationContext.getActivityComponent());
        }
    }

    @Override // from RemoteScreenObservationServiceCallbacks
    public void onServiceDied(AbstractRemoteService service) {
        // TODO(b/111276913): implement (remove session from PerUserSession?)
        if (mService.isDebug()) {
            Slog.d(TAG, "onServiceDied() for " + mId);
        }
        synchronized (mLock) {
            removeSelfLocked(/* notifyRemoteService= */ false);
        }
    }

    @Override // from RemoteScreenObservationServiceCallbacks
    public void onSessionLifecycleRequestFailureOrTimeout(boolean timedOut) {
        // TODO(b/111276913): log metrics on whether timed out or not
        if (mService.isDebug()) {
            Slog.d(TAG, "onSessionLifecycleRequestFailure(" + mId + "): timed out=" + timedOut);
        }
        synchronized (mLock) {
            removeSelfLocked(/* notifyRemoteService= */ false);
        }
    }

    /**
     * Gets global id, unique per {@link IntelligencePerUserService}.
     */
    public int getGlobalSessionId() {
        return mId.getGlobalId();
    }

    @GuardedBy("mLock")
    public void dumpLocked(@NonNull String prefix, @NonNull PrintWriter pw) {
        pw.print(prefix); pw.print("id: ");  mId.dump(pw); pw.println();
        pw.print(prefix); pw.print("context: ");  mInterationContext.dump(pw); pw.println();
        pw.print(prefix); pw.print("activity token: "); pw.println(mActivityToken);
    }
}