aboutsummaryrefslogtreecommitdiff
path: root/common/object/src/registry.rs
blob: d9da2fb01bc9783825020df6f61d7bd13abd9d5c (plain)
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
/*
    This file is part of jellything (https://codeberg.org/metamuffin/jellything)
    which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
    Copyright (C) 2026 metamuffin <metamuffin.org>
*/

use crate::Tag;
use std::{any::TypeId, collections::BTreeMap};

pub mod types {
    use crate::Object;
    use std::any::TypeId;

    pub const OBJECT: TypeId = TypeId::of::<Object>();
    pub const STR: TypeId = TypeId::of::<&str>();
    pub const U32: TypeId = TypeId::of::<u32>();
    pub const U64: TypeId = TypeId::of::<u64>();
}

#[derive(Default)]
pub struct Registry {
    tags: BTreeMap<Tag, TagInfo>,
}
impl Registry {
    pub fn add(&mut self, tag: Tag, info: TagInfo) {
        if let Some(other) = self.tags.get(&tag) {
            panic!(
                "Conflicting names for tag {}: {:?} vs {:?}",
                tag.0, info.name, other.name
            )
        }
        self.tags.insert(tag, info);
    }
    pub fn info(&self, tag: Tag) -> Option<&TagInfo> {
        self.tags.get(&tag)
    }
    pub fn name(&self, tag: Tag) -> &str {
        match self.tags.get(&tag) {
            Some(inf) => inf.name,
            None => "unknown",
        }
    }
}

pub struct TagInfo {
    pub name: &'static str,
    pub r#type: Option<TypeId>,
}

#[macro_export]
macro_rules! fields {
    ($($id:ident: $type:ty = $tag:literal $name:literal;)*) => {
        $(pub const $id: $crate::TypedTag<$type> = $crate::TypedTag($crate::Tag($tag), std::marker::PhantomData);)*
        pub(crate) fn register_fields(reg: &mut $crate::Registry) {
            $(reg.add($crate::Tag($tag), $crate::TagInfo { name: $name, r#type: Some(std::any::TypeId::of::<$type>()) });)*
        }
    };
}
#[macro_export]
macro_rules! enums {
    ($($id:ident = $tag:literal $name:literal;)*) => {
        $(pub const $id: $crate::Tag = $crate::Tag($tag);)*
        pub(crate) fn register_enums(reg: &mut $crate::Registry) {
            $(reg.add($crate::Tag($tag), $crate::TagInfo { name: $name, r#type: None });)*
        }
    };
}