1#![allow(deprecated)]
6
7mod error;
8mod named_package_ident;
9mod package_hash;
10mod package_id;
11mod package_ident;
12mod package_source;
13
14pub use self::{
15 error::PackageParseError,
16 named_package_ident::{NamedPackageIdent, Tag},
17 package_hash::PackageHash,
18 package_id::{NamedPackageId, PackageId},
19 package_ident::PackageIdent,
20 package_source::PackageSource,
21};
22
23use std::{
24 borrow::Cow,
25 collections::{BTreeMap, BTreeSet},
26 fmt::{self, Display},
27 path::{Path, PathBuf},
28 str::FromStr,
29};
30
31use indexmap::IndexMap;
32use semver::{Version, VersionReq};
33use serde::{de::Error as _, Deserialize, Serialize};
34use thiserror::Error;
35
36#[derive(Clone, Copy, Default, Debug, Deserialize, Serialize, PartialEq, Eq)]
41#[non_exhaustive]
42pub enum Abi {
43 #[default]
44 #[serde(rename = "none")]
45 None,
46 #[serde(rename = "wasi")]
47 Wasi,
48 #[serde(rename = "wasm4")]
49 WASM4,
50}
51
52impl Abi {
53 pub fn to_str(&self) -> &str {
55 match self {
56 Abi::Wasi => "wasi",
57 Abi::WASM4 => "wasm4",
58 Abi::None => "generic",
59 }
60 }
61
62 pub fn is_none(&self) -> bool {
64 matches!(self, Abi::None)
65 }
66
67 pub fn from_name(name: &str) -> Self {
69 name.parse().unwrap_or(Abi::None)
70 }
71}
72
73impl fmt::Display for Abi {
74 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
75 write!(f, "{}", self.to_str())
76 }
77}
78
79impl FromStr for Abi {
80 type Err = Box<dyn std::error::Error + Send + Sync>;
81
82 fn from_str(s: &str) -> Result<Self, Self::Err> {
83 match s.to_lowercase().as_str() {
84 "wasi" => Ok(Abi::Wasi),
85 "wasm4" => Ok(Abi::WASM4),
86 "generic" => Ok(Abi::None),
87 _ => Err(format!("Unknown ABI, \"{s}\"").into()),
88 }
89 }
90}
91
92pub static MANIFEST_FILE_NAME: &str = "wasmer.toml";
94
95const README_PATHS: &[&str; 5] = &[
96 "README",
97 "README.md",
98 "README.markdown",
99 "README.mdown",
100 "README.mkdn",
101];
102
103const LICENSE_PATHS: &[&str; 3] = &["LICENSE", "LICENSE.md", "COPYING"];
104
105#[derive(Clone, Debug, Deserialize, Serialize, derive_builder::Builder)]
109#[non_exhaustive]
110pub struct Package {
111 #[builder(setter(into, strip_option), default)]
113 pub name: Option<String>,
114 #[builder(setter(into, strip_option), default)]
116 pub version: Option<Version>,
117 #[builder(setter(into, strip_option), default)]
119 pub description: Option<String>,
120 #[builder(setter(into, strip_option), default)]
122 pub license: Option<String>,
123 #[serde(rename = "license-file")]
125 #[builder(setter(into, strip_option), default)]
126 pub license_file: Option<PathBuf>,
127 #[serde(skip_serializing_if = "Option::is_none")]
129 #[builder(setter(into, strip_option), default)]
130 pub readme: Option<PathBuf>,
131 #[serde(skip_serializing_if = "Option::is_none")]
133 #[builder(setter(into, strip_option), default)]
134 pub repository: Option<String>,
135 #[serde(skip_serializing_if = "Option::is_none")]
137 #[builder(setter(into, strip_option), default)]
138 pub homepage: Option<String>,
139 #[serde(rename = "wasmer-extra-flags")]
140 #[builder(setter(into, strip_option), default)]
141 #[deprecated(
142 since = "0.9.2",
143 note = "Use runner-specific command attributes instead"
144 )]
145 pub wasmer_extra_flags: Option<String>,
146 #[serde(
147 rename = "disable-command-rename",
148 default,
149 skip_serializing_if = "std::ops::Not::not"
150 )]
151 #[builder(default)]
152 #[deprecated(
153 since = "0.9.2",
154 note = "Does nothing. Prefer a runner-specific command attribute instead"
155 )]
156 pub disable_command_rename: bool,
157 #[serde(
163 rename = "rename-commands-to-raw-command-name",
164 default,
165 skip_serializing_if = "std::ops::Not::not"
166 )]
167 #[builder(default)]
168 #[deprecated(
169 since = "0.9.2",
170 note = "Does nothing. Prefer a runner-specific command attribute instead"
171 )]
172 pub rename_commands_to_raw_command_name: bool,
173 #[serde(skip_serializing_if = "Option::is_none")]
175 #[builder(setter(into, strip_option), default)]
176 pub entrypoint: Option<String>,
177 #[serde(default, skip_serializing_if = "std::ops::Not::not")]
179 #[builder(default)]
180 pub private: bool,
181}
182
183impl Package {
184 pub fn new_empty() -> Self {
185 PackageBuilder::default().build().unwrap()
186 }
187
188 pub fn builder(
190 name: impl Into<String>,
191 version: Version,
192 description: impl Into<String>,
193 ) -> PackageBuilder {
194 PackageBuilder::new(name, version, description)
195 }
196}
197
198impl PackageBuilder {
199 pub fn new(name: impl Into<String>, version: Version, description: impl Into<String>) -> Self {
200 let mut builder = PackageBuilder::default();
201 builder.name(name).version(version).description(description);
202 builder
203 }
204}
205
206#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
207#[serde(untagged)]
208pub enum Command {
209 V1(CommandV1),
210 V2(CommandV2),
211}
212
213impl Command {
214 pub fn get_name(&self) -> &str {
216 match self {
217 Self::V1(c) => &c.name,
218 Self::V2(c) => &c.name,
219 }
220 }
221
222 pub fn get_module(&self) -> &ModuleReference {
224 match self {
225 Self::V1(c) => &c.module,
226 Self::V2(c) => &c.module,
227 }
228 }
229}
230
231#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
238#[serde(deny_unknown_fields)] #[deprecated(since = "0.9.2", note = "Prefer the CommandV2 syntax")]
241pub struct CommandV1 {
242 pub name: String,
243 pub module: ModuleReference,
244 pub main_args: Option<String>,
245 pub package: Option<String>,
246}
247
248#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
250pub struct CommandV2 {
251 pub name: String,
253 pub module: ModuleReference,
255 pub runner: String,
259 pub annotations: Option<CommandAnnotations>,
261}
262
263impl CommandV2 {
264 pub fn get_annotations(&self, basepath: &Path) -> Result<Option<ciborium::Value>, String> {
267 match self.annotations.as_ref() {
268 Some(CommandAnnotations::Raw(v)) => Ok(Some(toml_to_cbor_value(v))),
269 Some(CommandAnnotations::File(FileCommandAnnotations { file, kind })) => {
270 let path = basepath.join(file.clone());
271 let file = std::fs::read_to_string(&path).map_err(|e| {
272 format!(
273 "Error reading {:?}.annotation ({:?}): {e}",
274 self.name,
275 path.display()
276 )
277 })?;
278 match kind {
279 FileKind::Json => {
280 let value: serde_json::Value =
281 serde_json::from_str(&file).map_err(|e| {
282 format!(
283 "Error reading {:?}.annotation ({:?}): {e}",
284 self.name,
285 path.display()
286 )
287 })?;
288 Ok(Some(json_to_cbor_value(&value)))
289 }
290 FileKind::Yaml => {
291 let value: serde_yaml::Value =
292 serde_yaml::from_str(&file).map_err(|e| {
293 format!(
294 "Error reading {:?}.annotation ({:?}): {e}",
295 self.name,
296 path.display()
297 )
298 })?;
299 Ok(Some(yaml_to_cbor_value(&value)))
300 }
301 }
302 }
303 None => Ok(None),
304 }
305 }
306}
307
308#[derive(Clone, Debug, PartialEq)]
314pub enum ModuleReference {
315 CurrentPackage {
317 module: String,
319 },
320 Dependency {
323 dependency: String,
325 module: String,
327 },
328}
329
330impl Serialize for ModuleReference {
331 fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
332 where
333 S: serde::Serializer,
334 {
335 self.to_string().serialize(serializer)
336 }
337}
338
339impl<'de> Deserialize<'de> for ModuleReference {
340 fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
341 where
342 D: serde::Deserializer<'de>,
343 {
344 let repr: Cow<'de, str> = Cow::deserialize(deserializer)?;
345 repr.parse().map_err(D::Error::custom)
346 }
347}
348
349impl FromStr for ModuleReference {
350 type Err = Box<dyn std::error::Error + Send + Sync>;
351
352 fn from_str(s: &str) -> Result<Self, Self::Err> {
353 match s.split_once(':') {
354 Some((dependency, module)) => {
355 if module.contains(':') {
356 return Err("Invalid format".into());
357 }
358
359 Ok(ModuleReference::Dependency {
360 dependency: dependency.to_string(),
361 module: module.to_string(),
362 })
363 }
364 None => Ok(ModuleReference::CurrentPackage {
365 module: s.to_string(),
366 }),
367 }
368 }
369}
370
371impl Display for ModuleReference {
372 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
373 match self {
374 ModuleReference::CurrentPackage { module } => Display::fmt(module, f),
375 ModuleReference::Dependency { dependency, module } => {
376 write!(f, "{dependency}:{module}")
377 }
378 }
379 }
380}
381
382fn toml_to_cbor_value(val: &toml::Value) -> ciborium::Value {
383 match val {
384 toml::Value::String(s) => ciborium::Value::Text(s.clone()),
385 toml::Value::Integer(i) => ciborium::Value::Integer(ciborium::value::Integer::from(*i)),
386 toml::Value::Float(f) => ciborium::Value::Float(*f),
387 toml::Value::Boolean(b) => ciborium::Value::Bool(*b),
388 toml::Value::Datetime(d) => ciborium::Value::Text(format!("{d}")),
389 toml::Value::Array(sq) => {
390 ciborium::Value::Array(sq.iter().map(toml_to_cbor_value).collect())
391 }
392 toml::Value::Table(m) => ciborium::Value::Map(
393 m.iter()
394 .map(|(k, v)| (ciborium::Value::Text(k.clone()), toml_to_cbor_value(v)))
395 .collect(),
396 ),
397 }
398}
399
400fn json_to_cbor_value(val: &serde_json::Value) -> ciborium::Value {
401 match val {
402 serde_json::Value::Null => ciborium::Value::Null,
403 serde_json::Value::Bool(b) => ciborium::Value::Bool(*b),
404 serde_json::Value::Number(n) => {
405 if let Some(i) = n.as_i64() {
406 ciborium::Value::Integer(ciborium::value::Integer::from(i))
407 } else if let Some(u) = n.as_u64() {
408 ciborium::Value::Integer(ciborium::value::Integer::from(u))
409 } else if let Some(f) = n.as_f64() {
410 ciborium::Value::Float(f)
411 } else {
412 ciborium::Value::Null
413 }
414 }
415 serde_json::Value::String(s) => ciborium::Value::Text(s.clone()),
416 serde_json::Value::Array(sq) => {
417 ciborium::Value::Array(sq.iter().map(json_to_cbor_value).collect())
418 }
419 serde_json::Value::Object(m) => ciborium::Value::Map(
420 m.iter()
421 .map(|(k, v)| (ciborium::Value::Text(k.clone()), json_to_cbor_value(v)))
422 .collect(),
423 ),
424 }
425}
426
427fn yaml_to_cbor_value(val: &serde_yaml::Value) -> ciborium::Value {
428 match val {
429 serde_yaml::Value::Null => ciborium::Value::Null,
430 serde_yaml::Value::Bool(b) => ciborium::Value::Bool(*b),
431 serde_yaml::Value::Number(n) => {
432 if let Some(i) = n.as_i64() {
433 ciborium::Value::Integer(ciborium::value::Integer::from(i))
434 } else if let Some(u) = n.as_u64() {
435 ciborium::Value::Integer(ciborium::value::Integer::from(u))
436 } else if let Some(f) = n.as_f64() {
437 ciborium::Value::Float(f)
438 } else {
439 ciborium::Value::Null
440 }
441 }
442 serde_yaml::Value::String(s) => ciborium::Value::Text(s.clone()),
443 serde_yaml::Value::Sequence(sq) => {
444 ciborium::Value::Array(sq.iter().map(yaml_to_cbor_value).collect())
445 }
446 serde_yaml::Value::Mapping(m) => ciborium::Value::Map(
447 m.iter()
448 .map(|(k, v)| (yaml_to_cbor_value(k), yaml_to_cbor_value(v)))
449 .collect(),
450 ),
451 serde_yaml::Value::Tagged(tag) => yaml_to_cbor_value(&tag.value),
452 }
453}
454
455#[derive(Debug, Clone, PartialEq, Deserialize, Serialize)]
457#[serde(untagged)]
458#[repr(C)]
459pub enum CommandAnnotations {
460 File(FileCommandAnnotations),
462 Raw(toml::Value),
464}
465
466#[derive(Debug, Clone, PartialEq, Eq, Deserialize, Serialize)]
468pub struct FileCommandAnnotations {
469 pub file: PathBuf,
471 pub kind: FileKind,
473}
474
475#[derive(Debug, Copy, Clone, PartialEq, PartialOrd, Ord, Eq, Deserialize, Serialize)]
477pub enum FileKind {
478 #[serde(rename = "yaml")]
480 Yaml,
481 #[serde(rename = "json")]
483 Json,
484}
485
486#[derive(Clone, Debug, PartialEq, Eq, Deserialize, Serialize)]
489pub struct Module {
490 pub name: String,
492 pub source: PathBuf,
495 #[serde(default = "Abi::default", skip_serializing_if = "Abi::is_none")]
497 pub abi: Abi,
498 #[serde(default)]
499 pub kind: Option<String>,
500 #[serde(skip_serializing_if = "Option::is_none")]
502 pub interfaces: Option<IndexMap<String, String>>,
503 pub bindings: Option<Bindings>,
506 #[serde(skip_serializing_if = "Option::is_none")]
508 pub annotations: Option<UserAnnotations>,
509}
510
511#[derive(Clone, Debug, PartialEq, Eq, Deserialize, Serialize, Default)]
513pub struct UserAnnotations {
514 pub suggested_compiler_optimizations: SuggestedCompilerOptimizations,
515}
516
517#[derive(Clone, Debug, PartialEq, Eq, Hash, Deserialize, Serialize, Default)]
519pub struct SuggestedCompilerOptimizations {
520 pub pass_params: Option<bool>,
521}
522
523impl SuggestedCompilerOptimizations {
524 pub const KEY: &'static str = "suggested_compiler_optimizations";
525 pub const PASS_PARAMS_KEY: &'static str = "pass_params";
526}
527
528#[derive(Clone, Debug, PartialEq, Eq)]
530pub enum Bindings {
531 Wit(WitBindings),
532 Wai(WaiBindings),
533}
534
535impl Bindings {
536 pub fn referenced_files(&self, base_directory: &Path) -> Result<Vec<PathBuf>, ImportsError> {
543 match self {
544 Bindings::Wit(WitBindings { wit_exports, .. }) => {
545 let path = base_directory.join(wit_exports);
549
550 if path.exists() {
551 Ok(vec![path])
552 } else {
553 Err(ImportsError::FileNotFound(path))
554 }
555 }
556 Bindings::Wai(wai) => wai.referenced_files(base_directory),
557 }
558 }
559}
560
561impl Serialize for Bindings {
562 fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
563 where
564 S: serde::Serializer,
565 {
566 match self {
567 Bindings::Wit(w) => w.serialize(serializer),
568 Bindings::Wai(w) => w.serialize(serializer),
569 }
570 }
571}
572
573impl<'de> Deserialize<'de> for Bindings {
574 fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
575 where
576 D: serde::Deserializer<'de>,
577 {
578 let value = toml::Value::deserialize(deserializer)?;
579
580 let keys = ["wit-bindgen", "wai-version"];
581 let [wit_bindgen, wai_version] = keys.map(|key| value.get(key).is_some());
582
583 match (wit_bindgen, wai_version) {
584 (true, false) => WitBindings::deserialize(value)
585 .map(Bindings::Wit)
586 .map_err(D::Error::custom),
587 (false, true) => WaiBindings::deserialize(value)
588 .map(Bindings::Wai)
589 .map_err(D::Error::custom),
590 (true, true) | (false, false) => {
591 let msg = format!(
592 "expected one of \"{}\" to be provided, but not both",
593 keys.join("\" or \""),
594 );
595 Err(D::Error::custom(msg))
596 }
597 }
598 }
599}
600
601#[derive(Clone, Debug, PartialEq, Eq, Deserialize, Serialize)]
602#[serde(rename_all = "kebab-case")]
603pub struct WitBindings {
604 pub wit_bindgen: Version,
606 pub wit_exports: PathBuf,
608}
609
610#[derive(Clone, Debug, PartialEq, Eq, Deserialize, Serialize)]
611#[serde(rename_all = "kebab-case")]
612pub struct WaiBindings {
613 pub wai_version: Version,
615 pub exports: Option<PathBuf>,
617 #[serde(default, skip_serializing_if = "Vec::is_empty")]
620 pub imports: Vec<PathBuf>,
621}
622
623impl WaiBindings {
624 fn referenced_files(&self, base_directory: &Path) -> Result<Vec<PathBuf>, ImportsError> {
625 let WaiBindings {
626 exports, imports, ..
627 } = self;
628
629 let initial_paths = exports
634 .iter()
635 .chain(imports)
636 .map(|relative_path| base_directory.join(relative_path));
637
638 let mut to_check: Vec<PathBuf> = Vec::new();
639
640 for path in initial_paths {
641 if !path.exists() {
642 return Err(ImportsError::FileNotFound(path));
643 }
644 to_check.push(path);
645 }
646
647 let mut files = BTreeSet::new();
648
649 while let Some(path) = to_check.pop() {
650 if files.contains(&path) {
651 continue;
652 }
653
654 to_check.extend(get_imported_wai_files(&path)?);
655 files.insert(path);
656 }
657
658 Ok(files.into_iter().collect())
659 }
660}
661
662fn get_imported_wai_files(path: &Path) -> Result<Vec<PathBuf>, ImportsError> {
667 let _wai_src = std::fs::read_to_string(path).map_err(|error| ImportsError::Read {
668 path: path.to_path_buf(),
669 error,
670 })?;
671
672 let parent_dir = path.parent()
673 .expect("All paths should have a parent directory because we joined them relative to the base directory");
674
675 let raw_imports: Vec<String> = Vec::new();
679
680 let mut resolved_paths = Vec::new();
683
684 for imported in raw_imports {
685 let absolute_path = parent_dir.join(imported);
686
687 if !absolute_path.exists() {
688 return Err(ImportsError::ImportedFileNotFound {
689 path: absolute_path,
690 referenced_by: path.to_path_buf(),
691 });
692 }
693
694 resolved_paths.push(absolute_path);
695 }
696
697 Ok(resolved_paths)
698}
699
700#[derive(Debug, thiserror::Error)]
702#[non_exhaustive]
703pub enum ImportsError {
704 #[error(
705 "The \"{}\" mentioned in the manifest doesn't exist",
706 _0.display(),
707 )]
708 FileNotFound(PathBuf),
709 #[error(
710 "The \"{}\" imported by \"{}\" doesn't exist",
711 path.display(),
712 referenced_by.display(),
713 )]
714 ImportedFileNotFound {
715 path: PathBuf,
716 referenced_by: PathBuf,
717 },
718 #[error("Unable to parse \"{}\" as a WAI file", path.display())]
719 WaiParse { path: PathBuf },
720 #[error("Unable to read \"{}\"", path.display())]
721 Read {
722 path: PathBuf,
723 #[source]
724 error: std::io::Error,
725 },
726}
727
728#[derive(Clone, Debug, Deserialize, Serialize, derive_builder::Builder)]
730#[non_exhaustive]
731pub struct Manifest {
732 pub package: Option<Package>,
734 #[serde(default, skip_serializing_if = "IndexMap::is_empty")]
736 #[builder(default)]
737 pub dependencies: IndexMap<String, VersionReq>,
738 #[serde(default, skip_serializing_if = "IndexMap::is_empty")]
741 #[builder(default)]
742 pub fs: IndexMap<String, PathBuf>,
743 #[serde(default, rename = "module", skip_serializing_if = "Vec::is_empty")]
745 #[builder(default)]
746 pub modules: Vec<Module>,
747 #[serde(default, rename = "command", skip_serializing_if = "Vec::is_empty")]
749 #[builder(default)]
750 pub commands: Vec<Command>,
751}
752
753impl Manifest {
754 pub fn new_empty() -> Self {
755 Self {
756 package: None,
757 dependencies: IndexMap::new(),
758 fs: IndexMap::new(),
759 modules: Vec::new(),
760 commands: Vec::new(),
761 }
762 }
763
764 pub fn builder(package: Package) -> ManifestBuilder {
766 ManifestBuilder::new(package)
767 }
768
769 pub fn parse(s: &str) -> Result<Self, toml::de::Error> {
771 toml::from_str(s)
772 }
773
774 pub fn find_in_directory<T: AsRef<Path>>(path: T) -> Result<Self, ManifestError> {
777 let path = path.as_ref();
778
779 if !path.is_dir() {
780 return Err(ManifestError::MissingManifest(path.to_path_buf()));
781 }
782 let manifest_path_buf = path.join(MANIFEST_FILE_NAME);
783 let contents = std::fs::read_to_string(&manifest_path_buf)
784 .map_err(|_e| ManifestError::MissingManifest(manifest_path_buf))?;
785 let mut manifest: Self = toml::from_str(contents.as_str())?;
786
787 if let Some(package) = manifest.package.as_mut() {
788 if package.readme.is_none() {
789 package.readme = locate_file(path, README_PATHS);
790 }
791
792 if package.license_file.is_none() {
793 package.license_file = locate_file(path, LICENSE_PATHS);
794 }
795 }
796 manifest.validate()?;
797
798 Ok(manifest)
799 }
800
801 pub fn validate(&self) -> Result<(), ValidationError> {
811 let mut modules = BTreeMap::new();
812
813 for module in &self.modules {
814 let is_duplicate = modules.insert(&module.name, module).is_some();
815
816 if is_duplicate {
817 return Err(ValidationError::DuplicateModule {
818 name: module.name.clone(),
819 });
820 }
821 }
822
823 let mut commands = BTreeMap::new();
824
825 for command in &self.commands {
826 let is_duplicate = commands.insert(command.get_name(), command).is_some();
827
828 if is_duplicate {
829 return Err(ValidationError::DuplicateCommand {
830 name: command.get_name().to_string(),
831 });
832 }
833
834 let module_reference = command.get_module();
835 match &module_reference {
836 ModuleReference::CurrentPackage { module } => {
837 if let Some(module) = modules.get(&module) {
838 if module.abi == Abi::None && module.interfaces.is_none() {
839 return Err(ValidationError::MissingABI {
840 command: command.get_name().to_string(),
841 module: module.name.clone(),
842 });
843 }
844 } else {
845 return Err(ValidationError::MissingModuleForCommand {
846 command: command.get_name().to_string(),
847 module: command.get_module().clone(),
848 });
849 }
850 }
851 ModuleReference::Dependency { dependency, .. } => {
852 if !self.dependencies.contains_key(dependency) {
855 return Err(ValidationError::MissingDependency {
856 command: command.get_name().to_string(),
857 dependency: dependency.clone(),
858 module_ref: module_reference.clone(),
859 });
860 }
861 }
862 }
863 }
864
865 if let Some(package) = &self.package {
866 if let Some(entrypoint) = package.entrypoint.as_deref() {
867 if !commands.contains_key(entrypoint) {
868 return Err(ValidationError::InvalidEntrypoint {
869 entrypoint: entrypoint.to_string(),
870 available_commands: commands.keys().map(ToString::to_string).collect(),
871 });
872 }
873 }
874 }
875
876 Ok(())
877 }
878
879 pub fn add_dependency(&mut self, dependency_name: String, dependency_version: VersionReq) {
881 self.dependencies
882 .insert(dependency_name, dependency_version);
883 }
884
885 pub fn remove_dependency(&mut self, dependency_name: &str) -> Option<VersionReq> {
887 self.dependencies.remove(dependency_name)
888 }
889
890 pub fn to_string(&self) -> anyhow::Result<String> {
892 let repr = toml::to_string_pretty(&self)?;
893 Ok(repr)
894 }
895
896 pub fn save(&self, path: impl AsRef<Path>) -> anyhow::Result<()> {
898 let manifest = toml::to_string_pretty(self)?;
899 std::fs::write(path, manifest).map_err(ManifestError::CannotSaveManifest)?;
900 Ok(())
901 }
902}
903
904fn locate_file(path: &Path, candidates: &[&str]) -> Option<PathBuf> {
905 for filename in candidates {
906 let path_buf = path.join(filename);
907 if path_buf.exists() {
908 return Some(filename.into());
909 }
910 }
911 None
912}
913
914impl ManifestBuilder {
915 pub fn new(package: Package) -> Self {
916 let mut builder = ManifestBuilder::default();
917 builder.package(Some(package));
918 builder
919 }
920
921 pub fn map_fs(&mut self, guest: impl Into<String>, host: impl Into<PathBuf>) -> &mut Self {
924 self.fs
925 .get_or_insert_with(IndexMap::new)
926 .insert(guest.into(), host.into());
927 self
928 }
929
930 pub fn with_dependency(&mut self, name: impl Into<String>, version: VersionReq) -> &mut Self {
932 self.dependencies
933 .get_or_insert_with(IndexMap::new)
934 .insert(name.into(), version);
935 self
936 }
937
938 pub fn with_module(&mut self, module: Module) -> &mut Self {
940 self.modules.get_or_insert_with(Vec::new).push(module);
941 self
942 }
943
944 pub fn with_command(&mut self, command: Command) -> &mut Self {
946 self.commands.get_or_insert_with(Vec::new).push(command);
947 self
948 }
949}
950
951#[derive(Debug, Error)]
953#[non_exhaustive]
954pub enum ManifestError {
955 #[error("Manifest file not found at \"{}\"", _0.display())]
956 MissingManifest(PathBuf),
957 #[error("Could not save manifest file: {0}.")]
958 CannotSaveManifest(#[source] std::io::Error),
959 #[error("Could not parse manifest because {0}.")]
960 TomlParseError(#[from] toml::de::Error),
961 #[error("There was an error validating the manifest")]
962 ValidationError(#[from] ValidationError),
963}
964
965#[derive(Debug, PartialEq, Error)]
967#[non_exhaustive]
968pub enum ValidationError {
969 #[error(
970 "missing ABI field on module, \"{module}\", used by command, \"{command}\"; an ABI of `wasi` is required",
971 )]
972 MissingABI { command: String, module: String },
973 #[error("missing module, \"{module}\", in manifest used by command, \"{command}\"")]
974 MissingModuleForCommand {
975 command: String,
976 module: ModuleReference,
977 },
978 #[error("The \"{command}\" command refers to a nonexistent dependency, \"{dependency}\" in \"{module_ref}\"")]
979 MissingDependency {
980 command: String,
981 dependency: String,
982 module_ref: ModuleReference,
983 },
984 #[error("The entrypoint, \"{entrypoint}\", isn't a valid command (commands: {})", available_commands.join(", "))]
985 InvalidEntrypoint {
986 entrypoint: String,
987 available_commands: Vec<String>,
988 },
989 #[error("Duplicate module, \"{name}\"")]
990 DuplicateModule { name: String },
991 #[error("Duplicate command, \"{name}\"")]
992 DuplicateCommand { name: String },
993}
994
995#[cfg(test)]
996mod tests {
997 use std::fmt::Debug;
998
999 use serde::{de::DeserializeOwned, Deserialize};
1000 use toml::toml;
1001
1002 use super::*;
1003
1004 #[test]
1005 fn test_to_string() {
1006 Manifest {
1007 package: Some(Package {
1008 name: Some("package/name".to_string()),
1009 version: Some(Version::parse("1.0.0").unwrap()),
1010 description: Some("test".to_string()),
1011 license: None,
1012 license_file: None,
1013 readme: None,
1014 repository: None,
1015 homepage: None,
1016 wasmer_extra_flags: None,
1017 disable_command_rename: false,
1018 rename_commands_to_raw_command_name: false,
1019 entrypoint: None,
1020 private: false,
1021 }),
1022 dependencies: IndexMap::new(),
1023 modules: vec![Module {
1024 name: "test".to_string(),
1025 abi: Abi::Wasi,
1026 bindings: None,
1027 interfaces: None,
1028 kind: Some("https://webc.org/kind/wasi".to_string()),
1029 source: Path::new("test.wasm").to_path_buf(),
1030 annotations: None,
1031 }],
1032 commands: Vec::new(),
1033 fs: vec![
1034 ("a".to_string(), Path::new("/a").to_path_buf()),
1035 ("b".to_string(), Path::new("/b").to_path_buf()),
1036 ]
1037 .into_iter()
1038 .collect(),
1039 }
1040 .to_string()
1041 .unwrap();
1042 }
1043
1044 #[test]
1045 fn interface_test() {
1046 let manifest_str = r#"
1047[package]
1048name = "test"
1049version = "0.0.0"
1050description = "This is a test package"
1051license = "MIT"
1052
1053[[module]]
1054name = "mod"
1055source = "target/wasm32-wasip1/release/mod.wasm"
1056interfaces = {"wasi" = "0.0.0-unstable"}
1057
1058[[module]]
1059name = "mod-with-exports"
1060source = "target/wasm32-wasip1/release/mod-with-exports.wasm"
1061bindings = { wit-exports = "exports.wit", wit-bindgen = "0.0.0" }
1062
1063[[command]]
1064name = "command"
1065module = "mod"
1066"#;
1067 let manifest: Manifest = Manifest::parse(manifest_str).unwrap();
1068 let modules = &manifest.modules;
1069 assert_eq!(
1070 modules[0].interfaces.as_ref().unwrap().get("wasi"),
1071 Some(&"0.0.0-unstable".to_string())
1072 );
1073
1074 assert_eq!(
1075 modules[1],
1076 Module {
1077 name: "mod-with-exports".to_string(),
1078 source: PathBuf::from("target/wasm32-wasip1/release/mod-with-exports.wasm"),
1079 abi: Abi::None,
1080 kind: None,
1081 interfaces: None,
1082 bindings: Some(Bindings::Wit(WitBindings {
1083 wit_exports: PathBuf::from("exports.wit"),
1084 wit_bindgen: "0.0.0".parse().unwrap()
1085 })),
1086 annotations: None
1087 },
1088 );
1089 }
1090
1091 #[test]
1092 fn parse_wit_bindings() {
1093 let table = toml! {
1094 name = "..."
1095 source = "..."
1096 bindings = { wit-bindgen = "0.1.0", wit-exports = "./file.wit" }
1097 };
1098
1099 let module = Module::deserialize(table).unwrap();
1100
1101 assert_eq!(
1102 module.bindings.as_ref().unwrap(),
1103 &Bindings::Wit(WitBindings {
1104 wit_bindgen: "0.1.0".parse().unwrap(),
1105 wit_exports: PathBuf::from("./file.wit"),
1106 }),
1107 );
1108 assert_round_trippable(&module);
1109 }
1110
1111 #[test]
1112 fn parse_wai_bindings() {
1113 let table = toml! {
1114 name = "..."
1115 source = "..."
1116 bindings = { wai-version = "0.1.0", exports = "./file.wai", imports = ["a.wai", "../b.wai"] }
1117 };
1118
1119 let module = Module::deserialize(table).unwrap();
1120
1121 assert_eq!(
1122 module.bindings.as_ref().unwrap(),
1123 &Bindings::Wai(WaiBindings {
1124 wai_version: "0.1.0".parse().unwrap(),
1125 exports: Some(PathBuf::from("./file.wai")),
1126 imports: vec![PathBuf::from("a.wai"), PathBuf::from("../b.wai")],
1127 }),
1128 );
1129 assert_round_trippable(&module);
1130 }
1131
1132 #[track_caller]
1133 fn assert_round_trippable<T>(value: &T)
1134 where
1135 T: Serialize + DeserializeOwned + PartialEq + Debug,
1136 {
1137 let repr = toml::to_string(value).unwrap();
1138 let round_tripped: T = toml::from_str(&repr).unwrap();
1139 assert_eq!(
1140 round_tripped, *value,
1141 "The value should convert to/from TOML losslessly"
1142 );
1143 }
1144
1145 #[test]
1146 fn imports_and_exports_are_optional_with_wai() {
1147 let table = toml! {
1148 name = "..."
1149 source = "..."
1150 bindings = { wai-version = "0.1.0" }
1151 };
1152
1153 let module = Module::deserialize(table).unwrap();
1154
1155 assert_eq!(
1156 module.bindings.as_ref().unwrap(),
1157 &Bindings::Wai(WaiBindings {
1158 wai_version: "0.1.0".parse().unwrap(),
1159 exports: None,
1160 imports: Vec::new(),
1161 }),
1162 );
1163 assert_round_trippable(&module);
1164 }
1165
1166 #[test]
1167 fn ambiguous_bindings_table() {
1168 let table = toml! {
1169 wai-version = "0.2.0"
1170 wit-bindgen = "0.1.0"
1171 };
1172
1173 let err = Bindings::deserialize(table).unwrap_err();
1174
1175 assert_eq!(
1176 err.to_string(),
1177 "expected one of \"wit-bindgen\" or \"wai-version\" to be provided, but not both\n"
1178 );
1179 }
1180
1181 #[test]
1182 fn bindings_table_that_is_neither_wit_nor_wai() {
1183 let table = toml! {
1184 wai-bindgen = "lol, this should have been wai-version"
1185 exports = "./file.wai"
1186 };
1187
1188 let err = Bindings::deserialize(table).unwrap_err();
1189
1190 assert_eq!(
1191 err.to_string(),
1192 "expected one of \"wit-bindgen\" or \"wai-version\" to be provided, but not both\n"
1193 );
1194 }
1195
1196 #[test]
1197 fn command_v2_isnt_ambiguous_with_command_v1() {
1198 let src = r#"
1199[package]
1200name = "hotg-ai/sine"
1201version = "0.12.0"
1202description = "sine"
1203
1204[dependencies]
1205"hotg-ai/train_test_split" = "0.12.1"
1206"hotg-ai/elastic_net" = "0.12.1"
1207
1208[[module]] # This is the same as atoms
1209name = "sine"
1210kind = "tensorflow-SavedModel" # It can also be "wasm" (default)
1211source = "models/sine"
1212
1213[[command]]
1214name = "run"
1215runner = "rune"
1216module = "sine"
1217annotations = { file = "Runefile.yml", kind = "yaml" }
1218"#;
1219
1220 let manifest: Manifest = toml::from_str(src).unwrap();
1221
1222 let commands = &manifest.commands;
1223 assert_eq!(commands.len(), 1);
1224 assert_eq!(
1225 commands[0],
1226 Command::V2(CommandV2 {
1227 name: "run".into(),
1228 module: "sine".parse().unwrap(),
1229 runner: "rune".into(),
1230 annotations: Some(CommandAnnotations::File(FileCommandAnnotations {
1231 file: "Runefile.yml".into(),
1232 kind: FileKind::Yaml,
1233 }))
1234 })
1235 );
1236 }
1237
1238 #[test]
1239 fn get_manifest() {
1240 let wasmer_toml = toml! {
1241 [package]
1242 name = "test"
1243 version = "1.0.0"
1244 repository = "test.git"
1245 homepage = "test.com"
1246 description = "The best package."
1247 };
1248 let manifest: Manifest = wasmer_toml.try_into().unwrap();
1249 if let Some(package) = manifest.package {
1250 assert!(!package.disable_command_rename);
1251 }
1252 }
1253
1254 #[test]
1255 fn parse_manifest_without_package_section() {
1256 let wasmer_toml = toml! {
1257 [[module]]
1258 name = "test-module"
1259 source = "data.wasm"
1260 abi = "wasi"
1261 };
1262 let manifest: Manifest = wasmer_toml.try_into().unwrap();
1263 assert!(manifest.package.is_none());
1264 }
1265
1266 #[test]
1267 fn get_commands() {
1268 let wasmer_toml = toml! {
1269 [package]
1270 name = "test"
1271 version = "1.0.0"
1272 repository = "test.git"
1273 homepage = "test.com"
1274 description = "The best package."
1275 [[module]]
1276 name = "test-pkg"
1277 module = "target.wasm"
1278 source = "source.wasm"
1279 description = "description"
1280 interfaces = {"wasi" = "0.0.0-unstable"}
1281 [[command]]
1282 name = "foo"
1283 module = "test"
1284 [[command]]
1285 name = "baz"
1286 module = "test"
1287 main_args = "$@"
1288 };
1289 let manifest: Manifest = wasmer_toml.try_into().unwrap();
1290 let commands = &manifest.commands;
1291 assert_eq!(2, commands.len());
1292 }
1293
1294 #[test]
1295 fn add_new_dependency() {
1296 let tmp_dir = tempfile::tempdir().unwrap();
1297 let tmp_dir_path: &std::path::Path = tmp_dir.as_ref();
1298 let manifest_path = tmp_dir_path.join(MANIFEST_FILE_NAME);
1299 let wasmer_toml = toml! {
1300 [package]
1301 name = "_/test"
1302 version = "1.0.0"
1303 description = "description"
1304 [[module]]
1305 name = "test"
1306 source = "test.wasm"
1307 interfaces = {}
1308 };
1309 let toml_string = toml::to_string(&wasmer_toml).unwrap();
1310 std::fs::write(manifest_path, toml_string).unwrap();
1311 let mut manifest = Manifest::find_in_directory(tmp_dir).unwrap();
1312
1313 let dependency_name = "dep_pkg";
1314 let dependency_version: VersionReq = "0.1.0".parse().unwrap();
1315
1316 manifest.add_dependency(dependency_name.to_string(), dependency_version.clone());
1317 assert_eq!(1, manifest.dependencies.len());
1318
1319 manifest.add_dependency(dependency_name.to_string(), dependency_version);
1321 assert_eq!(1, manifest.dependencies.len());
1322
1323 let dependency_name_2 = "dep_pkg_2";
1325 let dependency_version_2: VersionReq = "0.2.0".parse().unwrap();
1326 manifest.add_dependency(dependency_name_2.to_string(), dependency_version_2);
1327 assert_eq!(2, manifest.dependencies.len());
1328 }
1329
1330 #[test]
1331 fn duplicate_modules_are_invalid() {
1332 let wasmer_toml = toml! {
1333 [package]
1334 name = "some/package"
1335 version = "0.0.0"
1336 description = ""
1337 [[module]]
1338 name = "test"
1339 source = "test.wasm"
1340 [[module]]
1341 name = "test"
1342 source = "test.wasm"
1343 };
1344 let manifest = Manifest::deserialize(wasmer_toml).unwrap();
1345
1346 let error = manifest.validate().unwrap_err();
1347
1348 assert_eq!(
1349 error,
1350 ValidationError::DuplicateModule {
1351 name: "test".to_string()
1352 }
1353 );
1354 }
1355
1356 #[test]
1357 fn duplicate_commands_are_invalid() {
1358 let wasmer_toml = toml! {
1359 [package]
1360 name = "some/package"
1361 version = "0.0.0"
1362 description = ""
1363 [[module]]
1364 name = "test"
1365 source = "test.wasm"
1366 abi = "wasi"
1367 [[command]]
1368 name = "cmd"
1369 module = "test"
1370 [[command]]
1371 name = "cmd"
1372 module = "test"
1373 };
1374 let manifest = Manifest::deserialize(wasmer_toml).unwrap();
1375
1376 let error = manifest.validate().unwrap_err();
1377
1378 assert_eq!(
1379 error,
1380 ValidationError::DuplicateCommand {
1381 name: "cmd".to_string()
1382 }
1383 );
1384 }
1385
1386 #[test]
1387 fn nonexistent_entrypoint() {
1388 let wasmer_toml = toml! {
1389 [package]
1390 name = "some/package"
1391 version = "0.0.0"
1392 description = ""
1393 entrypoint = "this-doesnt-exist"
1394 [[module]]
1395 name = "test"
1396 source = "test.wasm"
1397 abi = "wasi"
1398 [[command]]
1399 name = "cmd"
1400 module = "test"
1401 };
1402 let manifest = Manifest::deserialize(wasmer_toml).unwrap();
1403
1404 let error = manifest.validate().unwrap_err();
1405
1406 assert_eq!(
1407 error,
1408 ValidationError::InvalidEntrypoint {
1409 entrypoint: "this-doesnt-exist".to_string(),
1410 available_commands: vec!["cmd".to_string()]
1411 }
1412 );
1413 }
1414
1415 #[test]
1416 fn command_with_nonexistent_module() {
1417 let wasmer_toml = toml! {
1418 [package]
1419 name = "some/package"
1420 version = "0.0.0"
1421 description = ""
1422 [[command]]
1423 name = "cmd"
1424 module = "this-doesnt-exist"
1425 };
1426 let manifest = Manifest::deserialize(wasmer_toml).unwrap();
1427
1428 let error = manifest.validate().unwrap_err();
1429
1430 assert_eq!(
1431 error,
1432 ValidationError::MissingModuleForCommand {
1433 command: "cmd".to_string(),
1434 module: "this-doesnt-exist".parse().unwrap()
1435 }
1436 );
1437 }
1438
1439 #[test]
1440 fn use_builder_api_to_create_simplest_manifest() {
1441 let package =
1442 Package::builder("my/package", "1.0.0".parse().unwrap(), "My awesome package")
1443 .build()
1444 .unwrap();
1445 let manifest = Manifest::builder(package).build().unwrap();
1446
1447 manifest.validate().unwrap();
1448 }
1449
1450 #[test]
1451 fn deserialize_command_referring_to_module_from_dependency() {
1452 let wasmer_toml = toml! {
1453 [package]
1454 name = "some/package"
1455 version = "0.0.0"
1456 description = ""
1457
1458 [dependencies]
1459 dep = "1.2.3"
1460
1461 [[command]]
1462 name = "cmd"
1463 module = "dep:module"
1464 };
1465 let manifest = Manifest::deserialize(wasmer_toml).unwrap();
1466
1467 let command = manifest
1468 .commands
1469 .iter()
1470 .find(|cmd| cmd.get_name() == "cmd")
1471 .unwrap();
1472
1473 assert_eq!(
1474 command.get_module(),
1475 &ModuleReference::Dependency {
1476 dependency: "dep".to_string(),
1477 module: "module".to_string()
1478 }
1479 );
1480 }
1481
1482 #[test]
1483 fn command_with_module_from_nonexistent_dependency() {
1484 let wasmer_toml = toml! {
1485 [package]
1486 name = "some/package"
1487 version = "0.0.0"
1488 description = ""
1489 [[command]]
1490 name = "cmd"
1491 module = "dep:module"
1492 };
1493 let manifest = Manifest::deserialize(wasmer_toml).unwrap();
1494
1495 let error = manifest.validate().unwrap_err();
1496
1497 assert_eq!(
1498 error,
1499 ValidationError::MissingDependency {
1500 command: "cmd".to_string(),
1501 dependency: "dep".to_string(),
1502 module_ref: ModuleReference::Dependency {
1503 dependency: "dep".to_string(),
1504 module: "module".to_string()
1505 }
1506 }
1507 );
1508 }
1509
1510 #[test]
1511 fn round_trip_dependency_module_ref() {
1512 let original = ModuleReference::Dependency {
1513 dependency: "my/dep".to_string(),
1514 module: "module".to_string(),
1515 };
1516
1517 let repr = original.to_string();
1518 let round_tripped: ModuleReference = repr.parse().unwrap();
1519
1520 assert_eq!(round_tripped, original);
1521 }
1522}