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

/// Recording OpenLibrary subjects and subject-entity links.
use num_enum::{IntoPrimitive, TryFromPrimitive};
use parse_display::*;

use crate::arrow::row::MappableColType;

use super::source::OLSubjects;

/// The type of a subject relationship.
#[derive(Debug, Clone, FromStr, Display, IntoPrimitive, TryFromPrimitive)]
#[display(style = "kebab-case")]
#[repr(u8)]
pub enum SubjectType {
    General = 0,
    Person,
    Place,
    Time,
}

impl MappableColType for SubjectType {
    type ColumnType = u8;
}

/// Schema for subject linking records.
#[derive(Debug, Clone)]
pub struct SubjectEntry {
    pub entity: i32,
    pub subj_type: SubjectType,
    pub subject: String,
}

impl OLSubjects {
    pub fn subject_records(self, entity: i32) -> Vec<SubjectEntry> {
        let mut records = Vec::new();
        for subject in self.subjects {
            records.push(SubjectEntry {
                entity,
                subj_type: SubjectType::General,
                subject: subject.into(),
            });
        }

        for subject in self.subject_people {
            records.push(SubjectEntry {
                entity,
                subj_type: SubjectType::Person,
                subject: subject.into(),
            });
        }

        for subject in self.subject_places {
            records.push(SubjectEntry {
                entity,
                subj_type: SubjectType::Place,
                subject: subject.into(),
            });
        }

        for subject in self.subject_times {
            records.push(SubjectEntry {
                entity,
                subj_type: SubjectType::Time,
                subject: subject.into(),
            });
        }

        records
    }
}