dbtui 0.3.2

Terminal database client with Vim-style navigation
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

//! UI diagnostic type for rendering error underlines.
//! The diagnostic engine itself lives in `sql_engine::diagnostics`.

/// Severity level — mirrors `sql_engine::diagnostics::DiagnosticSeverity`.
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
pub enum Severity {
    Error,
    Warning,
    Info,
    Hint,
}

/// Source of the diagnostic — which pass produced it.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Source {
    Syntax,
    Semantic,
    Lint,
    Server,
}

impl Source {
    pub fn label(&self) -> &str {
        match self {
            Source::Syntax => "syntax",
            Source::Semantic => "semantic",
            Source::Lint => "lint",
            Source::Server => "server",
        }
    }
}

/// A single diagnostic (error/warning on a specific range).
#[derive(Debug, Clone)]
pub struct Diagnostic {
    pub row: usize,
    pub col_start: usize,
    pub col_end: usize,
    pub message: String,
    pub severity: Severity,
    pub source: Source,
}

impl Diagnostic {
    /// Convert from engine diagnostic to UI diagnostic.
    pub fn from_engine(d: crate::sql_engine::diagnostics::Diagnostic) -> Self {
        use crate::sql_engine::diagnostics::{DiagnosticSeverity, DiagnosticSource};
        Self {
            row: d.row,
            col_start: d.col_start,
            col_end: d.col_end,
            message: d.message,
            severity: match d.severity {
                DiagnosticSeverity::Error => Severity::Error,
                DiagnosticSeverity::Warning => Severity::Warning,
                DiagnosticSeverity::Info => Severity::Info,
                DiagnosticSeverity::Hint => Severity::Hint,
            },
            source: match d.source {
                DiagnosticSource::Syntax => Source::Syntax,
                DiagnosticSource::Semantic => Source::Semantic,
                DiagnosticSource::Lint => Source::Lint,
                DiagnosticSource::Server => Source::Server,
            },
        }
    }
}