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

//! Validate cached message receipts before reporting acceptance from disk.
use crate::core::domain::{Conversation, identifier};
use crate::core::error::{AppError, Result};
use serde_json::Value;

pub fn validate(value: &Value, expected: Option<&Conversation>) -> Result<()> {
    let fields = value.as_object().ok_or_else(invalid)?;
    if fields.keys().any(|key| {
        !matches!(
            key.as_str(),
            "messageId" | "conversation" | "attachments" | "fileTransfer"
        )
    }) {
        return Err(invalid());
    }
    let id = value
        .get("messageId")
        .filter(|v| v.is_string())
        .and_then(message_id)
        .ok_or_else(invalid)?;
    if id.is_empty() {
        return Err(invalid());
    }
    let target = value
        .get("conversation")
        .and_then(Value::as_str)
        .ok_or_else(invalid)?
        .parse::<Conversation>()
        .map_err(|_| invalid())?;
    if expected.is_some_and(|expected| expected != &target) {
        return Err(invalid());
    }
    match value.get("attachments") {
        None if value.get("fileTransfer").is_none() => (),
        Some(Value::Array(files))
            if !files.is_empty()
                && files.len() <= 16
                && value["fileTransfer"] == "not_observable" =>
        {
            for file in files {
                attachment(file)?;
            }
        }
        _ => return Err(invalid()),
    }
    Ok(())
}

pub fn message_id(value: &Value) -> Option<String> {
    let id = identifier(value)?;
    let canonical = if id.starts_with('-') {
        let value = id.parse::<i64>().ok()?;
        if value == 0 {
            return None;
        }
        value.to_string()
    } else {
        let value = id.parse::<u64>().ok()?;
        if value == 0 {
            return None;
        }
        value.to_string()
    };
    (id == canonical).then_some(id)
}

pub fn attachment(value: &Value) -> Result<()> {
    let fields = value.as_object().ok_or_else(invalid)?;
    if fields.len() != 4 || !matches!(value["kind"].as_str(), Some("image" | "file")) {
        return Err(invalid());
    }
    let name = value["name"].as_str().ok_or_else(invalid)?;
    if std::path::Path::new(name).components().count() != 1 {
        return Err(invalid());
    }
    crate::napcat::install::archive::member_path(name).map_err(|_| invalid())?;
    value["sizeBytes"]
        .as_u64()
        .filter(|size| *size > 0 && *size <= crate::messaging::media::MAX_DOWNLOAD_BYTES)
        .ok_or_else(invalid)?;
    let hash = value["sha256"].as_str().ok_or_else(invalid)?;
    if hash.len() != 64
        || !hash
            .bytes()
            .all(|byte| byte.is_ascii_hexdigit() && !byte.is_ascii_uppercase())
    {
        return Err(invalid());
    }
    Ok(())
}

fn invalid() -> AppError {
    AppError::new(
        "REQUEST_RECEIPT_INVALID",
        "Stored message receipt is invalid; inspect the request state without sending again.",
        6,
    )
}