Revision control
Copy as Markdown
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
use std::collections::BTreeMap;
use crate::frontend::{
EnumBody, EnumVariantBody, ExampleBlock, FeatureBody, FeatureFieldBody, FieldBody,
InlineExampleBlock, ManifestFrontEnd, ObjectBody, Types,
};
use crate::intermediate_representation::{
EnumDef, FeatureDef, FeatureExample, FeatureManifest, ObjectDef, PropDef, TypeRef, VariantDef,
};
impl From<FeatureManifest> for ManifestFrontEnd {
fn from(value: FeatureManifest) -> Self {
let features = merge(&value, |fm| fm.iter_feature_defs().collect(), |f| &f.name);
let objects = merge(&value, |fm| fm.iter_object_defs().collect(), |o| &o.name);
let enums = merge(&value, |fm| fm.iter_enum_defs().collect(), |e| &e.name);
let about = value.about.description_only();
let channels = value.channel.into_iter().collect();
ManifestFrontEnd {
about: Some(about),
version: "1.0.0".to_string(),
channels,
includes: Default::default(),
imports: Default::default(),
features,
legacy_types: None,
types: Types { enums, objects },
}
}
}
fn merge<ListGetter, NameGetter, S, T>(
root: &FeatureManifest,
list_getter: ListGetter,
name_getter: NameGetter,
) -> BTreeMap<String, T>
where
S: Clone,
T: From<S>,
ListGetter: Fn(&FeatureManifest) -> Vec<&S>,
NameGetter: Fn(&S) -> &str,
{
let mut dest: BTreeMap<String, T> = BTreeMap::new();
for s in list_getter(root) {
dest.insert(name_getter(s).to_string(), s.to_owned().into());
}
for fm in root.all_imports.values() {
for s in list_getter(fm) {
dest.insert(name_getter(s).to_string(), s.to_owned().into());
}
}
dest
}
impl From<FeatureDef> for FeatureBody {
fn from(value: FeatureDef) -> Self {
let mut variables = BTreeMap::new();
for f in value.props {
variables.insert(f.name(), f.into());
}
let examples: Vec<_> = value.examples.into_iter().map(ExampleBlock::from).collect();
Self {
metadata: value.metadata,
variables,
default: None,
allow_coenrollment: value.allow_coenrollment,
examples,
}
}
}
impl From<FeatureExample> for ExampleBlock {
fn from(example: FeatureExample) -> Self {
ExampleBlock::Inline(example.into())
}
}
impl From<FeatureExample> for InlineExampleBlock {
fn from(example: FeatureExample) -> Self {
Self {
metadata: example.metadata,
value: example.value,
}
}
}
impl From<ObjectDef> for ObjectBody {
fn from(value: ObjectDef) -> Self {
let mut fields = BTreeMap::new();
for f in value.props {
fields.insert(f.name.clone(), f.into());
}
Self {
description: value.doc,
fields,
}
}
}
impl From<EnumDef> for EnumBody {
fn from(value: EnumDef) -> Self {
let mut variants = BTreeMap::new();
for v in value.variants {
variants.insert(v.name.clone(), v.into());
}
Self {
description: value.doc,
variants,
}
}
}
impl From<VariantDef> for EnumVariantBody {
fn from(value: VariantDef) -> Self {
Self {
description: value.doc,
}
}
}
impl From<PropDef> for FieldBody {
fn from(value: PropDef) -> Self {
Self {
description: value.doc,
variable_type: value.typ.to_string(),
default: Some(value.default),
}
}
}
impl From<PropDef> for FeatureFieldBody {
fn from(value: PropDef) -> Self {
Self {
pref_key: value.pref_key.clone(),
string_alias: value.string_alias.as_ref().map(TypeRef::to_string),
field: value.into(),
}
}
}