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

//! Poll struct and associated methods.

use crate::prelude::*;

#[non_exhaustive]
#[derive(Debug)]
/// A struct representing a Poll.
pub struct Poll {
    /// The username of the User that created this submission.
    pub by: String,
    /// The total amount of comments.
    pub comments: u64,
    /// The unique id of this submission.
    pub id: u64,
    pub(crate) kids: Vec<u64>,
    pub(crate) parts: Vec<u64>,
    /// The score of this submission.
    pub score: i64,
    /// The text of this submission.
    pub text: String,
    /// When this submission was made, as a Unix timestamp.
    pub time: u64,
    /// The title of this submission.
    pub title: String,
}

impl Poll {
    /// Returns the [User] that created this poll.
    pub fn by(&self, client: &Client) -> Result<User, HError> {
        client.get_user(&self.by)
    }

    /// Returns the top-level [Comments](Comment) of the poll.
    pub fn comments(&self, client: &Client) -> Result<Vec<Comment>, HError> {
        self.kids
            .iter()
            .map(|kid| client.get_comment(*kid))
            .collect()
    }

    /// Returns the [Poll Options](PollOption) of this poll.
    pub fn options(&self, client: &Client) -> Result<Vec<PollOption>, HError> {
        self.parts
            .iter()
            .map(|kid| client.get_poll_option(*kid))
            .collect()
    }
}

#[non_exhaustive]
#[derive(Debug)]
/// A poll option belonging to a poll.
pub struct PollOption {
    /// Username of the [User] that created this poll option.
    pub by: String,
    /// Unique id of this poll option.
    pub id: u64,
    /// The [Poll] it belongs to.
    pub poll: u64,
    /// The score of this poll option.
    pub score: i64,
    /// The text of this poll option.
    pub text: String,
    /// When this option was made, as a Unix timestamp.
    pub time: u64,
}

impl PollOption {
    /// Return the [User] that created this poll option.
    pub fn by(&self, client: &Client) -> Result<User, HError> {
        client.get_user(&self.by)
    }

    /// Return the [Poll] this option belongs to.
    pub fn poll(&self, client: &Client) -> Result<Poll, HError> {
        client.get_poll(self.poll)
    }
}