sqlparser/dialect/
sqlite.rs

1// Licensed under the Apache License, Version 2.0 (the "License");
2// you may not use this file except in compliance with the License.
3// You may obtain a copy of the License at
4//
5// http://www.apache.org/licenses/LICENSE-2.0
6//
7// Unless required by applicable law or agreed to in writing, software
8// distributed under the License is distributed on an "AS IS" BASIS,
9// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10// See the License for the specific language governing permissions and
11// limitations under the License.
12
13use crate::ast::Statement;
14use crate::dialect::Dialect;
15use crate::keywords::Keyword;
16use crate::parser::{Parser, ParserError};
17
18#[derive(Debug)]
19pub struct SQLiteDialect {}
20
21impl Dialect for SQLiteDialect {
22    // see https://www.sqlite.org/lang_keywords.html
23    // parse `...`, [...] and "..." as identifier
24    // TODO: support depending on the context tread '...' as identifier too.
25    fn is_delimited_identifier_start(&self, ch: char) -> bool {
26        ch == '`' || ch == '"' || ch == '['
27    }
28
29    fn is_identifier_start(&self, ch: char) -> bool {
30        // See https://www.sqlite.org/draft/tokenreq.html
31        ch.is_ascii_lowercase()
32            || ch.is_ascii_uppercase()
33            || ch == '_'
34            || ch == '$'
35            || ('\u{007f}'..='\u{ffff}').contains(&ch)
36    }
37
38    fn is_identifier_part(&self, ch: char) -> bool {
39        self.is_identifier_start(ch) || ch.is_ascii_digit()
40    }
41
42    fn parse_statement(&self, parser: &mut Parser) -> Option<Result<Statement, ParserError>> {
43        if parser.parse_keyword(Keyword::REPLACE) {
44            parser.prev_token();
45            Some(parser.parse_insert())
46        } else {
47            None
48        }
49    }
50}