1mod keymap_file;
2pub mod settings_file;
3pub mod watched_json;
4
5use anyhow::{bail, Result};
6use gpui::{
7 font_cache::{FamilyId, FontCache},
8 AssetSource,
9};
10use schemars::{
11 gen::{SchemaGenerator, SchemaSettings},
12 schema::{InstanceType, ObjectValidation, Schema, SchemaObject, SingleOrVec},
13 JsonSchema,
14};
15use serde::{de::DeserializeOwned, Deserialize, Serialize};
16use serde_json::Value;
17use sqlez::{
18 bindable::{Bind, Column, StaticColumnCount},
19 statement::Statement,
20};
21use std::{collections::HashMap, fmt::Write as _, num::NonZeroU32, str, sync::Arc};
22use theme::{Theme, ThemeRegistry};
23use tree_sitter::Query;
24use util::ResultExt as _;
25
26pub use keymap_file::{keymap_file_json_schema, KeymapFileContent};
27
28#[derive(Clone)]
29pub struct Settings {
30 pub buffer_font_family: FamilyId,
31 pub default_buffer_font_size: f32,
32 pub buffer_font_size: f32,
33 pub active_pane_magnification: f32,
34 pub cursor_blink: bool,
35 pub confirm_quit: bool,
36 pub hover_popover_enabled: bool,
37 pub show_completions_on_input: bool,
38 pub show_call_status_icon: bool,
39 pub vim_mode: bool,
40 pub autosave: Autosave,
41 pub default_dock_anchor: DockAnchor,
42 pub editor_defaults: EditorSettings,
43 pub editor_overrides: EditorSettings,
44 pub git: GitSettings,
45 pub git_overrides: GitSettings,
46 pub journal_defaults: JournalSettings,
47 pub journal_overrides: JournalSettings,
48 pub terminal_defaults: TerminalSettings,
49 pub terminal_overrides: TerminalSettings,
50 pub language_defaults: HashMap<Arc<str>, EditorSettings>,
51 pub language_overrides: HashMap<Arc<str>, EditorSettings>,
52 pub lsp: HashMap<Arc<str>, LspSettings>,
53 pub theme: Arc<Theme>,
54 pub telemetry_defaults: TelemetrySettings,
55 pub telemetry_overrides: TelemetrySettings,
56 pub auto_update: bool,
57}
58
59#[derive(Copy, Clone, Debug, Default, Serialize, Deserialize, JsonSchema)]
60pub struct TelemetrySettings {
61 diagnostics: Option<bool>,
62 metrics: Option<bool>,
63}
64
65impl TelemetrySettings {
66 pub fn metrics(&self) -> bool {
67 self.metrics.unwrap()
68 }
69
70 pub fn diagnostics(&self) -> bool {
71 self.diagnostics.unwrap()
72 }
73
74 pub fn set_metrics(&mut self, value: bool) {
75 self.metrics = Some(value);
76 }
77
78 pub fn set_diagnostics(&mut self, value: bool) {
79 self.diagnostics = Some(value);
80 }
81}
82
83#[derive(Copy, Clone, Debug, Default, Serialize, Deserialize, JsonSchema)]
84pub struct GitSettings {
85 pub git_gutter: Option<GitGutter>,
86 pub gutter_debounce: Option<u64>,
87}
88
89#[derive(Clone, Copy, Debug, Default, Serialize, Deserialize, JsonSchema)]
90#[serde(rename_all = "snake_case")]
91pub enum GitGutter {
92 #[default]
93 TrackedFiles,
94 Hide,
95}
96
97pub struct GitGutterConfig {}
98
99#[derive(Clone, Debug, Default, Serialize, Deserialize, JsonSchema)]
100pub struct EditorSettings {
101 pub tab_size: Option<NonZeroU32>,
102 pub hard_tabs: Option<bool>,
103 pub soft_wrap: Option<SoftWrap>,
104 pub preferred_line_length: Option<u32>,
105 pub format_on_save: Option<FormatOnSave>,
106 pub remove_trailing_whitespace_on_save: Option<bool>,
107 pub ensure_final_newline_on_save: Option<bool>,
108 pub formatter: Option<Formatter>,
109 pub enable_language_server: Option<bool>,
110}
111
112#[derive(Copy, Clone, Debug, Serialize, Deserialize, PartialEq, Eq, JsonSchema)]
113#[serde(rename_all = "snake_case")]
114pub enum SoftWrap {
115 None,
116 EditorWidth,
117 PreferredLineLength,
118}
119#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq, JsonSchema)]
120#[serde(rename_all = "snake_case")]
121pub enum FormatOnSave {
122 On,
123 Off,
124 LanguageServer,
125 External {
126 command: String,
127 arguments: Vec<String>,
128 },
129}
130
131#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq, JsonSchema)]
132#[serde(rename_all = "snake_case")]
133pub enum Formatter {
134 LanguageServer,
135 External {
136 command: String,
137 arguments: Vec<String>,
138 },
139}
140
141#[derive(Copy, Clone, Debug, Serialize, Deserialize, PartialEq, Eq, JsonSchema)]
142#[serde(rename_all = "snake_case")]
143pub enum Autosave {
144 Off,
145 AfterDelay { milliseconds: u64 },
146 OnFocusChange,
147 OnWindowChange,
148}
149
150#[derive(Clone, Debug, Serialize, Deserialize, JsonSchema)]
151pub struct JournalSettings {
152 pub path: Option<String>,
153 pub hour_format: Option<HourFormat>,
154}
155
156impl Default for JournalSettings {
157 fn default() -> Self {
158 Self {
159 path: Some("~".into()),
160 hour_format: Some(Default::default()),
161 }
162 }
163}
164
165#[derive(Clone, Debug, Serialize, Deserialize, JsonSchema)]
166#[serde(rename_all = "snake_case")]
167pub enum HourFormat {
168 Hour12,
169 Hour24,
170}
171
172impl Default for HourFormat {
173 fn default() -> Self {
174 Self::Hour12
175 }
176}
177
178#[derive(Clone, Debug, Default, Serialize, Deserialize, JsonSchema)]
179pub struct TerminalSettings {
180 pub shell: Option<Shell>,
181 pub working_directory: Option<WorkingDirectory>,
182 pub font_size: Option<f32>,
183 pub font_family: Option<String>,
184 pub env: Option<HashMap<String, String>>,
185 pub blinking: Option<TerminalBlink>,
186 pub alternate_scroll: Option<AlternateScroll>,
187 pub option_as_meta: Option<bool>,
188 pub copy_on_select: Option<bool>,
189}
190
191#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq, JsonSchema)]
192#[serde(rename_all = "snake_case")]
193pub enum TerminalBlink {
194 Off,
195 TerminalControlled,
196 On,
197}
198
199impl Default for TerminalBlink {
200 fn default() -> Self {
201 TerminalBlink::TerminalControlled
202 }
203}
204
205#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq, JsonSchema)]
206#[serde(rename_all = "snake_case")]
207pub enum Shell {
208 System,
209 Program(String),
210 WithArguments { program: String, args: Vec<String> },
211}
212
213impl Default for Shell {
214 fn default() -> Self {
215 Shell::System
216 }
217}
218
219#[derive(Clone, Copy, Debug, Serialize, Deserialize, PartialEq, Eq, JsonSchema)]
220#[serde(rename_all = "snake_case")]
221pub enum AlternateScroll {
222 On,
223 Off,
224}
225
226impl Default for AlternateScroll {
227 fn default() -> Self {
228 AlternateScroll::On
229 }
230}
231
232#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq, JsonSchema)]
233#[serde(rename_all = "snake_case")]
234pub enum WorkingDirectory {
235 CurrentProjectDirectory,
236 FirstProjectDirectory,
237 AlwaysHome,
238 Always { directory: String },
239}
240
241impl Default for WorkingDirectory {
242 fn default() -> Self {
243 Self::CurrentProjectDirectory
244 }
245}
246
247#[derive(PartialEq, Eq, Debug, Default, Copy, Clone, Hash, Serialize, Deserialize, JsonSchema)]
248#[serde(rename_all = "snake_case")]
249pub enum DockAnchor {
250 #[default]
251 Bottom,
252 Right,
253 Expanded,
254}
255
256impl StaticColumnCount for DockAnchor {}
257impl Bind for DockAnchor {
258 fn bind(&self, statement: &Statement, start_index: i32) -> anyhow::Result<i32> {
259 match self {
260 DockAnchor::Bottom => "Bottom",
261 DockAnchor::Right => "Right",
262 DockAnchor::Expanded => "Expanded",
263 }
264 .bind(statement, start_index)
265 }
266}
267
268impl Column for DockAnchor {
269 fn column(statement: &mut Statement, start_index: i32) -> anyhow::Result<(Self, i32)> {
270 String::column(statement, start_index).and_then(|(anchor_text, next_index)| {
271 Ok((
272 match anchor_text.as_ref() {
273 "Bottom" => DockAnchor::Bottom,
274 "Right" => DockAnchor::Right,
275 "Expanded" => DockAnchor::Expanded,
276 _ => bail!("Stored dock anchor is incorrect"),
277 },
278 next_index,
279 ))
280 })
281 }
282}
283
284#[derive(Clone, Debug, Default, Serialize, Deserialize, JsonSchema)]
285pub struct SettingsFileContent {
286 #[serde(default)]
287 pub projects_online_by_default: Option<bool>,
288 #[serde(default)]
289 pub buffer_font_family: Option<String>,
290 #[serde(default)]
291 pub buffer_font_size: Option<f32>,
292 #[serde(default)]
293 pub active_pane_magnification: Option<f32>,
294 #[serde(default)]
295 pub cursor_blink: Option<bool>,
296 #[serde(default)]
297 pub confirm_quit: Option<bool>,
298 #[serde(default)]
299 pub hover_popover_enabled: Option<bool>,
300 #[serde(default)]
301 pub show_completions_on_input: Option<bool>,
302 #[serde(default)]
303 pub show_call_status_icon: Option<bool>,
304 #[serde(default)]
305 pub vim_mode: Option<bool>,
306 #[serde(default)]
307 pub autosave: Option<Autosave>,
308 #[serde(default)]
309 pub default_dock_anchor: Option<DockAnchor>,
310 #[serde(flatten)]
311 pub editor: EditorSettings,
312 #[serde(default)]
313 pub journal: JournalSettings,
314 #[serde(default)]
315 pub terminal: TerminalSettings,
316 #[serde(default)]
317 pub git: Option<GitSettings>,
318 #[serde(default)]
319 #[serde(alias = "language_overrides")]
320 pub languages: HashMap<Arc<str>, EditorSettings>,
321 #[serde(default)]
322 pub lsp: HashMap<Arc<str>, LspSettings>,
323 #[serde(default)]
324 pub theme: Option<String>,
325 #[serde(default)]
326 pub telemetry: TelemetrySettings,
327 #[serde(default)]
328 pub auto_update: Option<bool>,
329}
330
331#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq, JsonSchema)]
332#[serde(rename_all = "snake_case")]
333pub struct LspSettings {
334 pub initialization_options: Option<Value>,
335}
336
337impl Settings {
338 /// Fill out the settings corresponding to the default.json file, overrides will be set later
339 pub fn defaults(
340 assets: impl AssetSource,
341 font_cache: &FontCache,
342 themes: &ThemeRegistry,
343 ) -> Self {
344 #[track_caller]
345 fn required<T>(value: Option<T>) -> Option<T> {
346 assert!(value.is_some(), "missing default setting value");
347 value
348 }
349
350 let defaults: SettingsFileContent = parse_json_with_comments(
351 str::from_utf8(assets.load("settings/default.json").unwrap().as_ref()).unwrap(),
352 )
353 .unwrap();
354
355 Self {
356 buffer_font_family: font_cache
357 .load_family(&[defaults.buffer_font_family.as_ref().unwrap()])
358 .unwrap(),
359 buffer_font_size: defaults.buffer_font_size.unwrap(),
360 active_pane_magnification: defaults.active_pane_magnification.unwrap(),
361 default_buffer_font_size: defaults.buffer_font_size.unwrap(),
362 confirm_quit: defaults.confirm_quit.unwrap(),
363 cursor_blink: defaults.cursor_blink.unwrap(),
364 hover_popover_enabled: defaults.hover_popover_enabled.unwrap(),
365 show_completions_on_input: defaults.show_completions_on_input.unwrap(),
366 show_call_status_icon: defaults.show_call_status_icon.unwrap(),
367 vim_mode: defaults.vim_mode.unwrap(),
368 autosave: defaults.autosave.unwrap(),
369 default_dock_anchor: defaults.default_dock_anchor.unwrap(),
370 editor_defaults: EditorSettings {
371 tab_size: required(defaults.editor.tab_size),
372 hard_tabs: required(defaults.editor.hard_tabs),
373 soft_wrap: required(defaults.editor.soft_wrap),
374 preferred_line_length: required(defaults.editor.preferred_line_length),
375 remove_trailing_whitespace_on_save: required(
376 defaults.editor.remove_trailing_whitespace_on_save,
377 ),
378 ensure_final_newline_on_save: required(
379 defaults.editor.ensure_final_newline_on_save,
380 ),
381 format_on_save: required(defaults.editor.format_on_save),
382 formatter: required(defaults.editor.formatter),
383 enable_language_server: required(defaults.editor.enable_language_server),
384 },
385 editor_overrides: Default::default(),
386 git: defaults.git.unwrap(),
387 git_overrides: Default::default(),
388 journal_defaults: defaults.journal,
389 journal_overrides: Default::default(),
390 terminal_defaults: defaults.terminal,
391 terminal_overrides: Default::default(),
392 language_defaults: defaults.languages,
393 language_overrides: Default::default(),
394 lsp: defaults.lsp.clone(),
395 theme: themes.get(&defaults.theme.unwrap()).unwrap(),
396 telemetry_defaults: defaults.telemetry,
397 telemetry_overrides: Default::default(),
398 auto_update: defaults.auto_update.unwrap(),
399 }
400 }
401
402 // Fill out the overrride and etc. settings from the user's settings.json
403 pub fn set_user_settings(
404 &mut self,
405 data: SettingsFileContent,
406 theme_registry: &ThemeRegistry,
407 font_cache: &FontCache,
408 ) {
409 if let Some(value) = &data.buffer_font_family {
410 if let Some(id) = font_cache.load_family(&[value]).log_err() {
411 self.buffer_font_family = id;
412 }
413 }
414 if let Some(value) = &data.theme {
415 if let Some(theme) = theme_registry.get(value).log_err() {
416 self.theme = theme;
417 }
418 }
419
420 merge(&mut self.buffer_font_size, data.buffer_font_size);
421 merge(
422 &mut self.active_pane_magnification,
423 data.active_pane_magnification,
424 );
425 merge(&mut self.default_buffer_font_size, data.buffer_font_size);
426 merge(&mut self.cursor_blink, data.cursor_blink);
427 merge(&mut self.confirm_quit, data.confirm_quit);
428 merge(&mut self.hover_popover_enabled, data.hover_popover_enabled);
429 merge(
430 &mut self.show_completions_on_input,
431 data.show_completions_on_input,
432 );
433 merge(&mut self.vim_mode, data.vim_mode);
434 merge(&mut self.autosave, data.autosave);
435 merge(&mut self.default_dock_anchor, data.default_dock_anchor);
436
437 // Ensure terminal font is loaded, so we can request it in terminal_element layout
438 if let Some(terminal_font) = &data.terminal.font_family {
439 font_cache.load_family(&[terminal_font]).log_err();
440 }
441
442 self.editor_overrides = data.editor;
443 self.git_overrides = data.git.unwrap_or_default();
444 self.journal_overrides = data.journal;
445 self.terminal_defaults.font_size = data.terminal.font_size;
446 self.terminal_overrides.copy_on_select = data.terminal.copy_on_select;
447 self.terminal_overrides = data.terminal;
448 self.language_overrides = data.languages;
449 self.telemetry_overrides = data.telemetry;
450 self.lsp = data.lsp;
451 merge(&mut self.auto_update, data.auto_update);
452 }
453
454 pub fn with_language_defaults(
455 mut self,
456 language_name: impl Into<Arc<str>>,
457 overrides: EditorSettings,
458 ) -> Self {
459 self.language_defaults
460 .insert(language_name.into(), overrides);
461 self
462 }
463
464 pub fn tab_size(&self, language: Option<&str>) -> NonZeroU32 {
465 self.language_setting(language, |settings| settings.tab_size)
466 }
467
468 pub fn hard_tabs(&self, language: Option<&str>) -> bool {
469 self.language_setting(language, |settings| settings.hard_tabs)
470 }
471
472 pub fn soft_wrap(&self, language: Option<&str>) -> SoftWrap {
473 self.language_setting(language, |settings| settings.soft_wrap)
474 }
475
476 pub fn preferred_line_length(&self, language: Option<&str>) -> u32 {
477 self.language_setting(language, |settings| settings.preferred_line_length)
478 }
479
480 pub fn remove_trailing_whitespace_on_save(&self, language: Option<&str>) -> bool {
481 self.language_setting(language, |settings| {
482 settings.remove_trailing_whitespace_on_save.clone()
483 })
484 }
485
486 pub fn ensure_final_newline_on_save(&self, language: Option<&str>) -> bool {
487 self.language_setting(language, |settings| {
488 settings.ensure_final_newline_on_save.clone()
489 })
490 }
491
492 pub fn format_on_save(&self, language: Option<&str>) -> FormatOnSave {
493 self.language_setting(language, |settings| settings.format_on_save.clone())
494 }
495
496 pub fn formatter(&self, language: Option<&str>) -> Formatter {
497 self.language_setting(language, |settings| settings.formatter.clone())
498 }
499
500 pub fn enable_language_server(&self, language: Option<&str>) -> bool {
501 self.language_setting(language, |settings| settings.enable_language_server)
502 }
503
504 fn language_setting<F, R>(&self, language: Option<&str>, f: F) -> R
505 where
506 F: Fn(&EditorSettings) -> Option<R>,
507 {
508 None.or_else(|| language.and_then(|l| self.language_overrides.get(l).and_then(&f)))
509 .or_else(|| f(&self.editor_overrides))
510 .or_else(|| language.and_then(|l| self.language_defaults.get(l).and_then(&f)))
511 .or_else(|| f(&self.editor_defaults))
512 .expect("missing default")
513 }
514
515 pub fn git_gutter(&self) -> GitGutter {
516 self.git_overrides.git_gutter.unwrap_or_else(|| {
517 self.git
518 .git_gutter
519 .expect("git_gutter should be some by setting setup")
520 })
521 }
522
523 fn terminal_setting<F, R: Default + Clone>(&self, f: F) -> R
524 where
525 F: Fn(&TerminalSettings) -> Option<&R>,
526 {
527 f(&self.terminal_overrides)
528 .or_else(|| f(&self.terminal_defaults))
529 .cloned()
530 .unwrap_or_else(|| R::default())
531 }
532
533 pub fn telemetry(&self) -> TelemetrySettings {
534 TelemetrySettings {
535 diagnostics: Some(self.telemetry_diagnostics()),
536 metrics: Some(self.telemetry_metrics()),
537 }
538 }
539
540 pub fn telemetry_diagnostics(&self) -> bool {
541 self.telemetry_overrides
542 .diagnostics
543 .or(self.telemetry_defaults.diagnostics)
544 .expect("missing default")
545 }
546
547 pub fn telemetry_metrics(&self) -> bool {
548 self.telemetry_overrides
549 .metrics
550 .or(self.telemetry_defaults.metrics)
551 .expect("missing default")
552 }
553
554 pub fn terminal_scroll(&self) -> AlternateScroll {
555 self.terminal_setting(|terminal_setting| terminal_setting.alternate_scroll.as_ref())
556 }
557
558 pub fn terminal_shell(&self) -> Shell {
559 self.terminal_setting(|terminal_setting| terminal_setting.shell.as_ref())
560 }
561
562 pub fn terminal_env(&self) -> HashMap<String, String> {
563 self.terminal_setting(|terminal_setting| terminal_setting.env.as_ref())
564 }
565
566 pub fn terminal_strategy(&self) -> WorkingDirectory {
567 self.terminal_setting(|terminal_setting| terminal_setting.working_directory.as_ref())
568 }
569
570 #[cfg(any(test, feature = "test-support"))]
571 pub fn test(cx: &gpui::AppContext) -> Settings {
572 Settings {
573 buffer_font_family: cx.font_cache().load_family(&["Monaco"]).unwrap(),
574 buffer_font_size: 14.,
575 active_pane_magnification: 1.,
576 default_buffer_font_size: 14.,
577 confirm_quit: false,
578 cursor_blink: true,
579 hover_popover_enabled: true,
580 show_completions_on_input: true,
581 show_call_status_icon: true,
582 vim_mode: false,
583 autosave: Autosave::Off,
584 default_dock_anchor: DockAnchor::Bottom,
585 editor_defaults: EditorSettings {
586 tab_size: Some(4.try_into().unwrap()),
587 hard_tabs: Some(false),
588 soft_wrap: Some(SoftWrap::None),
589 preferred_line_length: Some(80),
590 remove_trailing_whitespace_on_save: Some(true),
591 ensure_final_newline_on_save: Some(true),
592 format_on_save: Some(FormatOnSave::On),
593 formatter: Some(Formatter::LanguageServer),
594 enable_language_server: Some(true),
595 },
596 editor_overrides: Default::default(),
597 journal_defaults: Default::default(),
598 journal_overrides: Default::default(),
599 terminal_defaults: Default::default(),
600 terminal_overrides: Default::default(),
601 git: Default::default(),
602 git_overrides: Default::default(),
603 language_defaults: Default::default(),
604 language_overrides: Default::default(),
605 lsp: Default::default(),
606 theme: gpui::fonts::with_font_cache(cx.font_cache().clone(), Default::default),
607 telemetry_defaults: TelemetrySettings {
608 diagnostics: Some(true),
609 metrics: Some(true),
610 },
611 telemetry_overrides: Default::default(),
612 auto_update: true,
613 }
614 }
615
616 #[cfg(any(test, feature = "test-support"))]
617 pub fn test_async(cx: &mut gpui::TestAppContext) {
618 cx.update(|cx| {
619 let settings = Self::test(cx);
620 cx.set_global(settings);
621 });
622 }
623}
624
625pub fn settings_file_json_schema(
626 theme_names: Vec<String>,
627 language_names: &[String],
628) -> serde_json::Value {
629 let settings = SchemaSettings::draft07().with(|settings| {
630 settings.option_add_null_type = false;
631 });
632 let generator = SchemaGenerator::new(settings);
633 let mut root_schema = generator.into_root_schema_for::<SettingsFileContent>();
634
635 // Create a schema for a theme name.
636 let theme_name_schema = SchemaObject {
637 instance_type: Some(SingleOrVec::Single(Box::new(InstanceType::String))),
638 enum_values: Some(theme_names.into_iter().map(Value::String).collect()),
639 ..Default::default()
640 };
641
642 // Create a schema for a 'languages overrides' object, associating editor
643 // settings with specific langauges.
644 assert!(root_schema.definitions.contains_key("EditorSettings"));
645 let languages_object_schema = SchemaObject {
646 instance_type: Some(SingleOrVec::Single(Box::new(InstanceType::Object))),
647 object: Some(Box::new(ObjectValidation {
648 properties: language_names
649 .iter()
650 .map(|name| {
651 (
652 name.clone(),
653 Schema::new_ref("#/definitions/EditorSettings".into()),
654 )
655 })
656 .collect(),
657 ..Default::default()
658 })),
659 ..Default::default()
660 };
661
662 // Add these new schemas as definitions, and modify properties of the root
663 // schema to reference them.
664 root_schema.definitions.extend([
665 ("ThemeName".into(), theme_name_schema.into()),
666 ("Languages".into(), languages_object_schema.into()),
667 ]);
668 let root_schema_object = &mut root_schema.schema.object.as_mut().unwrap();
669
670 root_schema_object.properties.extend([
671 (
672 "theme".to_owned(),
673 Schema::new_ref("#/definitions/ThemeName".into()),
674 ),
675 (
676 "languages".to_owned(),
677 Schema::new_ref("#/definitions/Languages".into()),
678 ),
679 // For backward compatibility
680 (
681 "language_overrides".to_owned(),
682 Schema::new_ref("#/definitions/Languages".into()),
683 ),
684 ]);
685
686 serde_json::to_value(root_schema).unwrap()
687}
688
689/// Expects the key to be unquoted, and the value to be valid JSON
690/// (e.g. values should be unquoted for numbers and bools, quoted for strings)
691pub fn write_settings_key(
692 mut settings_content: String,
693 top_level_key: &str,
694 new_val: &str,
695) -> String {
696 let mut parser = tree_sitter::Parser::new();
697 parser.set_language(tree_sitter_json::language()).unwrap();
698 let tree = parser.parse(&settings_content, None).unwrap();
699
700 let mut cursor = tree_sitter::QueryCursor::new();
701
702 let query = Query::new(
703 tree_sitter_json::language(),
704 "
705 (document
706 (object
707 (pair
708 key: (string) @key
709 value: (_) @value)))
710 ",
711 )
712 .unwrap();
713
714 let mut first_key_start = None;
715 let mut existing_value_range = None;
716 let matches = cursor.matches(&query, tree.root_node(), settings_content.as_bytes());
717 for mat in matches {
718 if mat.captures.len() != 2 {
719 continue;
720 }
721
722 let key = mat.captures[0];
723 let value = mat.captures[1];
724
725 first_key_start.get_or_insert_with(|| key.node.start_byte());
726
727 if let Some(key_text) = settings_content.get(key.node.byte_range()) {
728 if key_text == format!("\"{top_level_key}\"") {
729 existing_value_range = Some(value.node.byte_range());
730 break;
731 }
732 }
733 }
734
735 match (first_key_start, existing_value_range) {
736 (None, None) => {
737 // No document, create a new object and overwrite
738 settings_content.clear();
739 write!(
740 settings_content,
741 "{{\n \"{}\": {new_val}\n}}\n",
742 top_level_key
743 )
744 .unwrap();
745 }
746
747 (_, Some(existing_value_range)) => {
748 // Existing theme key, overwrite
749 settings_content.replace_range(existing_value_range, &new_val);
750 }
751
752 (Some(first_key_start), None) => {
753 // No existing theme key, but other settings. Prepend new theme settings and
754 // match style of first key
755 let mut row = 0;
756 let mut column = 0;
757 for (ix, char) in settings_content.char_indices() {
758 if ix == first_key_start {
759 break;
760 }
761 if char == '\n' {
762 row += 1;
763 column = 0;
764 } else {
765 column += char.len_utf8();
766 }
767 }
768
769 let content = format!(r#""{top_level_key}": {new_val},"#);
770 settings_content.insert_str(first_key_start, &content);
771
772 if row > 0 {
773 settings_content.insert_str(
774 first_key_start + content.len(),
775 &format!("\n{:width$}", ' ', width = column),
776 )
777 } else {
778 settings_content.insert_str(first_key_start + content.len(), " ")
779 }
780 }
781 }
782
783 settings_content
784}
785
786fn merge<T: Copy>(target: &mut T, value: Option<T>) {
787 if let Some(value) = value {
788 *target = value;
789 }
790}
791
792pub fn parse_json_with_comments<T: DeserializeOwned>(content: &str) -> Result<T> {
793 Ok(serde_json::from_reader(
794 json_comments::CommentSettings::c_style().strip_comments(content.as_bytes()),
795 )?)
796}
797
798pub fn update_settings_file(
799 old_text: String,
800 old_file_content: SettingsFileContent,
801 update: impl FnOnce(&mut SettingsFileContent),
802) -> String {
803 let mut new_file_content = old_file_content.clone();
804 update(&mut new_file_content);
805
806 let old_json = to_json_object(old_file_content);
807 let new_json = to_json_object(new_file_content);
808
809 // Find changed fields
810 let mut diffs = vec![];
811 for (key, old_value) in old_json.iter() {
812 let new_value = new_json.get(key).unwrap();
813 if old_value != new_value {
814 if matches!(
815 new_value,
816 &Value::Null | &Value::Object(_) | &Value::Array(_)
817 ) {
818 unimplemented!("We only support updating basic values at the top level");
819 }
820
821 let new_json = serde_json::to_string_pretty(new_value)
822 .expect("Could not serialize new json field to string");
823
824 diffs.push((key, new_json));
825 }
826 }
827
828 let mut new_text = old_text;
829 for (key, new_value) in diffs {
830 new_text = write_settings_key(new_text, key, &new_value)
831 }
832 new_text
833}
834
835fn to_json_object(settings_file: SettingsFileContent) -> serde_json::Map<String, Value> {
836 let tmp = serde_json::to_value(settings_file).unwrap();
837 match tmp {
838 Value::Object(map) => map,
839 _ => unreachable!("SettingsFileContent represents a JSON map"),
840 }
841}
842
843#[cfg(test)]
844mod tests {
845 use super::*;
846 use unindent::Unindent;
847
848 fn assert_new_settings<S1: Into<String>, S2: Into<String>>(
849 old_json: S1,
850 update: fn(&mut SettingsFileContent),
851 expected_new_json: S2,
852 ) {
853 let old_json = old_json.into();
854 let old_content: SettingsFileContent = serde_json::from_str(&old_json).unwrap();
855 let new_json = update_settings_file(old_json, old_content, update);
856 assert_eq!(new_json, expected_new_json.into());
857 }
858
859 #[test]
860 fn test_update_telemetry_setting_multiple_fields() {
861 assert_new_settings(
862 r#"{
863 "telemetry": {
864 "metrics": false,
865 "diagnostics": false
866 }
867 }"#
868 .unindent(),
869 |settings| {
870 settings.telemetry.set_diagnostics(true);
871 settings.telemetry.set_metrics(true);
872 },
873 r#"{
874 "telemetry": {
875 "metrics": true,
876 "diagnostics": true
877 }
878 }"#
879 .unindent(),
880 );
881 }
882
883 #[test]
884 fn test_update_telemetry_setting_weird_formatting() {
885 assert_new_settings(
886 r#"{
887 "telemetry": { "metrics": false, "diagnostics": true }
888 }"#
889 .unindent(),
890 |settings| settings.telemetry.set_diagnostics(false),
891 r#"{
892 "telemetry": { "metrics": false, "diagnostics": false }
893 }"#
894 .unindent(),
895 );
896 }
897
898 #[test]
899 fn test_update_telemetry_setting_other_fields() {
900 assert_new_settings(
901 r#"{
902 "telemetry": {
903 "metrics": false,
904 "diagnostics": true
905 }
906 }"#
907 .unindent(),
908 |settings| settings.telemetry.set_diagnostics(false),
909 r#"{
910 "telemetry": {
911 "metrics": false,
912 "diagnostics": false
913 }
914 }"#
915 .unindent(),
916 );
917 }
918
919 #[test]
920 fn test_update_telemetry_setting_pre_existing() {
921 assert_new_settings(
922 r#"{
923 "telemetry": {
924 "diagnostics": true
925 }
926 }"#
927 .unindent(),
928 |settings| settings.telemetry.set_diagnostics(false),
929 r#"{
930 "telemetry": {
931 "diagnostics": false
932 }
933 }"#
934 .unindent(),
935 );
936 }
937
938 #[test]
939 fn test_update_telemetry_setting() {
940 assert_new_settings(
941 "{}",
942 |settings| settings.telemetry.set_diagnostics(true),
943 r#"{
944 "telemetry": {
945 "diagnostics": true
946 }
947 }"#
948 .unindent(),
949 );
950 }
951
952 #[test]
953 fn test_write_theme_into_settings_with_theme() {
954 let settings = r#"
955 {
956 "theme": "One Dark"
957 }
958 "#
959 .unindent();
960
961 let new_settings = r#"
962 {
963 "theme": "summerfruit-light"
964 }
965 "#
966 .unindent();
967
968 let settings_after_theme = write_settings_key(settings, "theme", "\"summerfruit-light\"");
969
970 assert_eq!(settings_after_theme, new_settings)
971 }
972
973 #[test]
974 fn test_write_theme_into_empty_settings() {
975 let settings = r#"
976 {
977 }
978 "#
979 .unindent();
980
981 let new_settings = r#"
982 {
983 "theme": "summerfruit-light"
984 }
985 "#
986 .unindent();
987
988 let settings_after_theme = write_settings_key(settings, "theme", "\"summerfruit-light\"");
989
990 assert_eq!(settings_after_theme, new_settings)
991 }
992
993 #[test]
994 fn test_write_theme_into_no_settings() {
995 let settings = "".to_string();
996
997 let new_settings = r#"
998 {
999 "theme": "summerfruit-light"
1000 }
1001 "#
1002 .unindent();
1003
1004 let settings_after_theme = write_settings_key(settings, "theme", "\"summerfruit-light\"");
1005
1006 assert_eq!(settings_after_theme, new_settings)
1007 }
1008
1009 #[test]
1010 fn test_write_theme_into_single_line_settings_without_theme() {
1011 let settings = r#"{ "a": "", "ok": true }"#.to_string();
1012 let new_settings = r#"{ "theme": "summerfruit-light", "a": "", "ok": true }"#;
1013
1014 let settings_after_theme = write_settings_key(settings, "theme", "\"summerfruit-light\"");
1015
1016 assert_eq!(settings_after_theme, new_settings)
1017 }
1018
1019 #[test]
1020 fn test_write_theme_pre_object_whitespace() {
1021 let settings = r#" { "a": "", "ok": true }"#.to_string();
1022 let new_settings = r#" { "theme": "summerfruit-light", "a": "", "ok": true }"#;
1023
1024 let settings_after_theme = write_settings_key(settings, "theme", "\"summerfruit-light\"");
1025
1026 assert_eq!(settings_after_theme, new_settings)
1027 }
1028
1029 #[test]
1030 fn test_write_theme_into_multi_line_settings_without_theme() {
1031 let settings = r#"
1032 {
1033 "a": "b"
1034 }
1035 "#
1036 .unindent();
1037
1038 let new_settings = r#"
1039 {
1040 "theme": "summerfruit-light",
1041 "a": "b"
1042 }
1043 "#
1044 .unindent();
1045
1046 let settings_after_theme = write_settings_key(settings, "theme", "\"summerfruit-light\"");
1047
1048 assert_eq!(settings_after_theme, new_settings)
1049 }
1050}