qq-cli 0.1.0

A modular Rust QQ task interface for coding agents
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

use std::time::{Duration, Instant};

use clap::Parser;
use qq_cli::cli::args::{Cli, Command, MessageCommand};
use qq_cli::messaging::content;
use serde_json::json;

#[test]
fn explicit_sources_preserve_literal_text_and_bound_utf8_input() {
    assert_eq!(
        content::text_from_reader(&b"[CQ:at,qq=all]"[..]).unwrap(),
        "[CQ:at,qq=all]"
    );
    for bytes in [vec![], vec![255], vec![b'x'; 65537]] {
        assert!(content::text_from_reader(bytes.as_slice()).is_err());
    }
    assert_eq!(
        content::text_from_reader(vec![b'x'; 65536].as_slice())
            .unwrap()
            .len(),
        65536
    );
    let temp = tempfile::tempdir().unwrap();
    let file = temp.path().join("input.txt");
    std::fs::write(&file, "synthetic file text").unwrap();
    let cli = Cli::try_parse_from([
        "qq-cli",
        "message",
        "send",
        "--conversation",
        "group:20001",
        "--text-file",
        file.to_str().unwrap(),
        "--mention",
        "10002",
        "--dry-run",
    ])
    .unwrap();
    let Command::Message {
        command: MessageCommand::Send(args),
    } = cli.command
    else {
        panic!()
    };
    let segments = content::prepare(
        &args.content,
        &args.conversation,
        Instant::now() + Duration::from_secs(1),
    )
    .unwrap();
    assert_eq!(
        segments,
        vec![
            json!({"type":"text","data":{"text":"synthetic file text"}}),
            json!({"type":"at","data":{"qq":"10002"}})
        ]
    );
}

#[test]
fn text_input_errors_and_private_mentions_fail_during_content_preparation() {
    for args in [
        vec![
            "qq-cli",
            "message",
            "send",
            "--conversation",
            "group:20001",
            "--image",
            "missing-image.png",
            "--text-file",
            "missing-input.txt",
            "--dry-run",
        ],
        vec![
            "qq-cli",
            "message",
            "send",
            "--conversation",
            "private:10002",
            "--mention",
            "10002",
            "--text-file",
            "missing-input.txt",
            "--dry-run",
        ],
    ] {
        let cli = Cli::try_parse_from(args).unwrap();
        let Command::Message {
            command: MessageCommand::Send(args),
        } = cli.command
        else {
            panic!()
        };
        let error = content::prepare(
            &args.content,
            &args.conversation,
            Instant::now() + Duration::from_secs(1),
        )
        .unwrap_err();
        assert!(matches!(
            error.code,
            "INPUT_INVALID" | "MENTION_UNSUPPORTED"
        ));
    }
}