/*
 * Copyright (C) 2020 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 android.car.test.mocks;

import static org.mockito.ArgumentMatchers.argThat;

import android.annotation.Nullable;
import android.annotation.UserIdInt;
import android.content.pm.UserInfo;
import android.hardware.automotive.vehicle.V2_0.VehiclePropValue;
import android.os.UserHandle;
import android.util.Log;

import org.mockito.ArgumentMatcher;

import java.util.Arrays;

/**
 * Provides custom mockito matcher for Android / Car objects.
 *
 */
public final class CarArgumentMatchers {

    private static final String TAG = CarArgumentMatchers.class.getSimpleName();

    /**
     * Matches if a {@link UserInfo} has the given {@code userId}.
     */
    public static UserInfo isUserInfo(@UserIdInt int userId) {
        return argThat(new UserInfoMatcher(userId));
    }

    /**
     * Matches if a {@link UserHandle} has the given {@code userId}.
     */
    public static UserHandle isUserHandle(@UserIdInt int userId) {
        return argThat(new UserHandleMatcher(userId));
    }

    /**
     * Matches if a {@link VehiclePropValue} has the given {@code prop}.
     */
    public static VehiclePropValue isProperty(int prop) {
        return argThat(new PropertyIdMatcher(prop));
    }

    /**
     * Matches if a {@link VehiclePropValue} has the given {@code prop} and {@code int32} values.
     */
    public static VehiclePropValue isPropertyWithValues(int prop, int...values) {
        return argThat(new PropertyIdMatcher(prop, values));
    }

    private static class UserInfoMatcher implements ArgumentMatcher<UserInfo> {

        public final @UserIdInt int userId;

        private UserInfoMatcher(@UserIdInt int userId) {
            this.userId = userId;
        }

        @Override
        public boolean matches(@Nullable UserInfo argument) {
            if (argument == null) {
                Log.w(TAG, "isUserInfo(): null argument");
                return false;
            }
            if (argument.id != userId) {
                Log.w(TAG, "isUserInfo(): argument mismatch (expected " + userId
                        + ", got " + argument.id);
                return false;
            }
            return true;
        }

        @Override
        public String toString() {
            return "isUserInfo(userId=" + userId + ")";
        }
    }

    private static class UserHandleMatcher implements ArgumentMatcher<UserHandle> {

        public final @UserIdInt int userId;

        private UserHandleMatcher(@UserIdInt int userId) {
            this.userId = userId;
        }

        @Override
        public boolean matches(UserHandle argument) {
            if (argument == null) {
                Log.w(TAG, "isUserHandle(): null argument");
                return false;
            }
            if (argument.getIdentifier() != userId) {
                Log.w(TAG, "isUserHandle(): argument mismatch (expected " + userId
                        + ", got " + argument.getIdentifier());
                return false;
            }
            return true;
        }

        @Override
        public String toString() {
            return "isUserHandle(userId=" + userId + ")";
        }
    }

    private static class PropertyIdMatcher implements ArgumentMatcher<VehiclePropValue> {

        final int mProp;
        private final int[] mValues;

        private PropertyIdMatcher(int prop) {
            this(prop, null);
        }

        private PropertyIdMatcher(int prop, int[] values) {
            mProp = prop;
            mValues = values;
        }

        @Override
        public boolean matches(VehiclePropValue argument) {
            Log.v(TAG, "PropertyIdMatcher: argument=" + argument);
            if (argument.prop != mProp) {
                Log.w(TAG, "PropertyIdMatcher: Invalid prop on " + argument);
                return false;
            }
            if (mValues == null) return true;
            // Make sure values match
            if (mValues.length != argument.value.int32Values.size()) {
                Log.w(TAG, "PropertyIdMatcher: number of values (expected " + mValues.length
                        + ") mismatch on " + argument);
                return false;
            }

            for (int i = 0; i < mValues.length; i++) {
                if (mValues[i] != argument.value.int32Values.get(i)) {
                    Log.w(TAG, "PropertyIdMatcher: value mismatch at index " + i + " on " + argument
                            + ": expected " + Arrays.toString(mValues));
                    return false;
                }
            }
            return true;
        }

        @Override
        public String toString() {
            return "prop: " + mProp + " values: " + Arrays.toString(mValues);
        }
    }

    private CarArgumentMatchers() {
        throw new UnsupportedOperationException("contains only static methods");
    }
}