1use anyhow::{anyhow, Context, Result};
2use collections::{BTreeMap, HashMap};
3use fs::Fs;
4use language::LanguageServerName;
5use semantic_version::SemanticVersion;
6use serde::{Deserialize, Serialize};
7use std::{
8 ffi::OsStr,
9 fmt,
10 path::{Path, PathBuf},
11 sync::Arc,
12};
13
14/// This is the old version of the extension manifest, from when it was `extension.json`.
15#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)]
16pub struct OldExtensionManifest {
17 pub name: String,
18 pub version: Arc<str>,
19
20 #[serde(default)]
21 pub description: Option<String>,
22 #[serde(default)]
23 pub repository: Option<String>,
24 #[serde(default)]
25 pub authors: Vec<String>,
26
27 #[serde(default)]
28 pub themes: BTreeMap<Arc<str>, PathBuf>,
29 #[serde(default)]
30 pub languages: BTreeMap<Arc<str>, PathBuf>,
31 #[serde(default)]
32 pub grammars: BTreeMap<Arc<str>, PathBuf>,
33}
34
35/// The schema version of the [`ExtensionManifest`].
36#[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Clone, Copy, Serialize, Deserialize)]
37pub struct SchemaVersion(pub i32);
38
39impl fmt::Display for SchemaVersion {
40 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
41 write!(f, "{}", self.0)
42 }
43}
44
45impl SchemaVersion {
46 pub const ZERO: Self = Self(0);
47
48 pub fn is_v0(&self) -> bool {
49 self == &Self::ZERO
50 }
51}
52
53#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)]
54pub struct ExtensionManifest {
55 pub id: Arc<str>,
56 pub name: String,
57 pub version: Arc<str>,
58 pub schema_version: SchemaVersion,
59
60 #[serde(default)]
61 pub description: Option<String>,
62 #[serde(default)]
63 pub repository: Option<String>,
64 #[serde(default)]
65 pub authors: Vec<String>,
66 #[serde(default)]
67 pub lib: LibManifestEntry,
68
69 #[serde(default)]
70 pub themes: Vec<PathBuf>,
71 #[serde(default)]
72 pub languages: Vec<PathBuf>,
73 #[serde(default)]
74 pub grammars: BTreeMap<Arc<str>, GrammarManifestEntry>,
75 #[serde(default)]
76 pub language_servers: BTreeMap<LanguageServerName, LanguageServerManifestEntry>,
77}
78
79#[derive(Clone, Default, PartialEq, Eq, Debug, Deserialize, Serialize)]
80pub struct LibManifestEntry {
81 pub kind: Option<ExtensionLibraryKind>,
82 pub version: Option<SemanticVersion>,
83}
84
85#[derive(Clone, PartialEq, Eq, Debug, Deserialize, Serialize)]
86pub enum ExtensionLibraryKind {
87 Rust,
88}
89
90#[derive(Clone, Default, PartialEq, Eq, Debug, Deserialize, Serialize)]
91pub struct GrammarManifestEntry {
92 pub repository: String,
93 #[serde(alias = "commit")]
94 pub rev: String,
95 #[serde(default)]
96 pub path: Option<String>,
97}
98
99#[derive(Clone, PartialEq, Eq, Debug, Deserialize, Serialize)]
100pub struct LanguageServerManifestEntry {
101 /// Deprecated in favor of `languages`.
102 #[serde(default)]
103 language: Option<Arc<str>>,
104 /// The list of languages this language server should work with.
105 #[serde(default)]
106 languages: Vec<Arc<str>>,
107 #[serde(default)]
108 pub language_ids: HashMap<String, String>,
109 #[serde(default)]
110 pub code_action_kinds: Option<Vec<lsp::CodeActionKind>>,
111}
112
113impl LanguageServerManifestEntry {
114 /// Returns the list of languages for the language server.
115 ///
116 /// Prefer this over accessing the `language` or `languages` fields directly,
117 /// as we currently support both.
118 ///
119 /// We can replace this with just field access for the `languages` field once
120 /// we have removed `language`.
121 pub fn languages(&self) -> impl IntoIterator<Item = Arc<str>> + '_ {
122 let language = if self.languages.is_empty() {
123 self.language.clone()
124 } else {
125 None
126 };
127 self.languages.iter().cloned().chain(language)
128 }
129}
130
131impl ExtensionManifest {
132 pub async fn load(fs: Arc<dyn Fs>, extension_dir: &Path) -> Result<Self> {
133 let extension_name = extension_dir
134 .file_name()
135 .and_then(OsStr::to_str)
136 .ok_or_else(|| anyhow!("invalid extension name"))?;
137
138 let mut extension_manifest_path = extension_dir.join("extension.json");
139 if fs.is_file(&extension_manifest_path).await {
140 let manifest_content = fs
141 .load(&extension_manifest_path)
142 .await
143 .with_context(|| format!("failed to load {extension_name} extension.json"))?;
144 let manifest_json = serde_json::from_str::<OldExtensionManifest>(&manifest_content)
145 .with_context(|| {
146 format!("invalid extension.json for extension {extension_name}")
147 })?;
148
149 Ok(manifest_from_old_manifest(manifest_json, extension_name))
150 } else {
151 extension_manifest_path.set_extension("toml");
152 let manifest_content = fs
153 .load(&extension_manifest_path)
154 .await
155 .with_context(|| format!("failed to load {extension_name} extension.toml"))?;
156 toml::from_str(&manifest_content)
157 .with_context(|| format!("invalid extension.json for extension {extension_name}"))
158 }
159 }
160}
161
162fn manifest_from_old_manifest(
163 manifest_json: OldExtensionManifest,
164 extension_id: &str,
165) -> ExtensionManifest {
166 ExtensionManifest {
167 id: extension_id.into(),
168 name: manifest_json.name,
169 version: manifest_json.version,
170 description: manifest_json.description,
171 repository: manifest_json.repository,
172 authors: manifest_json.authors,
173 schema_version: SchemaVersion::ZERO,
174 lib: Default::default(),
175 themes: {
176 let mut themes = manifest_json.themes.into_values().collect::<Vec<_>>();
177 themes.sort();
178 themes.dedup();
179 themes
180 },
181 languages: {
182 let mut languages = manifest_json.languages.into_values().collect::<Vec<_>>();
183 languages.sort();
184 languages.dedup();
185 languages
186 },
187 grammars: manifest_json
188 .grammars
189 .into_keys()
190 .map(|grammar_name| (grammar_name, Default::default()))
191 .collect(),
192 language_servers: Default::default(),
193 }
194}