aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--crates/ide/src/completion/complete_postfix/format_like.rs82
1 files changed, 20 insertions, 62 deletions
diff --git a/crates/ide/src/completion/complete_postfix/format_like.rs b/crates/ide/src/completion/complete_postfix/format_like.rs
index 53cb3af39..b663ba171 100644
--- a/crates/ide/src/completion/complete_postfix/format_like.rs
+++ b/crates/ide/src/completion/complete_postfix/format_like.rs
@@ -20,6 +20,18 @@ use crate::completion::{
20}; 20};
21use syntax::ast::{self, AstToken}; 21use syntax::ast::{self, AstToken};
22 22
23/// Mapping ("postfix completion item" => "macro to use")
24static KINDS: &[(&str, &str)] = &[
25 ("fmt", "format!"),
26 ("panic", "panic!"),
27 ("println", "println!"),
28 ("logd", "log::debug!"),
29 ("logt", "log::trace!"),
30 ("logi", "log::info!"),
31 ("logw", "log::warn!"),
32 ("loge", "log::error!"),
33];
34
23pub(super) fn add_format_like_completions( 35pub(super) fn add_format_like_completions(
24 acc: &mut Completions, 36 acc: &mut Completions,
25 ctx: &CompletionContext, 37 ctx: &CompletionContext,
@@ -36,11 +48,10 @@ pub(super) fn add_format_like_completions(
36 let mut parser = FormatStrParser::new(input); 48 let mut parser = FormatStrParser::new(input);
37 49
38 if parser.parse().is_ok() { 50 if parser.parse().is_ok() {
39 for kind in PostfixKind::all_suggestions() { 51 for (label, macro_name) in KINDS {
40 let snippet = parser.into_suggestion(*kind); 52 let snippet = parser.into_suggestion(macro_name);
41 let (label, detail) = kind.into_description();
42 53
43 postfix_snippet(ctx, cap, &dot_receiver, label, detail, &snippet).add_to(acc); 54 postfix_snippet(ctx, cap, &dot_receiver, label, macro_name, &snippet).add_to(acc);
44 } 55 }
45 } 56 }
46} 57}
@@ -66,59 +77,6 @@ pub struct FormatStrParser {
66 parsed: bool, 77 parsed: bool,
67} 78}
68 79
69#[derive(Debug, Clone, Copy)]
70pub enum PostfixKind {
71 Format,
72 Panic,
73 Println,
74 LogDebug,
75 LogTrace,
76 LogInfo,
77 LogWarn,
78 LogError,
79}
80
81impl PostfixKind {
82 pub fn all_suggestions() -> &'static [PostfixKind] {
83 &[
84 Self::Format,
85 Self::Panic,
86 Self::Println,
87 Self::LogDebug,
88 Self::LogTrace,
89 Self::LogInfo,
90 Self::LogWarn,
91 Self::LogError,
92 ]
93 }
94
95 pub fn into_description(self) -> (&'static str, &'static str) {
96 match self {
97 Self::Format => ("fmt", "format!"),
98 Self::Panic => ("panic", "panic!"),
99 Self::Println => ("println", "println!"),
100 Self::LogDebug => ("logd", "log::debug!"),
101 Self::LogTrace => ("logt", "log::trace!"),
102 Self::LogInfo => ("logi", "log::info!"),
103 Self::LogWarn => ("logw", "log::warn!"),
104 Self::LogError => ("loge", "log::error!"),
105 }
106 }
107
108 pub fn into_macro_name(self) -> &'static str {
109 match self {
110 Self::Format => "format!",
111 Self::Panic => "panic!",
112 Self::Println => "println!",
113 Self::LogDebug => "log::debug!",
114 Self::LogTrace => "log::trace!",
115 Self::LogInfo => "log::info!",
116 Self::LogWarn => "log::warn!",
117 Self::LogError => "log::error!",
118 }
119 }
120}
121
122#[derive(Debug, Clone, Copy, PartialEq)] 80#[derive(Debug, Clone, Copy, PartialEq)]
123enum State { 81enum State {
124 NotExpr, 82 NotExpr,
@@ -235,11 +193,11 @@ impl FormatStrParser {
235 Ok(()) 193 Ok(())
236 } 194 }
237 195
238 pub fn into_suggestion(&self, kind: PostfixKind) -> String { 196 pub fn into_suggestion(&self, macro_name: &str) -> String {
239 assert!(self.parsed, "Attempt to get a suggestion from not parsed expression"); 197 assert!(self.parsed, "Attempt to get a suggestion from not parsed expression");
240 198
241 let expressions_as_string = self.extracted_expressions.join(", "); 199 let expressions_as_string = self.extracted_expressions.join(", ");
242 format!(r#"{}("{}", {})"#, kind.into_macro_name(), self.output, expressions_as_string) 200 format!(r#"{}("{}", {})"#, macro_name, self.output, expressions_as_string)
243 } 201 }
244} 202}
245 203
@@ -300,13 +258,13 @@ mod tests {
300 #[test] 258 #[test]
301 fn test_into_suggestion() { 259 fn test_into_suggestion() {
302 let test_vector = &[ 260 let test_vector = &[
303 (PostfixKind::Println, "{}", r#"println!("{}", $1)"#), 261 ("println!", "{}", r#"println!("{}", $1)"#),
304 ( 262 (
305 PostfixKind::LogInfo, 263 "log::info!",
306 "{} {expr} {} {2 + 2}", 264 "{} {expr} {} {2 + 2}",
307 r#"log::info!("{} {} {} {}", $1, expr, $2, 2 + 2)"#, 265 r#"log::info!("{} {} {} {}", $1, expr, $2, 2 + 2)"#,
308 ), 266 ),
309 (PostfixKind::Format, "{expr:?}", r#"format!("{:?}", expr)"#), 267 ("format!", "{expr:?}", r#"format!("{:?}", expr)"#),
310 ]; 268 ];
311 269
312 for (kind, input, output) in test_vector { 270 for (kind, input, output) in test_vector {