r/src/spawn_tracker.c

// retoor <retoor@molodetz.nl>

#include "spawn_tracker.h"
#include "r_config.h"
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

struct spawn_tracker_t {
    int current_depth;
    int total_spawns;
    int max_depth;
    int max_total;
    pthread_mutex_t lock;
};

static struct spawn_tracker_t *instance = NULL;

spawn_tracker_handle spawn_tracker_get_instance(void) {
    if (instance) return instance;

    instance = calloc(1, sizeof(struct spawn_tracker_t));
    if (!instance) return NULL;

    r_config_handle cfg = r_config_get_instance();
    instance->max_depth = r_config_get_max_spawn_depth(cfg);
    instance->max_total = r_config_get_max_total_spawns(cfg);
    instance->current_depth = 0;
    instance->total_spawns = 0;
    pthread_mutex_init(&instance->lock, NULL);

    return instance;
}

void spawn_tracker_destroy(void) {
    if (!instance) return;
    pthread_mutex_destroy(&instance->lock);
    free(instance);
    instance = NULL;
}

bool spawn_tracker_can_spawn(spawn_tracker_handle tracker) {
    if (!tracker) return false;

    pthread_mutex_lock(&tracker->lock);
    bool allowed = (tracker->current_depth < tracker->max_depth) &&
                   (tracker->total_spawns < tracker->max_total);
    pthread_mutex_unlock(&tracker->lock);

    return allowed;
}

int spawn_tracker_begin(spawn_tracker_handle tracker) {
    if (!tracker) return -1;

    pthread_mutex_lock(&tracker->lock);
    tracker->current_depth++;
    tracker->total_spawns++;
    int depth_token = tracker->current_depth;
    fprintf(stderr, "[SpawnTracker] depth=%d/%d total=%d/%d\n",
            tracker->current_depth, tracker->max_depth,
            tracker->total_spawns, tracker->max_total);
    pthread_mutex_unlock(&tracker->lock);

    return depth_token;
}

void spawn_tracker_end(spawn_tracker_handle tracker, int depth_token) {
    if (!tracker) return;
    (void)depth_token;

    pthread_mutex_lock(&tracker->lock);
    if (tracker->current_depth > 0) {
        tracker->current_depth--;
    }
    fprintf(stderr, "[SpawnTracker] agent finished, depth=%d/%d total=%d/%d\n",
            tracker->current_depth, tracker->max_depth,
            tracker->total_spawns, tracker->max_total);
    pthread_mutex_unlock(&tracker->lock);
}

bool spawn_tracker_validate_result(const char *result) {
    if (!result) return false;
    size_t len = strlen(result);
    if (len < SPAWN_RESULT_MIN_LENGTH) return false;
    if (strstr(result, "Error:") == result) return false;
    return true;
}