java/src/com/android/textclassifier/logging/SelectionEventConverter.java

/*
 * Copyright (C) 2019 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.textclassifier.logging;

import android.view.textclassifier.SelectionEvent;
import android.view.textclassifier.TextClassificationContext;
import android.view.textclassifier.TextClassifierEvent;

import androidx.annotation.Nullable;

/** Helper class to convert a {@link SelectionEvent} to a {@link TextClassifierEvent}. */
public final class SelectionEventConverter {

    /** Converts a {@link SelectionEvent} to a {@link TextClassifierEvent}. */
    @Nullable
    public static TextClassifierEvent toTextClassifierEvent(SelectionEvent selectionEvent) {
        TextClassificationContext textClassificationContext = null;
        if (selectionEvent.getPackageName() != null && selectionEvent.getWidgetType() != null) {
            textClassificationContext =
                    new TextClassificationContext.Builder(
                                    selectionEvent.getPackageName(), selectionEvent.getWidgetType())
                            .setWidgetVersion(selectionEvent.getWidgetVersion())
                            .build();
        }
        if (selectionEvent.getInvocationMethod() == SelectionEvent.INVOCATION_LINK) {
            return new TextClassifierEvent.TextLinkifyEvent.Builder(
                            convertEventType(selectionEvent.getEventType()))
                    .setEventContext(textClassificationContext)
                    .setResultId(selectionEvent.getResultId())
                    .setEventIndex(selectionEvent.getEventIndex())
                    .setEntityTypes(selectionEvent.getEntityType())
                    .build();
        }
        if (selectionEvent.getInvocationMethod() == SelectionEvent.INVOCATION_MANUAL) {
            return new TextClassifierEvent.TextSelectionEvent.Builder(
                            convertEventType(selectionEvent.getEventType()))
                    .setEventContext(textClassificationContext)
                    .setResultId(selectionEvent.getResultId())
                    .setEventIndex(selectionEvent.getEventIndex())
                    .setEntityTypes(selectionEvent.getEntityType())
                    .setRelativeWordStartIndex(selectionEvent.getStart())
                    .setRelativeWordEndIndex(selectionEvent.getEnd())
                    .setRelativeSuggestedWordStartIndex(selectionEvent.getSmartStart())
                    .setRelativeSuggestedWordEndIndex(selectionEvent.getSmartEnd())
                    .build();
        }
        return null;
    }

    private static int convertEventType(int eventType) {
        switch (eventType) {
            case SelectionEvent.EVENT_SELECTION_STARTED:
                return TextClassifierEvent.TYPE_SELECTION_STARTED;
            case SelectionEvent.EVENT_SELECTION_MODIFIED:
                return TextClassifierEvent.TYPE_SELECTION_MODIFIED;
            case SelectionEvent.EVENT_SMART_SELECTION_SINGLE:
                return SelectionEvent.EVENT_SMART_SELECTION_SINGLE;
            case SelectionEvent.EVENT_SMART_SELECTION_MULTI:
                return SelectionEvent.EVENT_SMART_SELECTION_MULTI;
            case SelectionEvent.EVENT_AUTO_SELECTION:
                return SelectionEvent.EVENT_AUTO_SELECTION;
            case SelectionEvent.ACTION_OVERTYPE:
                return TextClassifierEvent.TYPE_OVERTYPE;
            case SelectionEvent.ACTION_COPY:
                return TextClassifierEvent.TYPE_COPY_ACTION;
            case SelectionEvent.ACTION_PASTE:
                return TextClassifierEvent.TYPE_PASTE_ACTION;
            case SelectionEvent.ACTION_CUT:
                return TextClassifierEvent.TYPE_CUT_ACTION;
            case SelectionEvent.ACTION_SHARE:
                return TextClassifierEvent.TYPE_SHARE_ACTION;
            case SelectionEvent.ACTION_SMART_SHARE:
                return TextClassifierEvent.TYPE_SMART_ACTION;
            case SelectionEvent.ACTION_DRAG:
                return TextClassifierEvent.TYPE_SELECTION_DRAG;
            case SelectionEvent.ACTION_ABANDON:
                return TextClassifierEvent.TYPE_SELECTION_DESTROYED;
            case SelectionEvent.ACTION_OTHER:
                return TextClassifierEvent.TYPE_OTHER_ACTION;
            case SelectionEvent.ACTION_SELECT_ALL:
                return TextClassifierEvent.TYPE_SELECT_ALL;
            case SelectionEvent.ACTION_RESET:
                return TextClassifierEvent.TYPE_SELECTION_RESET;
            default:
                return 0;
        }
    }
}