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
|
// -*- coding: utf-8 -*-
//
// Copyright (C) 2024 Michael Büsch <m@bues.ch>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
//
// SPDX-License-Identifier: GPL-2.0-or-later
#![forbid(unsafe_code)]
mod command;
use crate::command::{list::command_list, seen::command_seen};
use anyhow::{self as ah, Context as _};
use clap::{Parser, Subcommand};
use feedsdb::Db;
use std::{num::NonZeroUsize, sync::Arc, time::Duration};
use tokio::runtime;
#[derive(Parser, Debug, Clone)]
struct Opts {
/// The name of the database to use.
#[arg(long, default_value = "feeds")]
db: String,
/// Set the number async worker threads.
#[arg(long, default_value = "2")]
worker_threads: NonZeroUsize,
#[command(subcommand)]
command: Command,
}
#[derive(Subcommand, Debug, Clone)]
enum Command {
/// List all feeds from the database.
List,
/// Mark items as "seen".
Seen {
/// The feed ID to set to "seen".
/// Or alternatively "all" to set all feeds to seen.
id: String,
},
}
async fn async_main(opts: Opts) -> ah::Result<()> {
let opts = Arc::new(opts);
let db = Db::new(&opts.db).await.context("Database")?;
match &opts.command {
Command::List => command_list(&db).await,
Command::Seen { id } => command_seen(&db, id).await,
}
}
fn main() -> ah::Result<()> {
let opts = Opts::parse();
runtime::Builder::new_multi_thread()
.worker_threads(opts.worker_threads.into())
.max_blocking_threads(opts.worker_threads.into()) // one blocking per worker.
.thread_keep_alive(Duration::from_secs(1))
.enable_all()
.build()
.context("Tokio runtime builder")?
.block_on(async_main(opts))
}
// vim: ts=4 sw=4 expandtab
|
