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}