ligent/src/confirm.c

/* confirm.c - 写操作确认 */
#include "confirm.h"
#include "i18n.h"
#include "tools.h" /* TOOL_* 宏: 工具名只在那里定义一次 */
#include "util.h"

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <unistd.h>

#define BOLD "\033[1m"
#define DIM "\033[2m"
#define YEL "\033[33m"
#define OFF "\033[0m"

static int g_enabled = 1;
static ConfirmDescriber g_describer;
static int g_color;
static int g_turn = CONFIRM_ASK; /* 本轮的提问策略, 每轮复位 */

static const char *c(const char *code) { return g_color ? code : ""; }

/* 截断并压掉换行, 保持在一行内 */
static void preview(const char *s, size_t max, char *out, size_t outlen) {
    if (!s) {
        snprintf(out, outlen, "%s", t("confirm.val.empty"));
        return;
    }
    size_t n = 0;
    for (const char *p = s; *p && n + 1 < outlen; p++) {
        if (*p == '\n' || *p == '\r' || *p == '\t') {
            if (n + 1 < outlen)
                out[n++] = ' ';
            continue;
        }
        out[n++] = *p;
        if (n >= max)
            break;
    }
    out[n] = '\0';
    if (s[n] && n >= max) {
        size_t l = strlen(out);
        if (l + 4 < outlen)
            snprintf(out + l, outlen - l, "...");
    }
}

/* line 集合的渲染在 tools.c 的 tool_lines_label —— 调用摘要 (agent.c)用的是
 * 同一份, 两处对"改的是哪几行"的说法必须一致.  */

/* 把这次调用做什么讲清楚 —— 只说工具名等于没说, 用户没法判断 */
static void describe(const char *name, const JsonValue *args) {
    if (g_describer) {
        char *text = g_describer(name, args);
        if (text) {
            fprintf(stderr, "%s", text);
            free(text);
            return;
        }
    }
    char buf[256], num[32];
    const char *path = json_get_string(args, "path");

    if (strcmp(name, TOOL_RUN_COMMAND) == 0) {
        preview(json_get_string(args, "command"), 200, buf, sizeof buf);
        fprintf(stderr, "    %s %s%s%s\n", t("confirm.lbl.cmd"), c(BOLD), buf,
                c(OFF));
    } else if (strcmp(name, TOOL_WRITE_FILE) == 0) {
        const char *content = json_get_string(args, "content");
        preview(path, 160, buf, sizeof buf);
        snprintf(num, sizeof num, "%zu", content ? strlen(content) : 0);
        fprintf(stderr, "    %s %s%s%s %s\n", t("confirm.lbl.write"), c(BOLD),
                buf, c(OFF), tf("confirm.note.overwrite", "bytes", num, NULL));
    } else if (strcmp(name, TOOL_EDIT_FILE) == 0) {
        char lines[96];
        preview(path, 160, buf, sizeof buf);
        tool_lines_label(args, lines, sizeof lines);
        /* 行号必须打出来: 用户要确认的第一件事就是"改的是哪几行".
         * 没给 line 就是全文件替换 (工具不拒绝它), 摘要里直说"全文件".  */
        const char *scope = lines[0] ? lines : t("tool.call.all_lines");
        if (json_get_bool(args, "replace_all", 0))
            scope = tf("tool.call.scope_all", "scope", scope, NULL);
        fprintf(stderr, "    %s %s\"%s\" [%s]%s\n", t("confirm.lbl.edit"),
                c(BOLD), buf, scope, c(OFF));
        /* 然后摆出**带行号**的新旧对照, 再问 Y/n —— 用户就是照着它决定放不放行.
         * 旧版这里打的是"替换 <旧> -> <新>": 两边各截 60 字符、压成一行, 看不出
         * 到底动了几行、动的是哪几行, 长一点的改动更是只剩两个省略号.  */
        tool_render_diff(stderr, args, "    ", g_color);
    } else if (path) {
        preview(path, 160, buf, sizeof buf);
        fprintf(stderr, "    %s %s%s%s\n", t("confirm.lbl.path"), c(BOLD), buf,
                c(OFF));
    }
}

void confirm_set_describer(ConfirmDescriber fn) { g_describer = fn; }

void confirm_init(int enabled) {
    g_enabled = enabled ? 1 : 0;
    g_color = isatty(STDERR_FILENO) ? 1 : 0;
}

void confirm_set_enabled(int on) { g_enabled = on ? 1 : 0; }
int confirm_enabled(void) { return g_enabled; }

void confirm_reset_turn(void) { g_turn = CONFIRM_ASK; }
int confirm_turn_mode(void) { return g_turn; }

int confirm_tool(const char *name, const JsonValue *args) {
    if (!g_enabled)
        return 1;

    /* 本轮已经表过态就不再打扰. Always/neVer 只作用于这一轮:
     * 下一条用户消息进来时 agent_run 会把策略复位成 ASK.  */
    if (g_turn == CONFIRM_ALWAYS)
        return 1;
    if (g_turn == CONFIRM_NEVER)
        return 0;
    /* 名字都没有就没什么可描述的, 也没有什么"改动"值得拦 —— 交给工具层
     * 去回一句"未知工具". strcmp(NULL, ...) 会直接崩.  */
    if (!name)
        return 1;

    /* 非交互输入: 没有人能回答, 按默认值 (Y)放行.
     * 这也是"默认 Y"的自然推论 —— 问不到就等于不反对.  */
    if (!isatty(STDIN_FILENO))
        return 1;

    /* 提权命令会在参数里明说 elevated, describe() 已经把它打出来了 ——
     * 这里不需要额外的身份标注: 没有常驻提权状态, 一次命令一次身份.  */
    fprintf(stderr, "\n%s  </> %s %s%s\n", c(YEL), t("confirm.lbl.about"),
            name ? name : "?", c(OFF));
    describe(name, args);

    for (;;) {
        fprintf(stderr, "%s", t("confirm.prompt"));
        fflush(stderr);

        char line[128];
        errno = 0;
        if (!fgets(line, sizeof line, stdin)) {
            fputc('\n', stderr);
            /* 这里必须区分两种"读不到":
             *
             *   EINTR      用户按了 Ctrl-C. 他是在说"别问了" —— 必须拒绝.
             *   真 EOF     stdin 关了, 没人能回答 —— 才按默认值放行.
             *
             * 原来两者走同一个分支、都返回 1 (允许), 于是"在确认提示上按
             * Ctrl-C"变成了"批准这次操作". run_command 侥幸自己查了中断
             * 标志才没执行, write_file / edit_file 没查, 文件照写.
             *
             * 返回 0 之后 g_ligent_interrupt 仍是 1, agent_run 会把这一轮
             * 收掉 —— 和 Ctrl-C 在别处的语义 (中断当前请求)保持一致.  */
            if (errno == EINTR)
                return 0;
            return 1;
        }
        char *ans = str_trim(line);
        /* 回车 = Y; A/V 大写小写都收, 也收全词 */
        if (*ans == '\0' || strcmp(ans, "y") == 0 || strcmp(ans, "Y") == 0)
            return 1;
        if (strcmp(ans, "n") == 0 || strcmp(ans, "N") == 0)
            return 0;
        if (strcmp(ans, "a") == 0 || strcmp(ans, "A") == 0 ||
            strcasecmp(ans, "always") == 0) {
            g_turn = CONFIRM_ALWAYS;
            fprintf(stderr, "    %s%s%s\n", c(DIM), t("confirm.always"),
                    c(OFF));
            return 1;
        }
        if (strcmp(ans, "v") == 0 || strcmp(ans, "V") == 0 ||
            strcasecmp(ans, "never") == 0) {
            g_turn = CONFIRM_NEVER;
            fprintf(stderr, "    %s%s%s\n", c(DIM), t("confirm.never"), c(OFF));
            return 0;
        }
        fprintf(stderr, "    %s%s%s\n", c(DIM), t("confirm.bad_input"), c(OFF));
    }
}