/*
 * A O(1) Timer library
 *
 * Hannes Gredler, July 2020
 *
 * Copyright (C) 2020-2023, RtBrick, Inc.
 * SPDX-License-Identifier: BSD-3-Clause
 */
#include "timer.h"
#include "logging.h"

/**
 * Set timer expiration.
 */
static void
timer_set_expire(timer_s *timer, time_t sec, long nsec)
{
    timer->expire.tv_sec += sec;
    timer->expire.tv_nsec += nsec;

    /* Handle nsec overflow. */
    if(timer->expire.tv_nsec >= 1e9) {
        timer->expire.tv_nsec -= 1e9;
        timer->expire.tv_sec++;
    }

    timer->expired = false;
}

/**
 * Compare two timespecs.
 *
 * return -1 if ts1 is older than ts2
 * return +1 if ts1 is newer than ts2
 * return  0 if ts1 is equal to ts2
 */
static int
timespec_compare(struct timespec *ts1, struct timespec *ts2)
{
    if(ts1->tv_sec < ts2->tv_sec) {
        return -1;
    }

    if(ts1->tv_sec > ts2->tv_sec) {
        return +1;
    }

    if(ts1->tv_nsec < ts2->tv_nsec) {
        return -1;
    }

    if(ts1->tv_nsec > ts2->tv_nsec) {
        return +1;
    }

    return 0;
}

/**
 * Add two timestamps x and y, storing the result in result.
 */
void
timespec_add(struct timespec *result, struct timespec *x, struct timespec *y)
{
    result->tv_sec = x->tv_sec + y->tv_sec;
    result->tv_nsec = x->tv_nsec + y->tv_nsec;

    /* Avoid overflow of result->tv_nsec */
    if(result->tv_nsec >= 1e9) {
        result->tv_nsec -= 1e9;
        result->tv_sec += 1;
    }
}

/**
 * Subtract the timestamps x from y, storing the result in result.
 */
void
timespec_sub(struct timespec *result, struct timespec *x, struct timespec *y)
{

    if(x->tv_sec < y->tv_sec) {
        result->tv_sec = 0;
        result->tv_nsec = 0;
        return;
    }

    /* Avoid overflow of result->tv_nsec */
    if(x->tv_nsec < y->tv_nsec) {
        if(x->tv_sec == y->tv_sec) {
            result->tv_sec = 0;
            result->tv_nsec = 0;
            return;
        }
        result->tv_nsec = x->tv_nsec + 1e9 - y->tv_nsec;
        result->tv_sec = x->tv_sec - y->tv_sec - 1;
    } else {
        result->tv_sec = x->tv_sec - y->tv_sec;
        result->tv_nsec = x->tv_nsec - y->tv_nsec;
    }
}

/**
 * Format a timestamp in one of four buffers.
 * This way we can format upto 4 timespecs in one printf() call.
 */
char *
timespec_format(struct timespec *x)
{
    static char buffer[4][32];
    static int idx = 0;
    char *ret;

    ret = buffer[idx];
    idx = (idx+1) & 3;

    snprintf(ret, 32, "%lu.%06lus", x->tv_sec, x->tv_nsec / 1000);

    return ret;
}

/**
 * Enqueue a timer for change processing.
 */
static void
timer_change(timer_s *timer)
{
    timer_root_s *timer_root;

    /* Are we already on the timer change queue? */
    if(timer->on_change_list) {
        return;
    }

    timer_root = timer->timer_bucket->timer_root;
    CIRCLEQ_INSERT_TAIL(&timer_root->timer_change_qhead, timer, timer_change_qnode);
    timer->on_change_list = true;
}

static void
timer_enqueue_bucket(timer_root_s *root, timer_s *timer, time_t sec, long nsec)
{
    timer_bucket_s *timer_bucket;

    /* Find the bucket for insertion. */
    CIRCLEQ_FOREACH(timer_bucket, &root->timer_bucket_qhead, timer_bucket_qnode) {
        if(timer_bucket->sec != sec ||  timer_bucket->nsec != nsec) {
            continue;
        }
        /* Found it! */
        goto INSERT;
    }

    /* No bucket found that matches the timer values. 
     * Create a fresh bucket. */
    timer_bucket = calloc(1, sizeof(timer_bucket_s));
    if(!timer_bucket) {
        return;
    }

    CIRCLEQ_INSERT_TAIL(&root->timer_bucket_qhead, timer_bucket, timer_bucket_qnode);
    CIRCLEQ_INIT(&timer_bucket->timer_qhead);
    timer_bucket->sec = sec;
    timer_bucket->nsec = nsec;
    timer_bucket->timer_root = root;
    root->buckets++;

#ifdef BNGBLASTER_TIMER_LOGGING
    LOG(TIMER_DETAIL, "Add timer bucket %lu.%06lus\n",
        timer_bucket->sec, timer_bucket->nsec/1000);
#endif

 INSERT:
    timer->timer_bucket = timer_bucket;
    CIRCLEQ_INSERT_TAIL(&timer_bucket->timer_qhead, timer, timer_qnode);
    timer_bucket->timers++;
}

/**
 * Dequeue a timer from its timer_bucket.
 */
static void
timer_dequeue_bucket(timer_s *timer)
{
    timer_root_s *timer_root;
    timer_bucket_s *timer_bucket;

    timer_bucket = timer->timer_bucket;
    timer_root = timer_bucket->timer_root;

    CIRCLEQ_REMOVE(&timer_bucket->timer_qhead, timer, timer_qnode);
    timer_bucket->timers--;
    timer->timer_bucket = NULL;

    /* If the last timer of a bucket is gone, 
     * remove the bucket as well. */
    if(!timer_bucket->timers) {
        CIRCLEQ_REMOVE(&timer_root->timer_bucket_qhead, timer_bucket, timer_bucket_qnode);

#ifdef BNGBLASTER_TIMER_LOGGING
        LOG(TIMER_DETAIL, "  Delete timer bucket %lu.%06lus\n",
            timer_bucket->sec, timer_bucket->nsec/1000);
#endif

        free(timer_bucket);
        timer_root->buckets--;
    }
}

static void
timer_requeue(timer_s *timer, time_t sec, long nsec)
{
    timer_root_s *timer_root;
    timer_bucket_s *timer_bucket;

    timer_bucket = timer->timer_bucket;
    timer_root = timer_bucket->timer_root;

    timer_set_expire(timer, sec, nsec);

    /* If the expiration {sec,nsec} matches the bucket, then simply
     * timer dequeue and enqueue to keep correct temporal ordering.
     * If there is no match, do a slightly more expensive
     * bucket dequeue and enqueue. */
    if(timer_bucket->sec == sec && timer_bucket->nsec == nsec) {
        CIRCLEQ_REMOVE(&timer_bucket->timer_qhead, timer, timer_qnode);
        CIRCLEQ_INSERT_TAIL(&timer_bucket->timer_qhead, timer, timer_qnode);
    } else {
        timer_dequeue_bucket(timer);
        timer_enqueue_bucket(timer_root, timer, sec, nsec);
    }

#ifdef BNGBLASTER_TIMER_LOGGING
    LOG(TIMER_DETAIL, "  Reset %s timer, expire in %lu.%06lus\n", 
        timer->name, sec, nsec/1000);
#endif

}

/**
 * Smear all the timer of a given bucket to expire equi-distant.
 * Call this function periodically to avoid clustering of timers.
 */
void
timer_smear_bucket(timer_root_s *root, time_t sec, long nsec)
{
    timer_bucket_s *timer_bucket;
    timer_s *timer, *last_timer;
    struct timespec now, diff, step;
    long step_nsec;

    /* Find the bucket for smearing. */
    CIRCLEQ_FOREACH(timer_bucket, &root->timer_bucket_qhead, timer_bucket_qnode) {
        if(timer_bucket->sec != sec ||  timer_bucket->nsec != nsec) {
            continue;
        }

        /* Found the bucket. Next compute the timespan 
         * between now and last timer. */
        last_timer = CIRCLEQ_LAST(&timer_bucket->timer_qhead);
        if(timer_bucket->timers > 1 && last_timer) {
            clock_gettime(CLOCK_MONOTONIC, &now);
            timespec_sub(&diff, &last_timer->expire, &now);
            step_nsec = (diff.tv_sec * 1e9 + diff.tv_nsec) / (timer_bucket->timers); /* calculate smear step */
            step.tv_sec = step_nsec / 1e9;
            step.tv_nsec = step_nsec - (step.tv_sec * 1e9);

#ifdef BNGBLASTER_TIMER_LOGGING
            LOG(TIMER_DETAIL, "Smear %u timers in bucket %lu.%06lus\n", timer_bucket->timers, sec, nsec);
            LOG(TIMER_DETAIL, "Now %lu.%06lus, last expire %lu.%06lus, step %lu.%06lus\n",
                now.tv_sec, now.tv_nsec / 1000,
                last_timer->expire.tv_sec, last_timer->expire.tv_nsec / 1000,
                step.tv_sec, step.tv_nsec / 1000);
#endif

            /*  Now walk all timers and space them <step> apart. */
            CIRCLEQ_FOREACH(timer, &timer_bucket->timer_qhead, timer_qnode) {
                if(timer->no_smear) continue;
                timespec_add(&timer->expire, &now, &step);
                now = timer->expire;
#ifdef BNGBLASTER_TIMER_LOGGING
                LOG(TIMER_DETAIL, "  Smear %s -> expire %lu.%06lus\n", timer->name,
                    timer->expire.tv_sec, timer->expire.tv_nsec / 1000);
#endif
            }
        }
        return;
    }
}

/**
 * Smear all the timer of all buckets to expire equi-distant.
 */
void
timer_smear_all_buckets(timer_root_s *root)
{
    timer_bucket_s *timer_bucket;
    timer_s *timer, *last_timer;
    struct timespec now, diff, step;
    long step_nsec;

    /* Iterate over all buckets for smearing. */
    CIRCLEQ_FOREACH(timer_bucket, &root->timer_bucket_qhead, timer_bucket_qnode) {
        /* Compute the time span between now and last timer. */
        last_timer = CIRCLEQ_LAST(&timer_bucket->timer_qhead);
        if(timer_bucket->timers > 1 && last_timer) {
            clock_gettime(CLOCK_MONOTONIC, &now);
            timespec_sub(&diff, &last_timer->expire, &now);
            step_nsec = (diff.tv_sec * 1e9 + diff.tv_nsec) / (timer_bucket->timers); /* calculate smear step */
            step.tv_sec = step_nsec / 1e9;
            step.tv_nsec = step_nsec - (step.tv_sec * 1e9);

#ifdef BNGBLASTER_TIMER_LOGGING
            LOG(TIMER_DETAIL, "Smear %u timers in bucket %lu.%06lus\n", timer_bucket->timers, timer_bucket->sec, timer_bucket->nsec);
            LOG(TIMER_DETAIL, "Now %lu.%06lus, last expire %lu.%06lus, step %lu.%06lus\n",
                now.tv_sec, now.tv_nsec / 1000,
                last_timer->expire.tv_sec, last_timer->expire.tv_nsec / 1000,
                step.tv_sec, step.tv_nsec / 1000);
#endif

            /* Now walk all timers and space them <step> apart. */
            CIRCLEQ_FOREACH(timer, &timer_bucket->timer_qhead, timer_qnode) {
                if(timer->no_smear) continue;
                timespec_add(&timer->expire, &now, &step);
                now = timer->expire;
#ifdef BNGBLASTER_TIMER_LOGGING
                LOG(TIMER_DETAIL, "  Smear %s -> expire %lu.%06lus\n", timer->name,
                    last_timer->expire.tv_sec, last_timer->expire.tv_nsec / 1000);
#endif

            }
        }
    }
}

void
timer_no_smear(timer_s *timer)
{
    if(timer) {
        timer->no_smear = true;
    }
}

/**
 * We do not delete timers, but rather dequeue them and move them to
 * the garbage collection queue, where they may get recycled.
 */
static void
timer_del_internal(timer_s *timer)
{
    timer_bucket_s *timer_bucket;
    timer_root_s *timer_root;

#ifdef BNGBLASTER_TIMER_LOGGING
    LOG(TIMER, "  Delete %s timer\n", timer->name);
#endif

    timer_bucket = timer->timer_bucket;
    if(timer_bucket) {
        timer_root = timer_bucket->timer_root;
        timer_dequeue_bucket(timer);
        /* Add to GC list */
        CIRCLEQ_INSERT_TAIL(&timer_root->timer_gc_qhead, timer, timer_qnode);
        timer_root->gc++;
        if(timer->ptimer) {
            *timer->ptimer = NULL; /* delete references to this timer */
            timer->ptimer= NULL;
        }
    }
}

/**
 * Mark a timer for deletion.
 */
void
timer_del(timer_s *timer)
{
    if(timer) {
        timer->delete = true;
        timer_change(timer);
    }
}

/**
 * Deferred processing of all timers.
 */
static void
timer_process_changes(timer_root_s *root)
{
    timer_s *timer;
    timer_bucket_s *timer_bucket;

    struct timespec now;
    clock_gettime(CLOCK_MONOTONIC, &now);

    while(!CIRCLEQ_EMPTY(&root->timer_change_qhead)) {
        timer = CIRCLEQ_FIRST(&root->timer_change_qhead);
        timer_bucket = timer->timer_bucket;

        /* Changes are only processed once.
         * Take this timer off the change list. */
        CIRCLEQ_REMOVE(&root->timer_change_qhead, timer, timer_change_qnode);
        timer->on_change_list = false;

        if(timer->delete) {
            /* Delete. */
            timer_del_internal(timer);
            continue;
        }

        if(timer->periodic) {
            /* Requeue. */
            if(timer->reset) {
                /* Reset timer start time. */
                timer->expire.tv_sec = now.tv_sec;
                timer->expire.tv_nsec = now.tv_nsec;
            }
            timer_requeue(timer, timer_bucket->sec, timer_bucket->nsec);
            continue;
        }
    }
}

/**
 * Enqueue a timer with a given callback function onto 
 * the hierarchical timer list.
 */
void
timer_add(timer_root_s *root, timer_s **ptimer, char *name,
          time_t sec, long nsec,
          void *data, void (*cb)(timer_s *))
{
    timer_s *timer = *ptimer;

    /* This timer already is enqueued. Requeue. */
    if(timer) {
        clock_gettime(CLOCK_MONOTONIC, &timer->expire);
        timer_requeue(timer, sec, nsec);
        /* Update data and cb if there was a change.
         * Do the reformatting of name only during a change. */
        if(timer->data != data || timer->cb != cb) {
            strncpy(timer->name, name, sizeof(timer->name));
            timer->data = data;
            timer->cb = cb;
        }
        return;
    }

    if(CIRCLEQ_EMPTY(&root->timer_gc_qhead)) {
        /* GC queue is empty, make a fresh allocation. */
        timer = calloc(1, sizeof(timer_s));
    } else {
        /* Dequeue the first entry on the GC list and recycle. */
        timer = CIRCLEQ_FIRST(&root->timer_gc_qhead);
        CIRCLEQ_REMOVE(&root->timer_gc_qhead, timer, timer_qnode);
        root->gc--;
        memset(timer, 0, sizeof(timer_s));
    }

    if(!timer) {
        return;
    }

    /* Store name, data, callback and misc. data. */
    strncpy(timer->name, name, sizeof(timer->name));
    timer->data = data;
    timer->cb = cb;
    clock_gettime(CLOCK_MONOTONIC, &timer->expire);
    timer_set_expire(timer, sec, nsec);
    timer->ptimer = ptimer;
    *ptimer = timer;

    /* Enqueue it into the correct timer bucket. */
    timer_enqueue_bucket(root, timer, sec, nsec);

#ifdef BNGBLASTER_TIMER_LOGGING
    LOG(TIMER, "Add %s timer, expire in %lu.%06lus\n", timer->name, sec, nsec/1000);
#endif
}

void
timer_add_periodic(timer_root_s *root, timer_s **ptimer, char *name,
                   time_t sec, long nsec, 
                   void *data, void (*cb)(timer_s *))
{
    timer_s *timer;

    timer_add(root, ptimer, name, sec, nsec, data, cb);

    timer = *ptimer;
    if(timer) {
        timer->periodic = true;
        timer->reset = true;
    }
}

/**
 * Process the timer queue.
 *
 * @param root timer root
 */
void
timer_walk(timer_root_s *root)
{
    timer_s *timer;
    timer_bucket_s *timer_bucket;
    struct timespec now, min, sleep, rem;
    int res;

    /* No buckets filled and we're done. */
    if(CIRCLEQ_EMPTY(&root->timer_bucket_qhead)) {
        return;
    }

    clock_gettime(CLOCK_MONOTONIC, &now);

#ifdef BNGBLASTER_TIMER_LOGGING
    LOG(TIMER_DETAIL, "Walk timer queue, now %lu.%06lus\n",
        now.tv_sec, now.tv_nsec / 1000);
#endif

    min.tv_sec = 0;
    min.tv_nsec = 0;

    /* Walk all buckets. */
    CIRCLEQ_FOREACH(timer_bucket, &root->timer_bucket_qhead, timer_bucket_qnode) {

#ifdef BNGBLASTER_TIMER_LOGGING
        LOG(TIMER_DETAIL, "  Checking timer bucket %lu.%06lus\n",
            timer_bucket->sec, timer_bucket->nsec/1000);
#endif

        /* First pass. Call into expired nodes. */
        CIRCLEQ_FOREACH(timer, &timer_bucket->timer_qhead, timer_qnode) {

            /* Hitting the first non-expired timer means
             * we're done processing this buckets queue. */
            if(timespec_compare(&timer->expire, &now) == 1) {
                break;
            }

            /* Everything from here one is expired. */
            timer->expired = true;

            /* Execute callback. */
            if(timer->cb) {
                timer->timestamp = &now;
                (*timer->cb)(timer);
#ifdef BNGBLASTER_TIMER_LOGGING
                LOG(TIMER_DETAIL, "  Firing %s timer\n", timer->name);
#endif
            }

            if(timer->periodic) {
                /* Periodic timers are simple de-queued and
                 * re-inserted at the tail of this buckets queue. */
                timer_change(timer);
            } else {
                /* Everything else gets deleted. */
                timer_del(timer);
            }
        }
    }

    /* Process all changes from the last timer run. */
    timer_process_changes(root);

    /* Second pass. Figure out min sleep time. */
    CIRCLEQ_FOREACH(timer_bucket, &root->timer_bucket_qhead, timer_bucket_qnode) {
        CIRCLEQ_FOREACH(timer, &timer_bucket->timer_qhead, timer_qnode) {

            /* Ignore deleted timers that wait for change processing. */
            if(timer->delete) {
                continue;
            }

            /* First timer in the queue becomes the actual minimum. */
            if(min.tv_sec == 0 && min.tv_nsec == 0) {
                min.tv_sec = timer->expire.tv_sec;
                min.tv_nsec = timer->expire.tv_nsec;
            }

            /* Find the min timer. */
            if(timespec_compare(&timer->expire, &min) == -1) {
                min.tv_sec = timer->expire.tv_sec;
                min.tv_nsec = timer->expire.tv_nsec;
#ifdef BNGBLASTER_TIMER_LOGGING
                LOG(TIMER_DETAIL, "New minimum sleep (%s) timer, found %lu.%06lus\n",
                    timer->name, min.tv_sec, min.tv_nsec / 1000);
#endif
            }
            /* Hitting the first non-expired timer means
             * we're done processing this buckets queue. */
            if(timespec_compare(&timer->expire, &now) == 1) {
                break;
            }
        }
    }

    /* Calculate the sleep timer. */
#ifdef BNGBLASTER_TIMER_LOGGING
    LOG(TIMER_DETAIL, "  Now %lu.%06lus\n", now.tv_sec, now.tv_nsec / 1000);
    LOG(TIMER_DETAIL, "  Min %lu.%06lus\n", min.tv_sec, min.tv_nsec / 1000);
#endif
    clock_gettime(CLOCK_MONOTONIC, &now);
    if(timespec_compare(&now, &min) == -1) {
        timespec_sub(&sleep, &min, &now);
#ifdef BNGBLASTER_TIMER_LOGGING
        LOG(TIMER_DETAIL, "  Sleep %lu.%06lus\n", sleep.tv_sec, sleep.tv_nsec / 1000);
#endif
        res = nanosleep(&sleep, &rem);
        if(res == -1) {
            switch (errno) {
                case EINTR: /* Ctrl-C */
                    break;
                default:
                    LOG(ERROR, "Timer Error: nanosleep %s (%d)\n", strerror(errno), errno);
                    break;
            }
        }
    }
}

/**
 * Init a timer root.
 *
 * @param root timer root
 */
void
timer_init_root(timer_root_s *timer_root)
{
    CIRCLEQ_INIT(&timer_root->timer_bucket_qhead);
    CIRCLEQ_INIT(&timer_root->timer_gc_qhead);
    CIRCLEQ_INIT(&timer_root->timer_change_qhead);
}

/**
 * Flush all timers hanging off a timer root.
 *
 * @param root timer root
 */
void
timer_flush_root(timer_root_s *timer_root)
{
    timer_s *timer;
    timer_bucket_s *timer_bucket;

    /* First step. Walk all timers and move them onto the GC thread. */
    CIRCLEQ_FOREACH(timer_bucket, &timer_root->timer_bucket_qhead, timer_bucket_qnode) {
        CIRCLEQ_FOREACH(timer, &timer_bucket->timer_qhead, timer_qnode) {
            timer_del(timer);
        }
    }
    timer_process_changes(timer_root);

    /* Second step. Run the GC queue. */
    while(!CIRCLEQ_EMPTY(&timer_root->timer_gc_qhead)) {
        timer = CIRCLEQ_FIRST(&timer_root->timer_gc_qhead);
        CIRCLEQ_REMOVE(&timer_root->timer_gc_qhead, timer, timer_qnode);
        timer_root->gc--;
        free(timer);
    }
}