/*
 * Copyright (C) 2009 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.
 */

#define LOG_TAG "VibratorService"

#include <android/hardware/vibrator/1.4/IVibrator.h>
#include <android/hardware/vibrator/BnVibratorCallback.h>
#include <android/hardware/vibrator/IVibrator.h>
#include <binder/IServiceManager.h>

#include "jni.h"
#include <nativehelper/JNIHelp.h>
#include "android_runtime/AndroidRuntime.h"
#include "core_jni_helpers.h"

#include <utils/misc.h>
#include <utils/Log.h>
#include <hardware/vibrator.h>

#include <inttypes.h>
#include <stdio.h>

using android::hardware::Return;
using android::hardware::Void;
using android::hardware::vibrator::V1_0::EffectStrength;
using android::hardware::vibrator::V1_0::Status;
using android::hardware::vibrator::V1_1::Effect_1_1;

namespace V1_0 = android::hardware::vibrator::V1_0;
namespace V1_1 = android::hardware::vibrator::V1_1;
namespace V1_2 = android::hardware::vibrator::V1_2;
namespace V1_3 = android::hardware::vibrator::V1_3;
namespace V1_4 = android::hardware::vibrator::V1_4;
namespace aidl = android::hardware::vibrator;

namespace android {

static jmethodID sMethodIdOnComplete;

// TODO(b/141828236): remove HIDL 1.4 and re-write all of this code to remove
// shim
class VibratorShim : public V1_4::IVibrator {
  public:
    VibratorShim(const sp<aidl::IVibrator>& vib) : mVib(vib) {}

    Return<V1_0::Status> on(uint32_t timeoutMs) override {
        return on_1_4(timeoutMs, nullptr);
    }

    Return<V1_0::Status> off() override {
        return toHidlStatus(mVib->off());
    }

    Return<bool> supportsAmplitudeControl() override {
        int32_t cap = 0;
        if (!mVib->getCapabilities(&cap).isOk()) return false;
        if (mUnderExternalControl) {
           return (cap & aidl::IVibrator::CAP_EXTERNAL_AMPLITUDE_CONTROL) > 0;
        } else {
           return (cap & aidl::IVibrator::CAP_AMPLITUDE_CONTROL) > 0;
        }
    }

    Return<V1_0::Status> setAmplitude(uint8_t amplitude) override {
        return toHidlStatus(mVib->setAmplitude(amplitude));
    }

    Return<void> perform(V1_0::Effect effect, V1_0::EffectStrength strength,
                         perform_cb _hidl_cb) override {
        return perform_1_4(static_cast<V1_3::Effect>(effect), strength, nullptr, _hidl_cb);
    }

    Return<void> perform_1_1(V1_1::Effect_1_1 effect, V1_0::EffectStrength strength,
                             perform_1_1_cb _hidl_cb) override {
        return perform_1_4(static_cast<V1_3::Effect>(effect), strength, nullptr, _hidl_cb);
    }

    Return<void> perform_1_2(V1_2::Effect effect, V1_0::EffectStrength strength,
                             perform_1_2_cb _hidl_cb) override {
        return perform_1_4(static_cast<V1_3::Effect>(effect), strength, nullptr, _hidl_cb);
    }

    Return<bool> supportsExternalControl() override {
        int32_t cap = 0;
        if (!mVib->getCapabilities(&cap).isOk()) return false;
        return (cap & aidl::IVibrator::CAP_EXTERNAL_CONTROL) > 0;
    }

    Return<V1_0::Status> setExternalControl(bool enabled) override {
        Return<V1_0::Status> status = toHidlStatus(mVib->setExternalControl(enabled));
        if (status.isOk() && status == V1_0::Status::OK) {
            mUnderExternalControl = enabled;
        }
        return status;
    }

    Return<void> perform_1_3(V1_3::Effect effect, V1_0::EffectStrength strength,
                             perform_1_3_cb _hidl_cb) override {
        return perform_1_4(static_cast<V1_3::Effect>(effect), strength, nullptr, _hidl_cb);
    }

    Return<uint32_t> getCapabilities() override {
        static_assert(static_cast<int32_t>(V1_4::Capabilities::ON_COMPLETION_CALLBACK) ==
                      static_cast<int32_t>(aidl::IVibrator::CAP_ON_CALLBACK));
        static_assert(static_cast<int32_t>(V1_4::Capabilities::PERFORM_COMPLETION_CALLBACK) ==
                      static_cast<int32_t>(aidl::IVibrator::CAP_PERFORM_CALLBACK));

        int32_t cap;
        if (!mVib->getCapabilities(&cap).isOk()) return 0;
        return (cap & (aidl::IVibrator::CAP_ON_CALLBACK |
                       aidl::IVibrator::CAP_PERFORM_CALLBACK)) > 0;
    }

    Return<V1_0::Status> on_1_4(uint32_t timeoutMs,
                                const sp<V1_4::IVibratorCallback>& callback) override {
        sp<aidl::IVibratorCallback> cb = callback ? new CallbackShim(callback) : nullptr;
        return toHidlStatus(mVib->on(timeoutMs, cb));
    }

    Return<void> perform_1_4(V1_3::Effect effect, V1_0::EffectStrength strength,
                             const sp<V1_4::IVibratorCallback>& callback,
                             perform_1_4_cb _hidl_cb) override {
        static_assert(static_cast<uint8_t>(V1_0::EffectStrength::LIGHT) ==
                      static_cast<uint8_t>(aidl::EffectStrength::LIGHT));
        static_assert(static_cast<uint8_t>(V1_0::EffectStrength::MEDIUM) ==
                      static_cast<uint8_t>(aidl::EffectStrength::MEDIUM));
        static_assert(static_cast<uint8_t>(V1_0::EffectStrength::STRONG) ==
                      static_cast<uint8_t>(aidl::EffectStrength::STRONG));
        static_assert(static_cast<uint8_t>(V1_3::Effect::CLICK) ==
                      static_cast<uint8_t>(aidl::Effect::CLICK));
        static_assert(static_cast<uint8_t>(V1_3::Effect::DOUBLE_CLICK) ==
                      static_cast<uint8_t>(aidl::Effect::DOUBLE_CLICK));
        static_assert(static_cast<uint8_t>(V1_3::Effect::TICK) ==
                      static_cast<uint8_t>(aidl::Effect::TICK));
        static_assert(static_cast<uint8_t>(V1_3::Effect::THUD) ==
                      static_cast<uint8_t>(aidl::Effect::THUD));
        static_assert(static_cast<uint8_t>(V1_3::Effect::POP) ==
                      static_cast<uint8_t>(aidl::Effect::POP));
        static_assert(static_cast<uint8_t>(V1_3::Effect::HEAVY_CLICK) ==
                      static_cast<uint8_t>(aidl::Effect::HEAVY_CLICK));
        static_assert(static_cast<uint8_t>(V1_3::Effect::RINGTONE_1) ==
                      static_cast<uint8_t>(aidl::Effect::RINGTONE_1));
        static_assert(static_cast<uint8_t>(V1_3::Effect::RINGTONE_2) ==
                      static_cast<uint8_t>(aidl::Effect::RINGTONE_2));
        static_assert(static_cast<uint8_t>(V1_3::Effect::RINGTONE_15) ==
                      static_cast<uint8_t>(aidl::Effect::RINGTONE_15));
        static_assert(static_cast<uint8_t>(V1_3::Effect::TEXTURE_TICK) ==
                      static_cast<uint8_t>(aidl::Effect::TEXTURE_TICK));

        sp<aidl::IVibratorCallback> cb = callback ? new CallbackShim(callback) : nullptr;
        int timeoutMs = 0;
        Return<V1_0::Status> status = toHidlStatus(
            mVib->perform(static_cast<aidl::Effect>(effect),
                          static_cast<aidl::EffectStrength>(strength), cb, &timeoutMs));

        if (status.isOk()) {
            _hidl_cb(status, timeoutMs);
            return android::hardware::Status::ok();
        } else {
            return android::hardware::details::StatusOf<V1_0::Status, void>(status);
        }
    }
  private:
    sp<aidl::IVibrator> mVib;
    bool mUnderExternalControl = false;

    Return<V1_0::Status> toHidlStatus(const android::binder::Status& status) {
        switch(status.exceptionCode()) {
            using android::hardware::Status;
            case Status::EX_NONE: return V1_0::Status::OK;
            case Status::EX_ILLEGAL_ARGUMENT: return V1_0::Status::BAD_VALUE;
            case Status::EX_UNSUPPORTED_OPERATION: return V1_0::Status::UNSUPPORTED_OPERATION;
            case Status::EX_TRANSACTION_FAILED: {
                return Status::fromStatusT(status.transactionError());
            }
        }
        return V1_0::Status::UNKNOWN_ERROR;
    }

    class CallbackShim : public aidl::BnVibratorCallback {
      public:
        CallbackShim(const sp<V1_4::IVibratorCallback>& cb) : mCb(cb) {}
        binder::Status onComplete() {
            mCb->onComplete();
            return binder::Status::ok(); // oneway, local call
        }
      private:
        sp<V1_4::IVibratorCallback> mCb;
    };
};

class VibratorCallback : public V1_4::IVibratorCallback {
    public:
        VibratorCallback(JNIEnv *env, jobject vibration) :
        mVibration(MakeGlobalRefOrDie(env, vibration)) {}

        ~VibratorCallback() {
            JNIEnv *env = AndroidRuntime::getJNIEnv();
            env->DeleteGlobalRef(mVibration);
        }

        Return<void> onComplete() override {
            auto env = AndroidRuntime::getJNIEnv();
            env->CallVoidMethod(mVibration, sMethodIdOnComplete);
            return Void();
        }

    private:
        jobject mVibration;
};

static constexpr int NUM_TRIES = 2;

// Creates a Return<R> with STATUS::EX_NULL_POINTER.
template<class R>
inline Return<R> NullptrStatus() {
    using ::android::hardware::Status;
    return Return<R>{Status::fromExceptionCode(Status::EX_NULL_POINTER)};
}

template <typename I>
class HalWrapper {
  public:
    static std::unique_ptr<HalWrapper> Create() {
        sp<aidl::IVibrator> aidlVib = waitForVintfService<aidl::IVibrator>();
        if (aidlVib) {
            return std::unique_ptr<HalWrapper>(new HalWrapper(new VibratorShim(aidlVib)));
        }

        // Assume that if getService returns a nullptr, HAL is not available on the
        // device.
        auto hal = I::getService();
        return hal ? std::unique_ptr<HalWrapper>(new HalWrapper(std::move(hal))) : nullptr;
    }

    // Helper used to transparently deal with the vibrator HAL becoming unavailable.
    template<class R, class... Args0, class... Args1>
    Return<R> call(Return<R> (I::* fn)(Args0...), Args1&&... args1) {
        // Return<R> doesn't have a default constructor, so make a Return<R> with
        // STATUS::EX_NONE.
        using ::android::hardware::Status;
        Return<R> ret{Status::fromExceptionCode(Status::EX_NONE)};

        // Note that ret is guaranteed to be changed after this loop.
        for (int i = 0; i < NUM_TRIES; ++i) {
            ret = (mHal == nullptr) ? NullptrStatus<R>()
                    : (*mHal.*fn)(std::forward<Args1>(args1)...);

            if (ret.isOk()) {
                break;
            }

            ALOGE("Failed to issue command to vibrator HAL. Retrying.");

            // Restoring connection to the HAL.
            sp<aidl::IVibrator> aidlVib = checkVintfService<aidl::IVibrator>();
            if (aidlVib) {
                mHal = new VibratorShim(aidlVib);
            } else {
                mHal = I::tryGetService();
            }
        }
        return ret;
    }

  private:
    HalWrapper(sp<I> &&hal) : mHal(std::move(hal)) {}

  private:
    sp<I> mHal;
};

template <typename I>
static auto getHal() {
    static auto sHalWrapper = HalWrapper<I>::Create();
    return sHalWrapper.get();
}

template<class R, class I, class... Args0, class... Args1>
Return<R> halCall(Return<R> (I::* fn)(Args0...), Args1&&... args1) {
    auto hal = getHal<I>();
    return hal ? hal->call(fn, std::forward<Args1>(args1)...) : NullptrStatus<R>();
}

template<class R>
bool isValidEffect(jlong effect) {
    if (effect < 0) {
        return false;
    }
    R val = static_cast<R>(effect);
    auto iter = hardware::hidl_enum_range<R>();
    return val >= *iter.begin() && val <= *std::prev(iter.end());
}

static void vibratorInit(JNIEnv *env, jclass clazz)
{
    halCall(&V1_0::IVibrator::ping).isOk();
}

static jboolean vibratorExists(JNIEnv* /* env */, jclass /* clazz */)
{
    return halCall(&V1_0::IVibrator::ping).isOk() ? JNI_TRUE : JNI_FALSE;
}

static void vibratorOn(JNIEnv* /* env */, jclass /* clazz */, jlong timeout_ms)
{
    Status retStatus = halCall(&V1_0::IVibrator::on, timeout_ms).withDefault(Status::UNKNOWN_ERROR);
    if (retStatus != Status::OK) {
        ALOGE("vibratorOn command failed (%" PRIu32 ").", static_cast<uint32_t>(retStatus));
    }
}

static void vibratorOff(JNIEnv* /* env */, jclass /* clazz */)
{
    Status retStatus = halCall(&V1_0::IVibrator::off).withDefault(Status::UNKNOWN_ERROR);
    if (retStatus != Status::OK) {
        ALOGE("vibratorOff command failed (%" PRIu32 ").", static_cast<uint32_t>(retStatus));
    }
}

static jlong vibratorSupportsAmplitudeControl(JNIEnv*, jclass) {
    return halCall(&V1_0::IVibrator::supportsAmplitudeControl).withDefault(false);
}

static void vibratorSetAmplitude(JNIEnv*, jclass, jint amplitude) {
    Status status = halCall(&V1_0::IVibrator::setAmplitude, static_cast<uint32_t>(amplitude))
        .withDefault(Status::UNKNOWN_ERROR);
    if (status != Status::OK) {
      ALOGE("Failed to set vibrator amplitude (%" PRIu32 ").",
            static_cast<uint32_t>(status));
    }
}

static jboolean vibratorSupportsExternalControl(JNIEnv*, jclass) {
    return halCall(&V1_3::IVibrator::supportsExternalControl).withDefault(false);
}

static void vibratorSetExternalControl(JNIEnv*, jclass, jboolean enabled) {
    Status status = halCall(&V1_3::IVibrator::setExternalControl, static_cast<uint32_t>(enabled))
        .withDefault(Status::UNKNOWN_ERROR);
    if (status != Status::OK) {
      ALOGE("Failed to set vibrator external control (%" PRIu32 ").",
            static_cast<uint32_t>(status));
    }
}

static jlong vibratorPerformEffect(JNIEnv* env, jclass, jlong effect, jlong strength,
                                   jobject vibration) {
    Status status;
    uint32_t lengthMs;
    auto callback = [&status, &lengthMs](Status retStatus, uint32_t retLengthMs) {
        status = retStatus;
        lengthMs = retLengthMs;
    };
    EffectStrength effectStrength(static_cast<EffectStrength>(strength));

    Return<void> ret;
    if (auto hal = getHal<V1_4::IVibrator>(); hal && isValidEffect<V1_3::Effect>(effect)) {
        sp<VibratorCallback> effectCallback = new VibratorCallback(env, vibration);
        ret = hal->call(&V1_4::IVibrator::perform_1_4, static_cast<V1_3::Effect>(effect),
                effectStrength, effectCallback, callback);
    } else if (isValidEffect<V1_0::Effect>(effect)) {
        ret = halCall(&V1_0::IVibrator::perform, static_cast<V1_0::Effect>(effect),
                effectStrength, callback);
    } else if (isValidEffect<Effect_1_1>(effect)) {
        ret = halCall(&V1_1::IVibrator::perform_1_1, static_cast<Effect_1_1>(effect),
                           effectStrength, callback);
    } else if (isValidEffect<V1_2::Effect>(effect)) {
        ret = halCall(&V1_2::IVibrator::perform_1_2, static_cast<V1_2::Effect>(effect),
                           effectStrength, callback);
    } else if (isValidEffect<V1_3::Effect>(effect)) {
        ret = halCall(&V1_3::IVibrator::perform_1_3, static_cast<V1_3::Effect>(effect),
                           effectStrength, callback);
    } else {
        ALOGW("Unable to perform haptic effect, invalid effect ID (%" PRId32 ")",
                static_cast<int32_t>(effect));
        return -1;
    }

    if (!ret.isOk()) {
        ALOGW("Failed to perform effect (%" PRId32 ")", static_cast<int32_t>(effect));
        return -1;
    }

    if (status == Status::OK) {
        return lengthMs;
    } else if (status != Status::UNSUPPORTED_OPERATION) {
        // Don't warn on UNSUPPORTED_OPERATION, that's a normal event and just means the motor
        // doesn't have a pre-defined waveform to perform for it, so we should just give the
        // opportunity to fall back to the framework waveforms.
        ALOGE("Failed to perform haptic effect: effect=%" PRId64 ", strength=%" PRId32
                ", error=%" PRIu32 ").", static_cast<int64_t>(effect),
                static_cast<int32_t>(strength), static_cast<uint32_t>(status));
    }

    return -1;
}

static jlong vibratorGetCapabilities(JNIEnv*, jclass) {
    return halCall(&V1_4::IVibrator::getCapabilities).withDefault(0);
}

static const JNINativeMethod method_table[] = {
    { "vibratorExists", "()Z", (void*)vibratorExists },
    { "vibratorInit", "()V", (void*)vibratorInit },
    { "vibratorOn", "(J)V", (void*)vibratorOn },
    { "vibratorOff", "()V", (void*)vibratorOff },
    { "vibratorSupportsAmplitudeControl", "()Z", (void*)vibratorSupportsAmplitudeControl},
    { "vibratorSetAmplitude", "(I)V", (void*)vibratorSetAmplitude},
    { "vibratorPerformEffect", "(JJLcom/android/server/VibratorService$Vibration;)J",
        (void*)vibratorPerformEffect},
    { "vibratorSupportsExternalControl", "()Z", (void*)vibratorSupportsExternalControl},
    { "vibratorSetExternalControl", "(Z)V", (void*)vibratorSetExternalControl},
    { "vibratorGetCapabilities", "()J", (void*)vibratorGetCapabilities},
};

int register_android_server_VibratorService(JNIEnv *env)
{
    sMethodIdOnComplete = GetMethodIDOrDie(env,
            FindClassOrDie(env, "com/android/server/VibratorService$Vibration"),
            "onComplete", "()V");
    return jniRegisterNativeMethods(env, "com/android/server/VibratorService",
            method_table, NELEM(method_table));
}

};