src/com/android/systemui/car/statusicon/ui/LocationStatusIconController.java

/*
 * Copyright (C) 2021 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.systemui.car.statusicon.ui;

import android.app.ActivityManager;
import android.car.Car;
import android.car.user.CarUserManager;
import android.content.BroadcastReceiver;
import android.content.Context;
import android.content.Intent;
import android.content.IntentFilter;
import android.content.res.Resources;
import android.location.LocationManager;
import android.os.UserHandle;
import android.util.Log;

import com.android.systemui.R;
import com.android.systemui.car.CarServiceProvider;
import com.android.systemui.car.statusicon.StatusIconController;
import com.android.systemui.dagger.qualifiers.Main;

import javax.inject.Inject;

/**
 * A controller for the read-only icon that shows location active status.
 */
public class LocationStatusIconController extends StatusIconController {

    private static final String TAG = LocationStatusIconController.class.getSimpleName();
    private static final IntentFilter INTENT_FILTER_LOCATION_MODE_CHANGED = new IntentFilter(
            LocationManager.MODE_CHANGED_ACTION);

    private final Context mContext;
    private final LocationManager mLocationManager;
    private final CarServiceProvider mCarServiceProvider;

    private boolean mIsLocationActive;
    private boolean mUserLifecycleListenerRegistered;

    private final BroadcastReceiver mLocationReceiver = new BroadcastReceiver() {
        @Override
        public void onReceive(Context context, Intent intent) {
            updateIconVisibilityForCurrentUser();
        }
    };

    private final CarUserManager.UserLifecycleListener mUserLifecycleListener =
            new CarUserManager.UserLifecycleListener() {
                @Override
                public void onEvent(CarUserManager.UserLifecycleEvent event) {
                    mContext.getMainExecutor().execute(() -> {
                        if (event.getEventType()
                                == CarUserManager.USER_LIFECYCLE_EVENT_TYPE_SWITCHING) {
                            updateIconVisibilityForCurrentUser();
                        }
                    });
                }
            };

    @Inject
    LocationStatusIconController(
            Context context,
            @Main Resources resources,
            CarServiceProvider carServiceProvider) {

        mContext = context;
        mLocationManager = context.getSystemService(LocationManager.class);
        mCarServiceProvider = carServiceProvider;

        context.registerReceiverForAllUsers(mLocationReceiver, INTENT_FILTER_LOCATION_MODE_CHANGED,
                /* broadcastPermission= */ null, /* scheduler= */ null);
        registerForUserChangeEvents();
        setIconDrawableToDisplay(resources.getDrawable(R.drawable.ic_location, context.getTheme()));
        updateIconVisibilityForCurrentUser();
    }

    @Override
    protected void updateStatus() {
        setIconVisibility(mIsLocationActive);
        onStatusUpdated();
    }

    private void updateIconVisibilityForCurrentUser() {
        int fgUserId = ActivityManager.getCurrentUser();
        UserHandle fgUserHandle = UserHandle.of(fgUserId);
        mIsLocationActive = mLocationManager.isLocationEnabledForUser(fgUserHandle);
        updateStatus();
    }

    private void registerForUserChangeEvents() {
        mCarServiceProvider.addListener(car -> {
            CarUserManager carUserManager = (CarUserManager) car.getCarManager(
                    Car.CAR_USER_SERVICE);
            if (carUserManager != null && !mUserLifecycleListenerRegistered) {
                carUserManager.addListener(Runnable::run, mUserLifecycleListener);
                mUserLifecycleListenerRegistered = true;
            } else {
                Log.e(TAG, "CarUserManager could not be obtained.");
            }
        });
    }
}