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

use std::collections::HashMap;

use crate::judger::log::Log;

use super::LifeStatus::{*};
use super::RespBoxesMut;

pub fn make_votelist(list: &Vec<&mut Box<dyn super::Responder>>) -> Vec<(usize, String)> {
    list.iter().map(|x| {
        (x.get_id(), x.name())
    })
    .collect()
}


/// 进行一次投票。结果格式为 (投票详情字符串，选票指向的玩家的 id)。
fn get_votes(
    voters: &mut RespBoxesMut, 
    list: &Vec<(usize, String)>, 
    msg: String
) -> Vec<(String, usize)> {
    voters.iter_mut()
        .map(|x| x.vote(&msg, list.clone()))
        .collect()
}

/// 进行计数，返回的是 `Vec<(编号, 数量)>`。
fn count_votes(list: Vec<usize>) -> Vec<(usize, usize)> {
    let mut count = HashMap::new();
    list.iter().for_each(|x| { count
        .entry(*x)
        .and_modify(|x| *x += 1)
        .or_insert(1);
    });
    count.into_iter().collect()
}

fn max_votes(count: Vec<(usize, usize)>) -> Vec<usize> {
    let max_num = count.iter().map(|x| x.1).max().unwrap();
    count.into_iter()
        .filter_map(|x| {
            if x.1 == max_num { Some(x.0) }
            else { None }
        })
        .collect()
}

/// chat 内筛选死活。
pub fn chat(chater: RespBoxesMut, log: &mut Log) {
    let (chater, mut dead): (Vec<_>, Vec<_>) = chater.into_iter()
        .partition(|x| x.status() == Alive);
    let mut succ = chater;
    let mut prev: RespBoxesMut = Vec::new();
    succ.iter_mut().for_each(|x| x.send_begin());
    while !succ.is_empty() {
        let mut it = succ.into_iter();
        let speaking = it.next().unwrap();
        speaking.send_begin();
        let words = format!("{}：{}", speaking.name(), speaking.rec_text());
        succ = it.collect();
        log.write(&words);
        dead.iter_mut().for_each(|x| x.send_msg(&words));
        prev.iter_mut().for_each(|x| x.send_msg(&words));
        succ.iter_mut().for_each(|x| x.send_msg(&words));
        prev.push(speaking);
    }
    prev.into_iter().for_each(|x| x.send_end());
}

fn send_detail(voters: &mut RespBoxesMut, s: String) {
    voters.iter_mut().for_each(|x| x.send_msg(&s));
}

/// 投票。如果出现平票，则递归调用自己再次投票。投票结果返回的是用户的 id。函数内不筛选死活。
pub fn vote(
    voters: &mut RespBoxesMut, 
    list: Vec<(usize, String)>, 
    msg: String,
    log: &mut Log
) -> usize {
    if list.len() == 1 {
        voters.iter_mut().for_each(|x| x.send_end());
        return list[0].0;
    }
    let (detail_str, res): (Vec<_>, Vec<_>) = 
        get_votes(voters, &list, msg).into_iter().unzip();
    let detail = detail_str.join("\n") + "\n";
    send_detail(voters, detail.clone());
    log.write("投票详情:");
    log.write(&detail);
    let count_list = count_votes(res);
    let again_list = max_votes(count_list);
    let cur_res: Vec<_> = list.into_iter()
        .filter(|x| again_list.contains(&x.0))
        .collect();
    vote(voters, cur_res, "请在平票玩家中再次投票。".to_string(), log)
}