/*
 * Copyright 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.
 */

#pragma once

#include <array>
#include <functional>
#include <mutex>
#include <thread>

#include <android-base/thread_annotations.h>

#include <scheduler/TimeKeeper.h>

namespace android::scheduler {

class Timer : public TimeKeeper {
public:
    Timer();
    ~Timer();

    nsecs_t now() const final;

    // NB: alarmAt and alarmCancel are threadsafe; with the last-returning function being effectual
    //     Most users will want to serialize thes calls so as to be aware of the timer state.
    void alarmAt(std::function<void()>, nsecs_t time) final;
    void alarmCancel() final;

    void dump(std::string&) const final;

protected:
    // For unit testing
    int mEpollFd = -1;

private:
    enum class DebugState {
        Reset,
        Running,
        Waiting,
        Reading,
        InCallback,
        Terminated,

        ftl_last = Terminated
    };

    void reset() EXCLUDES(mMutex);
    void cleanup() REQUIRES(mMutex);
    void setDebugState(DebugState) EXCLUDES(mMutex);
    void setCallback(std::function<void()>&&) REQUIRES(mMutex);

    int mTimerFd = -1;

    std::array<int, 2> mPipes = {-1, -1};

    std::thread mDispatchThread;
    void threadMain();
    bool dispatch();
    void endDispatch();

    mutable std::mutex mMutex;

    std::function<void()> mCallback GUARDED_BY(mMutex);
    bool mExpectingCallback GUARDED_BY(mMutex) = false;
    DebugState mDebugState GUARDED_BY(mMutex);
};

} // namespace android::scheduler