yerba 0.4.2

YAML Editing and Refactoring with Better Accuracy
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

use clap::ValueEnum;
use yaml_parser::SyntaxKind;

#[derive(Debug, Clone, PartialEq, ValueEnum)]
pub enum KeyStyle {
  /// Unquoted key (host:)
  Plain,
  /// Single-quoted key ('host':)
  Single,
  /// Double-quoted key ("host":)
  Double,
}

impl std::str::FromStr for KeyStyle {
  type Err = String;

  fn from_str(string: &str) -> Result<Self, Self::Err> {
    match string {
      "plain" => Ok(KeyStyle::Plain),
      "single" | "single-quoted" => Ok(KeyStyle::Single),
      "double" | "double-quoted" => Ok(KeyStyle::Double),
      _ => Err(format!("unknown key style: '{}'. Valid options: plain, single, double", string)),
    }
  }
}

impl KeyStyle {
  pub(crate) fn to_syntax_kind(&self) -> SyntaxKind {
    match self {
      KeyStyle::Plain => SyntaxKind::PLAIN_SCALAR,
      KeyStyle::Single => SyntaxKind::SINGLE_QUOTED_SCALAR,
      KeyStyle::Double => SyntaxKind::DOUBLE_QUOTED_SCALAR,
    }
  }
}

#[derive(Debug, Clone, PartialEq, ValueEnum)]
pub enum QuoteStyle {
  /// Unquoted value (host: localhost)
  Plain,
  /// Single-quoted value (host: 'localhost')
  #[value(alias = "single-quoted")]
  Single,
  /// Double-quoted value (host: "localhost")
  #[value(alias = "double-quoted")]
  Double,
  /// Literal block scalar, strip trailing newline (|-)
  #[value(alias = "block-literal", alias = "|-")]
  Literal,
  /// Literal block scalar, keep one trailing newline (|)
  #[value(alias = "|")]
  LiteralClip,
  /// Literal block scalar, keep all trailing newlines (|+)
  #[value(alias = "|+")]
  LiteralKeep,
  /// Folded block scalar, strip trailing newline (>-)
  #[value(alias = "block-folded", alias = ">-")]
  Folded,
  /// Folded block scalar, keep one trailing newline (>)
  #[value(alias = ">")]
  FoldedClip,
  /// Folded block scalar, keep all trailing newlines (>+)
  #[value(alias = ">+")]
  FoldedKeep,
}

impl std::str::FromStr for QuoteStyle {
  type Err = String;

  fn from_str(string: &str) -> Result<Self, Self::Err> {
    let normalized = string.replace('_', "-");

    match normalized.as_str() {
      "plain" => Ok(QuoteStyle::Plain),
      "single" | "single-quoted" => Ok(QuoteStyle::Single),
      "double" | "double-quoted" => Ok(QuoteStyle::Double),
      "literal" | "block-literal" | "|-" => Ok(QuoteStyle::Literal),
      "literal-clip" | "|" => Ok(QuoteStyle::LiteralClip),
      "literal-keep" | "|+" => Ok(QuoteStyle::LiteralKeep),
      "folded" | "block-folded" | ">-" => Ok(QuoteStyle::Folded),
      "folded-clip" | ">" => Ok(QuoteStyle::FoldedClip),
      "folded-keep" | ">+" => Ok(QuoteStyle::FoldedKeep),
      _ => Err(format!("unknown quote style: '{}'", string)),
    }
  }
}

impl QuoteStyle {
  pub(crate) fn to_syntax_kind(&self) -> SyntaxKind {
    match self {
      QuoteStyle::Plain => SyntaxKind::PLAIN_SCALAR,
      QuoteStyle::Single => SyntaxKind::SINGLE_QUOTED_SCALAR,
      QuoteStyle::Double => SyntaxKind::DOUBLE_QUOTED_SCALAR,
      QuoteStyle::Literal | QuoteStyle::LiteralClip | QuoteStyle::LiteralKeep | QuoteStyle::Folded | QuoteStyle::FoldedClip | QuoteStyle::FoldedKeep => {
        SyntaxKind::BLOCK_SCALAR_TEXT
      }
    }
  }

  pub(crate) fn is_block_scalar(&self) -> bool {
    matches!(
      self,
      QuoteStyle::Literal | QuoteStyle::LiteralClip | QuoteStyle::LiteralKeep | QuoteStyle::Folded | QuoteStyle::FoldedClip | QuoteStyle::FoldedKeep
    )
  }

  pub(crate) fn block_header(&self) -> &'static str {
    match self {
      QuoteStyle::Literal => "|-",
      QuoteStyle::LiteralClip => "|",
      QuoteStyle::LiteralKeep => "|+",
      QuoteStyle::Folded => ">-",
      QuoteStyle::FoldedClip => ">",
      QuoteStyle::FoldedKeep => ">+",
      _ => "",
    }
  }
}