libraries/rawkit/rawkit-proc-macros/src/build_camera_data.rs

use proc_macro::TokenStream;
use quote::{ToTokens, quote};
use std::fs;
use std::path::Path;
use toml::{Table, Value};

enum CustomValue {
	String(String),
	Integer(i64),
	Float(f64),
	Boolean(bool),
	Array(Vec<CustomValue>),
}

impl ToTokens for CustomValue {
	fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {
		match self {
			CustomValue::String(x) => x.to_tokens(tokens),
			CustomValue::Integer(x) => {
				let x: proc_macro2::TokenStream = format!("{:?}", x).parse().unwrap();
				x.to_tokens(tokens)
			}
			CustomValue::Float(x) => {
				let x: proc_macro2::TokenStream = format!("{:?}", x).parse().unwrap();
				x.to_tokens(tokens)
			}
			CustomValue::Boolean(x) => x.to_tokens(tokens),
			CustomValue::Array(x) => quote! { [ #( #x ),* ] }.to_tokens(tokens),
		}
	}
}

impl From<Value> for CustomValue {
	fn from(value: Value) -> Self {
		match value {
			Value::String(x) => CustomValue::String(x),
			Value::Integer(x) => CustomValue::Integer(x),
			Value::Float(x) => CustomValue::Float(x),
			Value::Boolean(x) => CustomValue::Boolean(x),
			Value::Array(x) => CustomValue::Array(x.into_iter().map(|x| x.into()).collect()),
			_ => panic!("Unsupported data type"),
		}
	}
}

pub fn build_camera_data() -> TokenStream {
	let mut camera_data: Vec<(String, Table)> = Vec::new();

	let mut path = Path::new(&std::env::var("CARGO_MANIFEST_DIR").unwrap()).to_path_buf();
	path.push("camera_data");

	fs::read_dir(path).unwrap().for_each(|entry| {
		let company_name_path = entry.unwrap().path();
		if !company_name_path.is_dir() {
			panic!("camera_data should only contain folders of company names")
		}

		let company_name = company_name_path.file_name().unwrap().to_str().unwrap().to_string();

		fs::read_dir(company_name_path).unwrap().for_each(|entry| {
			let model_path = entry.unwrap().path();
			if !model_path.is_file() || model_path.extension().unwrap() != "toml" {
				panic!("The folders within camera_data should only contain toml files")
			}

			let name = company_name.clone() + " " + model_path.file_stem().unwrap().to_str().unwrap();

			let mut values: Table = toml::from_str(&fs::read_to_string(model_path).unwrap()).unwrap();

			if let Some(val) = values.get_mut("xyz_to_camera") {
				*val = Value::Array(val.as_array().unwrap().iter().map(|x| Value::Integer((x.as_float().unwrap() * 10_000.) as i64)).collect());
			}

			camera_data.push((name, values))
		});
	});

	let x: Vec<_> = camera_data
		.iter()
		.map(|(name, camera_data)| {
			let keys: Vec<_> = camera_data.keys().map(|key| syn::Ident::new(key, proc_macro2::Span::call_site())).collect();
			let values: Vec<CustomValue> = camera_data.values().cloned().map(|x| x.into()).collect();

			quote! {
				(
					#name,
					CameraData {
						#( #keys: #values, )*
						..CameraData::DEFAULT
					}
				)
			}
		})
		.collect();

	quote!([ #(#x),* ]).into()
}