/*
 * Copyright (C) 2024-present ScyllaDB
 */

/*
 * SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
 */

#pragma once

#include "gms/inet_address.hh"
#include "tasks/task_manager.hh"
#include "utils/chunked_vector.hh"

namespace tasks {

struct task_identity {
    locator::host_id host_id;
    task_id task_id;
};

struct task_status {
    tasks::task_id task_id;
    std::string type;
    task_kind kind;
    std::string scope;
    task_manager::task_state state;
    is_abortable is_abortable;
    db_clock::time_point start_time;
    db_clock::time_point end_time;
    std::string error;
    tasks::task_id parent_id;
    uint64_t sequence_number;
    unsigned shard;
    std::string keyspace;
    std::string table;
    std::string entity;
    std::string progress_units;
    task_manager::task::progress progress;
    utils::chunked_vector<task_identity> children;
};

struct task_stats {
    tasks::task_id task_id;
    std::string type;
    task_kind kind;
    std::string scope;
    task_manager::task_state state;
    uint64_t sequence_number;
    std::string keyspace;
    std::string table;
    std::string entity;
    unsigned shard;
    db_clock::time_point start_time;
    db_clock::time_point end_time;
};

struct status_helper;

class task_handler {
private:
    task_manager& _tm;
    task_id _id;
public:
    explicit task_handler(task_manager& tm, task_id id) noexcept
        : _tm(tm)
        , _id(id)
    {}
    future<task_stats> get_stats();
    future<task_status> get_status();
    future<task_status> wait_for_task(std::optional<std::chrono::seconds> timeout);
    future<utils::chunked_vector<task_status>> get_status_recursively(bool local);
    future<> abort();
private:
    future<status_helper> get_status_helper();
};

class task_not_abortable : public std::exception {
    sstring _cause;
public:
    explicit task_not_abortable(task_id tid)
        : _cause(format("task with id {} cannot be aborted", tid))
    { }

    virtual const char* what() const noexcept override { return _cause.c_str(); }
};

}