world-count 0.1.4

use aoko::no_std::{functions::ext::{Utf8Ext, AnyExt1}, pipelines::pipe::Pipe};
use itertools::Itertools;
use std::fs;

// prefix punctuation
const PP: [char; 6] = ['"', '\'', '(', '[', '{', '<'];

// suffix punctuation
const SP: [char; 12] = [',', '.', '!', '"', '\'', ':', ';', ')', ']', '}', '>', '-'];

pub fn batch_en(dict: String, r#in: String, out: String) {
    // 读文件
    fs::read(r#in).unwrap()
        // 转小写
        .to_ascii_lowercase()
        // 转字符串
        .to_str().unwrap()
        // 判断是否有字典 —— (1. 如果没有)
        .pipe(|s| if let None = dict.chars().next() {
            // 按空格切割
            s.split_whitespace()
            // 映射 + 过滤
            .filter_map(|s| s

                // 去前缀
                .trim_start_matches(|c| PP.contains(&c))
                // 去后缀
                .trim_end_matches(|c| SP.contains(&c))
                // 去空串，计数`1`
                .if_not_then(|s| s.chars().next().is_none(), |s| (s, 1)))
            // 按类分组(小写)
            .into_grouping_map_by(|t| t.0) // Map { "word": [("word", 1), ...], ... }
            // 聚合:   累计             K          V  (K, 计数)
            .fold(0, |acc, _, (_, count)| acc + count) // Map { "word": count, ... }
            // 迭代
            .into_iter()
            // 排序
            .sorted_by_key(|t| t.1)
        // 判断是否有字典 —— (2. 如果有)
        } else {
            // 读字典
            fs::read(dict).unwrap()
                // 转小写
                .to_ascii_lowercase()
                // 转字符串
                .to_str().unwrap()
                // 按行读(一行一词组)
                .lines()
                // 映射，[("word": count), ...]
                .flat_map(|phrase| s.matches(phrase).counts())
                // 排序
                .sorted_by_key(|t| t.1)
        })
            // 逆序(从大到小)(高频到低频)
            .rev()
            // 映射，转字符串
            .map(|t| format!("{:?}", t))
            // 合并字符串加换行
            .join("\n")
            // 输出文件
            .pipe(|byt| fs::write(out, byt).unwrap())
}