Local appearance and material libraries (#863)

Adds local scalar appearance and material libraries: a catalog in the io crate with safe writes, engine commands to list and apply library entries, and a library dialog in the UI, with docs.
This commit is contained in:
selune authored and GitHub committed 2026-10-14 12:21:35 -04:00
1 parent 8ea1fd643f
commit d399effbf7
21 files changed
+1285 -2

No files matched your search

+285
View File
@@ -0,0 +1,285 @@
//! Portable scalar libraries; no renderer, network, external resources or design links.
use crate::{IoError, Result};
use serde::{Deserialize, Serialize};
use sha2::{Digest, Sha256};
use solidcraft_doc::{Document, UserMaterial, appearance::Look, valid_density};
use std::collections::BTreeMap;
pub const FORMAT: &str = "solidcraft-library/1";
pub const MAX_BYTES: usize = 4 * 1024 * 1024;
pub const MAX_ENTRIES: usize = 1024;
/// Bound source inspection even when many assignments reuse the same definition.
pub const MAX_SOURCE_REFERENCES: usize = 100_000;
fn invalid(message: impl Into<String>) -> IoError {
IoError::Invalid(message.into())
}
fn text(value: &str, max: usize) -> bool {
!value.is_empty() && value == value.trim() && value.len() <= max && !value.chars().any(char::is_control)
}
fn identifier(value: &str) -> bool {
value.len() <= 64
&& value.as_bytes().first().is_some_and(u8::is_ascii_lowercase)
&& value.bytes().all(|b| b.is_ascii_lowercase() || b.is_ascii_digit() || b"._-".contains(&b))
}
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct Provenance {
pub license: String,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub author: Option<String>,
/// Information only: never fetched or treated as a local resource path.
#[serde(default, skip_serializing_if = "Option::is_none")]
pub source_url: Option<String>,
}
impl Default for Provenance {
fn default() -> Self {
Self { license: "LicenseRef-User-Provided".into(), author: None, source_url: None }
}
}
impl Provenance {
fn validate(&self) -> Result<()> {
if !text(&self.license, 128)
|| self.author.as_ref().is_some_and(|s| !text(s, 256))
|| self.source_url.as_ref().is_some_and(|s| !text(s, 2048))
{
return Err(invalid("invalid library provenance (license 128, author 256, source URL 2048 bytes)"));
}
Ok(())
}
}
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct ScalarLook {
pub name: String,
pub color: [u8; 3],
pub opacity: f64,
}
impl ScalarLook {
pub fn from_look(look: &Look) -> Self {
Self { name: look.name.clone(), color: look.color, opacity: look.opacity }
}
pub fn to_look(&self) -> Look {
Look { name: self.name.clone(), color: self.color, opacity: self.opacity }
}
fn validate(&self) -> Result<()> {
if !text(&self.name, 128) || !self.opacity.is_finite() || !(0.0..=1.0).contains(&self.opacity) {
return Err(invalid("library appearance needs a 1–128 byte name and finite opacity 0–1"));
}
Ok(())
}
}
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct ScalarMaterial {
pub name: String,
pub density_g_cm3: f64,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub appearance: Option<ScalarLook>,
}
impl ScalarMaterial {
pub fn from_material(material: &UserMaterial) -> Self {
Self { name: material.name.clone(), density_g_cm3: material.density, appearance: material.appearance.as_ref().map(ScalarLook::from_look) }
}
pub fn to_material(&self, name: String) -> UserMaterial {
UserMaterial { name, density: self.density_g_cm3, appearance: self.appearance.as_ref().map(ScalarLook::to_look) }
}
fn validate(&self) -> Result<()> {
if !text(&self.name, 128) || !valid_density(self.density_g_cm3) {
return Err(invalid("invalid library material name or density"));
}
if let Some(look) = &self.appearance {
look.validate()?;
}
Ok(())
}
}
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
#[serde(tag = "kind", content = "value", rename_all = "snake_case", deny_unknown_fields)]
pub enum Definition {
Appearance(ScalarLook),
Material(ScalarMaterial),
}
impl Definition {
pub fn name(&self) -> &str {
match self {
Self::Appearance(l) => &l.name,
Self::Material(m) => &m.name,
}
}
fn validate(&self) -> Result<()> {
match self {
Self::Appearance(l) => l.validate(),
Self::Material(m) => m.validate(),
}
}
fn kind(&self) -> &str {
match self {
Self::Appearance(_) => "appearance",
Self::Material(_) => "material",
}
}
}
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct Entry {
pub id: String,
pub revision: String,
pub provenance: Provenance,
pub definition: Definition,
}
impl Entry {
/// Revision is SHA-256 of compact serde JSON `[provenance, definition]` in this v1 contract.
pub fn content_revision(&self) -> Result<String> {
self.provenance.validate()?;
self.definition.validate()?;
let bytes = serde_json::to_vec(&(&self.provenance, &self.definition)).map_err(|e| invalid(e.to_string()))?;
Ok(format!("{:x}", Sha256::digest(bytes)))
}
pub fn new(pack_id: &str, definition: Definition, provenance: Provenance) -> Result<Self> {
if !identifier(pack_id) {
return Err(invalid("invalid library identifier"));
}
let mut out = Self { id: String::new(), revision: String::new(), provenance, definition };
out.revision = out.content_revision()?;
let prefix = out.revision.get(..16).ok_or_else(|| invalid("invalid library revision"))?;
out.id = format!("{pack_id}/{}-{prefix}", out.definition.kind());
Ok(out)
}
fn validate(&self, pack_id: &str) -> Result<()> {
let (pack, local) = self.id.split_once('/').ok_or_else(|| invalid("entry ID must be qualified by its library ID"))?;
if pack != pack_id || !identifier(local) || self.revision != self.content_revision()? {
return Err(invalid("invalid entry identity or content revision"));
}
Ok(())
}
}
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct Catalog {
pub format: String,
pub id: String,
pub name: String,
#[serde(deserialize_with = "bounded_entries")]
pub entries: Vec<Entry>,
}
fn bounded_entries<'de, D: serde::Deserializer<'de>>(deserializer: D) -> std::result::Result<Vec<Entry>, D::Error> {
struct Entries;
impl<'de> serde::de::Visitor<'de> for Entries {
type Value = Vec<Entry>;
fn expecting(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
formatter.write_str("at most 1024 scalar library entries")
}
fn visit_seq<A: serde::de::SeqAccess<'de>>(self, mut sequence: A) -> std::result::Result<Self::Value, A::Error> {
let mut entries = Vec::new();
loop {
if entries.len() == MAX_ENTRIES {
if sequence.next_element::<serde::de::IgnoredAny>()?.is_some() {
return Err(serde::de::Error::custom("scalar library exceeds 1024 entries"));
}
break;
}
let Some(entry) = sequence.next_element::<Entry>()? else { break };
let revision = entry.content_revision().map_err(serde::de::Error::custom)?;
if entry.revision != revision || entry.id.len() > 129 {
return Err(serde::de::Error::custom("invalid scalar library revision or identity"));
}
entries.try_reserve(1).map_err(|_| serde::de::Error::custom("could not reserve scalar library entries"))?;
entries.push(entry);
}
Ok(entries)
}
}
deserializer.deserialize_seq(Entries)
}
impl Catalog {
pub fn validate(&self) -> Result<()> {
if self.format != FORMAT || !identifier(&self.id) || !text(&self.name, 128) || self.entries.len() > MAX_ENTRIES {
return Err(invalid("unsupported or invalid scalar library (v1, 1024 entries, 128 byte name)"));
}
let mut ids = std::collections::BTreeSet::new();
for entry in &self.entries {
entry.validate(&self.id)?;
if !ids.insert(&entry.id) {
return Err(invalid("duplicate library entry identity"));
}
}
Ok(())
}
pub fn decode(bytes: &[u8]) -> Result<Self> {
if bytes.len() > MAX_BYTES {
return Err(invalid("scalar library exceeds 4 MiB"));
}
let catalog: Self = serde_json::from_slice(bytes).map_err(|e| invalid(format!("invalid scalar library: {e}")))?;
catalog.validate()?;
Ok(catalog)
}
pub fn encode(&self) -> Result<Vec<u8>> {
self.validate()?;
let bytes = serde_json::to_vec_pretty(self).map_err(|e| invalid(e.to_string()))?;
if bytes.len() > MAX_BYTES {
return Err(invalid("scalar library exceeds 4 MiB"));
}
Ok(bytes)
}
pub fn read(path: &str) -> Result<Self> {
// Check the VFS size first as well: its browser read clones the stored bytes.
if crate::vfs::len(path).map_err(|e| invalid(e.to_string()))? > MAX_BYTES as u64 {
return Err(invalid("scalar library exceeds 4 MiB"));
}
Self::decode(&crate::vfs::read(path, MAX_BYTES).map_err(|e| invalid(e.to_string()))?)
}
pub fn from_document(doc: &Document, id: &str, name: &str, provenance: Provenance) -> Result<Self> {
if !identifier(id) || !text(name, 128) {
return Err(invalid("invalid scalar library identifier or name"));
}
provenance.validate()?;
let references = doc
.appearances
.bodies
.len()
.saturating_add(doc.appearances.components.len())
.saturating_add(doc.appearances.faces.len())
.saturating_add(doc.user_materials.len().saturating_mul(2));
if references > MAX_SOURCE_REFERENCES {
return Err(invalid("library export exceeds 100000 source references"));
}
let mut entries = BTreeMap::new();
let mut insert = |definition| -> Result<()> {
let entry = Entry::new(id, definition, provenance.clone())?;
if let Some(previous) = entries.get(&entry.id) {
if previous != &entry {
return Err(invalid("library content identifier collision"));
}
} else {
if entries.len() >= MAX_ENTRIES {
return Err(invalid("scalar library exceeds 1024 unique entries"));
}
entries.insert(entry.id.clone(), entry);
}
Ok(())
};
for look in doc.appearances.bodies.values().chain(doc.appearances.components.values()).chain(doc.appearances.faces.iter().map(|f| &f.look)) {
insert(Definition::Appearance(ScalarLook::from_look(look)))?;
}
for material in &doc.user_materials {
insert(Definition::Material(ScalarMaterial::from_material(material)))?;
if let Some(look) = &material.appearance {
insert(Definition::Appearance(ScalarLook::from_look(look)))?;
}
}
let out = Self { format: FORMAT.into(), id: id.into(), name: name.into(), entries: entries.into_values().collect() };
out.validate()?;
Ok(out)
}
}
#[cfg(test)]
#[path = "catalog_tests.rs"]
mod tests;
+125
View File
@@ -0,0 +1,125 @@
use super::*;
use serde_json::json;
fn catalog() -> Catalog {
let entry = Entry::new(
"user.test",
Definition::Appearance(ScalarLook { name: "Warm oak colour".into(), color: [180, 120, 70], opacity: 0.8 }),
Provenance::default(),
)
.unwrap();
Catalog { format: FORMAT.into(), id: "user.test".into(), name: "Original test library".into(), entries: vec![entry] }
}
#[test]
fn catalog_roundtrip_is_deterministic_and_revision_covers_provenance_and_definition() {
let original = catalog();
let bytes = original.encode().unwrap();
let restored = Catalog::decode(&bytes).unwrap();
assert_eq!(original, restored);
assert_eq!(bytes, restored.encode().unwrap());
let mut changed = restored.clone();
changed.entries[0].provenance.license = "CC0-1.0".into();
assert!(changed.validate().is_err());
changed.entries[0].revision = changed.entries[0].content_revision().unwrap();
assert_ne!(changed.entries[0].revision, original.entries[0].revision);
changed.validate().unwrap();
if let Definition::Appearance(look) = &mut changed.entries[0].definition {
look.color = [1, 2, 3];
}
assert!(changed.validate().is_err());
}
#[test]
fn strict_schema_rejects_unknown_kinds_fields_versions_and_bad_identity() {
let value = serde_json::to_value(catalog()).unwrap();
for changed in [
{
let mut v = value.clone();
v["format"] = json!("solidcraft-library/2");
v
},
{
let mut v = value.clone();
v["entries"][0]["definition"]["kind"] = json!("texture_set");
v
},
{
let mut v = value.clone();
v["entries"][0]["definition"]["value"]["roughness"] = json!(0.5);
v
},
{
let mut v = value.clone();
v["entries"][0]["id"] = json!("other.pack/appearance-a");
v
},
{
let mut v = value.clone();
v["entries"][0]["revision"] = json!("0".repeat(64));
v
},
{
let mut v = value.clone();
v["id"] = json!("USER.TEST");
v
},
{
let mut v = value.clone();
v["unknown"] = json!(true);
v
},
] {
assert!(Catalog::decode(&serde_json::to_vec(&changed).unwrap()).is_err());
}
}
#[test]
fn scalar_validation_refuses_nonfinite_density_opacity_counts_strings_and_oversized_input() {
let mut invalid = catalog();
invalid.entries.push(invalid.entries[0].clone());
assert!(invalid.validate().is_err());
invalid.entries = vec![catalog().entries[0].clone(); MAX_ENTRIES + 1];
assert!(invalid.validate().is_err());
assert!(Catalog::decode(&serde_json::to_vec(&invalid).unwrap()).is_err());
let entry = catalog().entries.remove(0);
let mut boundary = catalog();
boundary.entries = (0..MAX_ENTRIES).map(|i| Entry { id: format!("user.test/entry-{i}"), ..entry.clone() }).collect();
Catalog::decode(&boundary.encode().unwrap()).unwrap();
boundary.entries.push(Entry { id: "user.test/one-more".into(), ..entry });
assert!(Catalog::decode(&serde_json::to_vec(&boundary).unwrap()).is_err());
for opacity in [f64::NAN, f64::INFINITY, -0.1, 1.1] {
let mut entry = catalog().entries.remove(0);
if let Definition::Appearance(look) = &mut entry.definition {
look.opacity = opacity;
}
assert!(entry.content_revision().is_err());
}
for density in [0.0, -1.0, 1000.1, f64::NAN, f64::INFINITY] {
assert!(
Entry::new(
"user.test",
Definition::Material(ScalarMaterial { name: "Timber".into(), density_g_cm3: density, appearance: None }),
Provenance::default()
)
.is_err()
);
}
for name in ["".into(), "x".repeat(129), "bad\nname".into(), " leading".into()] {
assert!(Entry::new("user.test", Definition::Appearance(ScalarLook { name, color: [1, 2, 3], opacity: 1.0 }), Provenance::default()).is_err());
}
assert!(Catalog::decode(&vec![b' '; MAX_BYTES + 1]).is_err());
let source = Provenance { source_url: Some("x".repeat(2049)), ..Default::default() };
assert!(Entry::new("user.test", catalog().entries[0].definition.clone(), source).is_err());
}
#[test]
fn design_export_deduplicates_exact_looks_and_keeps_physical_properties_distinct() {
let mut doc = Document::default();
let look = ScalarLook { name: "Warm oak colour".into(), color: [180, 120, 70], opacity: 0.8 }.to_look();
doc.appearances.bodies.insert("Body1".into(), look.clone());
doc.appearances.bodies.insert("Body2".into(), look.clone());
doc.user_materials.push(UserMaterial { name: "Timber density".into(), density: 0.7, appearance: Some(look) });
let library = Catalog::from_document(&doc, "user.test", "Test", Provenance::default()).unwrap();
assert_eq!(library.entries.len(), 2);
assert!(library.entries.iter().any(|e| matches!(&e.definition,Definition::Material(m) if m.density_g_cm3==0.7)));
assert_eq!(doc.user_materials[0].density, 0.7);
assert_eq!(library, Catalog::decode(&library.encode().unwrap()).unwrap());
assert!(Catalog::from_document(&Document::default(), "user.test", "Test", Provenance { license: String::new(), ..Default::default() }).is_err());
}
+1
View File
@@ -6,6 +6,7 @@
#![deny(clippy::unwrap_used, clippy::expect_used, clippy::panic, clippy::unimplemented, clippy::todo, clippy::unreachable)]
#![forbid(unsafe_code)]
pub mod catalog;
mod design;
mod design_metadata;
mod dxf;
+59
View File
@@ -129,6 +129,29 @@ pub fn write_atomic(path: &Path, bytes: &[u8], backup: bool) -> std::io::Result<
write_steps(&resolve_links(path)?, bytes, backup, None)
}
/// Publish a complete new file without replacing an existing destination, even if another
/// writer creates it after this call starts. Unsupported filesystems return an error.
pub fn create_atomic(path: &Path, bytes: &[u8]) -> std::io::Result<()> {
if cfg!(target_arch = "wasm32") {
let name = path.to_string_lossy();
if crate::vfs::exists(&name) {
return Err(std::io::Error::new(std::io::ErrorKind::AlreadyExists, "destination already exists"));
}
return crate::vfs::write(&name, bytes);
}
let tmp = temp_beside(path, "create");
// Only remove a temporary file that this invocation successfully created.
let mut file = OpenOptions::new().write(true).create_new(true).open(&tmp)?;
let result = file.write_all(bytes).and_then(|_| file.sync_all());
drop(file);
let result = result.and_then(|_| fs::hard_link(&tmp, path));
let _ = fs::remove_file(&tmp);
if result.is_ok() {
sync_dir(path);
}
result
}
/// Leftover temporary files of saves that never finished (a crash part way), next to `path`
/// (next to the file a link at `path` points to, where a save puts them).
pub fn stale_temps(path: &Path) -> Vec<PathBuf> {
@@ -150,6 +173,42 @@ pub fn stale_temps(path: &Path) -> Vec<PathBuf> {
mod tests {
use super::*;
#[test]
fn create_does_not_replace_an_existing_destination() {
let d = dir("create");
let p = d.join("render.png");
create_atomic(&p, b"first complete image").unwrap();
assert_eq!(create_atomic(&p, b"replacement").unwrap_err().kind(), std::io::ErrorKind::AlreadyExists);
assert_eq!(fs::read(&p).unwrap(), b"first complete image");
assert!(stale_temps(&p).is_empty());
fs::remove_dir_all(d).unwrap();
}
#[test]
fn concurrent_creators_publish_exactly_one_whole_file() {
let d = dir("create-race");
let p = d.join("render.png");
let gate = std::sync::Arc::new(std::sync::Barrier::new(2));
let workers: Vec<_> = [b"first image".as_slice(), b"second image".as_slice()]
.into_iter()
.map(|bytes| {
let gate = gate.clone();
let p = p.clone();
std::thread::spawn(move || {
gate.wait();
create_atomic(&p, bytes)
})
})
.collect();
let results: Vec<_> = workers.into_iter().map(|w| w.join().unwrap()).collect();
assert_eq!(results.iter().filter(|r| r.is_ok()).count(), 1);
assert_eq!(results.iter().filter_map(|r| r.as_ref().err()).next().unwrap().kind(), std::io::ErrorKind::AlreadyExists);
let bytes = fs::read(&p).unwrap();
assert!(bytes == b"first image" || bytes == b"second image");
assert!(stale_temps(&p).is_empty());
fs::remove_dir_all(d).unwrap();
}
fn dir(tag: &str) -> PathBuf {
let d = std::env::temp_dir().join(format!("solidcraft-safe-write-{tag}-{}", std::process::id()));
let _ = fs::remove_dir_all(&d);