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

/*
Copyright (C) 2022 Kunal Mehta <legoktm@debian.org>

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 3 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 <http://www.gnu.org/licenses/>.
 */

use crate::config::{Auth, Config};

/// Build a `mwbot::Bot` instance programmatically
pub struct Builder {
    config: Config,
}

impl Builder {
    pub(crate) fn new(api_url: String, rest_url: String) -> Self {
        Self {
            config: Config {
                api_url,
                rest_url,
                auth: None,
                general: Default::default(),
                edit: Default::default(),
            },
        }
    }

    pub fn set_botpassword(
        mut self,
        username: String,
        password: String,
    ) -> Self {
        self.config.auth = Some(Auth::BotPassword { username, password });
        self
    }

    pub fn set_oauth2_token(mut self, username: String, token: String) -> Self {
        self.config.auth = Some(Auth::OAuth2 {
            username,
            oauth2_token: token,
        });
        self
    }

    pub fn set_maxlag(mut self, maxlag: u32) -> Self {
        self.config.general.maxlag = Some(maxlag);
        self
    }

    pub fn set_retry_limit(mut self, retry_limit: u32) -> Self {
        self.config.general.retry_limit = Some(retry_limit);
        self
    }

    pub fn set_mark_as_bot(mut self, mark_as_bot: bool) -> Self {
        self.config.edit.mark_as_bot = Some(mark_as_bot);
        self
    }

    pub fn set_save_delay(mut self, save_delay: u64) -> Self {
        self.config.edit.save_delay = Some(save_delay);
        self
    }

    pub fn set_respect_nobots(mut self, respect_nobots: bool) -> Self {
        self.config.edit.respect_nobots = Some(respect_nobots);
        self
    }

    pub async fn build(self) -> Result<crate::Bot, crate::ConfigError> {
        crate::Bot::from_config(self.config).await
    }
}