tokrs 0.3.0

A CLI tool for counting tokens
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

//! 自定义AppError枚举
//!
//! 为AppError实现Display和Error trait
//!
use std::{error::Error, fmt, path::Path};

#[derive(Debug)]
pub enum AppError {
    Io {
        operation: &'static str,
        path: String,
        source: std::io::Error,
    },
    Corrupted {
        path: String,
        source: serde_json::Error,
    },
    Sqlite {
        path: String,
        source: rusqlite::Error,
    },
}

/// 用于使用端快速的生成 `AppError::Io` 这种错误类型
pub fn io_err(operation: &'static str, path: &Path, err: std::io::Error) -> AppError {
    AppError::Io {
        operation,
        path: path.to_string_lossy().to_string(),
        source: err,
    }
}

/// 用于使用端快速的生成 `AppError::Corrupted` 这种错误类型
pub fn json_err(path: &Path, err: serde_json::Error) -> AppError {
    AppError::Corrupted {
        path: path.to_string_lossy().to_string(),
        source: err,
    }
}

pub fn sqlite_err(path: &Path, err: rusqlite::Error) -> AppError {
    AppError::Sqlite {
        path: path.to_string_lossy().to_string(),
        source: err,
    }
}

impl fmt::Display for AppError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            AppError::Io {
                operation,
                path,
                source,
            } => {
                write!(f, "failed to {} '{}': {}", operation, path, source)
            }
            AppError::Corrupted { path, source } => {
                write!(f, "file '{}' is corrupted: {}", path, source)
            }
            AppError::Sqlite { path, source } => {
                write!(f, "sqlite error on '{}': {}", path, source)
            }
        }
    }
}

impl Error for AppError {
    fn source(&self) -> Option<&(dyn Error + 'static)> {
        match self {
            AppError::Io {
                operation: _,
                path: _,
                source,
            } => Some(source),
            AppError::Corrupted { path: _, source } => Some(source),
            AppError::Sqlite { path: _, source } => Some(source),
        }
    }
}

#[cfg(test)]
#[path = "tests/error_test.rs"]
mod tests;