/*
 * Copyright (c) 2025 Huawei Device Co., Ltd.
 * 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.
 */

#ifndef HISTREAMER_FOUNDATION_OSAL_TASK_INNER_H
#define HISTREAMER_FOUNDATION_OSAL_TASK_INNER_H

#include <atomic>
#include <functional>
#include <string>
#include <list>
#include <map>

#include "osal/task/task.h"
#include "osal/task/condition_variable.h"
#include "osal/task/mutex.h"
#include "osal/task/autolock.h"
#include "osal/task/pipeline_threadpool.h"

#ifdef MEDIA_FOUNDATION_FFRT
    #include "ffrt.h"
#else
    #include <map>
#endif


namespace OHOS {
namespace Media {

class TaskInner : public std::enable_shared_from_this<TaskInner> {
public:
    explicit TaskInner(const std::string& name, const std::string& groupId, TaskType type,
        TaskPriority priority, bool singleLoop)
    {
        (void)name;
        (void)groupId;
        (void)type;
        (void)priority;
        (void)singleLoop;
    }

    virtual ~TaskInner() {}

    virtual void Init() {}

    virtual void DeInit() {}

    virtual void Start() {}

    virtual void Stop() {}

    virtual void StopAsync() {}

    virtual void Pause() {}

    virtual void PauseAsync() {}

    virtual void RegisterJob(const std::function<int64_t()>& job)
    {
        (void)job;
    }

    virtual void SubmitJobOnce(const std::function<void()>& job, int64_t delay, bool wait)
    {
        (void)job;
        (void)delay;
        (void)wait;
    }
    
    virtual void SubmitJob(const std::function<void()>& job, int64_t delay, bool wait)
    {
        (void)job;
        (void)delay;
        (void)wait;
    }

    virtual bool IsTaskRunning() { return runningState_ == RunningState::STARTED; }

    virtual void UpdateDelayTime(int64_t delayUs)
    {
        (void)delayUs;
    }

    void SetEnableStateChangeLog(bool enable) { isStateLogEnabled_ = enable; }

    int64_t NextJobUs()
    {
        return 0;
    }

    void HandleJob() {}

    static void SleepInTask(unsigned ms)
    {
        (void)ms;
    }

    void UpdateThreadPriority(const uint32_t newPriority, const std::string &strBundleName)
    {
        (void)newPriority;
        (void)strBundleName;
    }

private:
    enum class RunningState : int {
        STARTED,
        PAUSING,
        PAUSED,
        STOPPING,
        STOPPED,
    };

    const std::string name_;
    std::atomic<RunningState> runningState_{RunningState::PAUSED};
    std::atomic<bool> jobState_{false};
    std::function<int64_t()> job_;
    bool singleLoop_ = false;
    int64_t topProcessUs_ {-1};
    bool topIsJob_ = false;
    std::shared_ptr<PipeLineThread> pipelineThread_;
    std::atomic<bool> isStateLogEnabled_{true};
#ifdef MEDIA_FOUNDATION_FFRT
    void DoJob(const std::function<void()>& job);
    std::shared_ptr<ffrt::queue> jobQueue_;
    Mutex stateMutex_;
    ConditionVariable syncCond_;
    ffrt::recursive_mutex jobMutex_;
#else
    void UpdateTop() {}

    int64_t InsertJob(const std::function<void()>& job, int64_t delayUs, bool inJobQueue)
    {
        (void)job;
        (void)delayUs;
        (void)inJobQueue;
        return 0;
    }

    Mutex stateMutex_{};
    FairMutex jobMutex_{};
    ConditionVariable syncCond_{};
    ConditionVariable replyCond_{};
    std::map<int64_t, std::function<void()>> msgQueue_;  // msg will be sorted by timeUs
    std::map<int64_t, std::function<void()>> jobQueue_;  // msg will be sorted by timeUs
#endif
};
} // namespace Media
} // namespace OHOS
#endif // HISTREAMER_FOUNDATION_OSAL_TASK_H