第29章 状态机和更稳的命令解析

#include <ctype.h>   // isspace 判断空白字符
#include <stdio.h>   // printf 打印结果

#define MAX_ARGS 8   // 最多拆出 8 个参数

// 状态机的三种状态：不在参数中 / 在普通参数中 / 在引号参数中
enum State {
    OUTSIDE,   // 还没进入参数，遇到空格跳过
    IN_WORD,   // 正在读一个空格分隔的参数
    IN_QUOTE   // 正在读双引号内的参数，空格不算分隔符
};

// 把一行命令拆成参数数组，返回参数个数；出错返回负数
int split_command(char *line, char *args[], int max_args)
{
    enum State state = OUTSIDE;   // 初始状态：不在参数中
    int argc = 0;                 // 当前已收集的参数个数

    // 逐个字符扫描整行
    for (char *p = line; *p != '\0'; p++) {
        if (state == OUTSIDE) {
            if (isspace((unsigned char)*p)) {
                continue;         // 空白字符：继续跳过
            }

            if (argc >= max_args) {
                return -1;        // 参数数量超上限
            }

            if (*p == '"') {
                // 遇到左双引号：参数从引号后一个字符开始
                args[argc++] = p + 1;
                state = IN_QUOTE;   // 进入引号状态
            } else {
                // 普通字符：参数从这里开始
                args[argc++] = p;
                state = IN_WORD;    // 进入单词状态
            }
        } else if (state == IN_WORD) {
            if (isspace((unsigned char)*p)) {
                *p = '\0';          // 写入 \0 切断当前参数
                state = OUTSIDE;    // 退出单词状态，回到空闲
            }
        } else if (state == IN_QUOTE) {
            if (*p == '"') {
                *p = '\0';          // 右引号改写为 \0，切断当前参数
                state = OUTSIDE;    // 退出引号状态，回到空闲
            }
        }
    }

    if (state == IN_QUOTE) {
        return -2;   // 扫描结束仍在引号中：引号未闭合
    }

    return argc;   // 返回拆出的参数个数
}

void print_args(char *args[], int argc)
{
    for (int i = 0; i < argc; i++) {
        printf("args[%d] = <%s>\n", i, args[i]);
    }
}

int main(void)
{
    // 测试输入：双引号包裹的姓名
    char line[] = "INSERT 5 \"Alice Wang\" 92";
    char *args[MAX_ARGS];

    int argc = split_command(line, args, MAX_ARGS);
    if (argc < 0) {
        printf("Parse failed, error code: %d\n", argc);
        return 1;
    }

    print_args(args, argc);
    return 0;
}

char buf[MAX_LINE];       // 可修改的缓冲区，split_command 会往里写 \0
char *args[MAX_ARGS];     // 参数指针数组

snprintf(buf, sizeof(buf), "%s", line);   // 把输入行复制到 buf，原 line 可能是只读的

int argc = split_command(buf, args, MAX_ARGS);  // 用状态机拆词
if (argc == -1) {
    printf("Too many arguments\n");
    return 1;
}
if (argc == -2) {
    printf("Unclosed quote\n");
    return 1;
}
if (argc == 0) {          // 空行或全空白字符
    return 1;
}

uppercase(args[0]);       // 命令名转大写，统一比较

if (strcmp(args[0], "INSERT") == 0) {
    if (argc != 4) {      // INSERT 固定 4 参数：命令名 + id + name + score
        printf("Usage: INSERT <id> <name> <score>\n");
        return 1;
    }

    int id;
    int score;

    // 解析整数 id（args[1]）和 score（args[3]），args[2] 是名字
    if (!parse_int(args[1], &id) || !parse_int(args[3], &score)) {
        printf("Usage: INSERT <id> <name> <score>\n");
        return 1;
    }

    struct Student s = {id, "", score};            // 构造学生记录
    snprintf(s.name, sizeof(s.name), "%s", args[2]); // 复制名字（引号已被 \0 替代）

    if (db_insert_logged(db, logfile, s)) {
        printf("Insert succeeded\n");
    } else {
        printf("Insert failed\n");
    }
    return 1;
}