cses-cli 0.1.3

CSES CLI is a lightweight tool for using CSES from the command line.
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

use std::io::Write;

use anyhow::Result;
use console::style;

use super::Ui;
use crate::entities::{Scope, ScopeItem, ScopeItemRaw};
use crate::ui::table::{Table, TableAlign, TableCell};
use crate::ui::util::styled_task_status_or_score;
use crate::{service, RP};

pub fn list_courses(ui: &mut Ui<impl RP>) -> Result<()> {
    let courses = service::courses(&mut ui.res)?;

    if courses.is_empty() {
        return Ok(writeln!(ui.term, "No available courses!")?);
    }

    writeln!(ui.term, "Courses: ")?;
    for course in courses {
        writeln!(ui.term, "\n{} ({})", course.name, course.id)?;
        writeln!(ui.term, "  {}", course.description)?;
    }
    Ok(())
}

pub fn list_content(ui: &mut Ui<impl RP>, scope: &Scope) -> Result<()> {
    let scope_content = service::scope_content(&mut ui.res, scope)?;

    if scope_content.sections.is_empty() {
        return Ok(writeln!(
            ui.term,
            "{}",
            match scope {
                Scope::Course(_) => "No course content!",
                Scope::Contest(_) => "No contest content!",
            }
        )?);
    }

    for section in scope_content.sections {
        writeln!(ui.term, "\n{}", style(&section.header).bold())?;
        if let Some(text) = &section.text {
            writeln!(ui.term, "{}", text)?;
        }

        let table = create_item_table(&section.list)?;
        write!(ui.term, "{}", table)?;
    }
    writeln!(ui.term)?;
    Ok(())
}

pub fn create_item_table(list: &[ScopeItemRaw]) -> Result<Table> {
    let mut table = Table::new(vec![0, 0, 0, 0]);

    for scope_item_raw in list {
        let scope_item_as_enum = scope_item_raw.as_enum()?;
        match scope_item_as_enum {
            ScopeItem::Text { id, name, link } => {
                table.add_row(vec![
                    TableCell::from(id).align(TableAlign::Right),
                    TableCell::from(name),
                    TableCell::empty(),
                    TableCell::from(link),
                ]);
            }
            ScopeItem::Link { name, link } => {
                table.add_row(vec![
                    TableCell::empty(),
                    TableCell::from(name),
                    TableCell::empty(),
                    TableCell::from(link),
                ]);
            }
            ScopeItem::Task {
                name,
                id,
                link,
                status,
                score,
            } => {
                table.add_row(vec![
                    TableCell::from(id).align(TableAlign::Right),
                    TableCell::from(name),
                    TableCell::styled(styled_task_status_or_score(status, score))
                        .align(TableAlign::Center),
                    TableCell::from(link),
                ]);
            }
        }
    }
    Ok(table)
}