-
Notifications
You must be signed in to change notification settings - Fork 81
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(kotlin): add kotlin.rs with
impl Language for Kotlin and add `…
…PatternLanguage::Kotlin` fixes #570
- Loading branch information
1 parent
b5f1dba
commit 07f0922
Showing
7 changed files
with
165 additions
and
34 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,107 @@ | ||
| use crate::language::{fields_for_nodes, Field, MarzanoLanguage, NodeTypes, SortId, TSLanguage}; | ||
| use grit_util::Language; | ||
| use marzano_util::node_with_source::NodeWithSource; | ||
| use std::sync::OnceLock; | ||
|
|
||
| static NODE_TYPES_STRING: &str = | ||
| include_str!("../../../resources/node-types/kotlin-node-types.json"); | ||
| static NODE_TYPES: OnceLock<Vec<Vec<Field>>> = OnceLock::new(); | ||
| static LANGUAGE: OnceLock<TSLanguage> = OnceLock::new(); | ||
|
|
||
| #[cfg(not(feature = "builtin-parser"))] | ||
| fn language() -> TSLanguage { | ||
| unimplemented!( | ||
| "tree-sitter parser must be initialized before use when [builtin-parser] is off." | ||
| ) | ||
| } | ||
| #[cfg(feature = "builtin-parser")] | ||
| fn language() -> TSLanguage { | ||
| tree_sitter_kotlin::language().into() | ||
| } | ||
|
|
||
| #[derive(Debug, Clone, Copy)] | ||
| pub struct Kotlin { | ||
| node_types: &'static [Vec<Field>], | ||
| metavariable_sort: SortId, | ||
| comment_sorts: [SortId; 2], | ||
| language: &'static TSLanguage, | ||
| } | ||
|
|
||
| impl NodeTypes for Kotlin { | ||
| fn node_types(&self) -> &[Vec<Field>] { | ||
| self.node_types | ||
| } | ||
| } | ||
|
|
||
| impl Kotlin { | ||
| pub(crate) fn new(lang: Option<TSLanguage>) -> Self { | ||
| let language = LANGUAGE.get_or_init(|| lang.unwrap_or_else(language)); | ||
| let node_types = NODE_TYPES.get_or_init(|| fields_for_nodes(language, NODE_TYPES_STRING)); | ||
| let metavariable_sort = language.id_for_node_kind("grit_metavariable", true); | ||
| let comment_sorts = [ | ||
| language.id_for_node_kind("line_comment", true), | ||
| language.id_for_node_kind("multiline_comment", true), | ||
| ]; | ||
| Self { | ||
| node_types, | ||
| metavariable_sort, | ||
| comment_sorts, | ||
| language, | ||
| } | ||
| } | ||
| pub(crate) fn is_initialized() -> bool { | ||
| LANGUAGE.get().is_some() | ||
| } | ||
| } | ||
| impl Language for Kotlin { | ||
| use_marzano_delegate!(); | ||
|
|
||
| fn language_name(&self) -> &'static str { | ||
| "Kotlin" | ||
| } | ||
|
|
||
| fn snippet_context_strings(&self) -> &[(&'static str, &'static str)] { | ||
| &[ | ||
| ("", ""), | ||
| ("import ", ""), | ||
| ("val GRIT_VAR = ", ""), | ||
| ("const val GRIT_VAR = ", ""), | ||
| ("class GRIT_CLASS { ", " }"), | ||
| ("class GRIT_CLASS { ", " fun GRIT_FUNCTION() {} }"), | ||
| ("GRIT_FN(", ") {}"), | ||
| ("fun GRIT_FN(", ") {}"), | ||
| ("fun GRIT_FN(GRIT_ARG:", ") {}"), | ||
| ] | ||
| } | ||
| } | ||
|
|
||
| impl<'a> MarzanoLanguage<'a> for Kotlin { | ||
| fn get_ts_language(&self) -> &TSLanguage { | ||
| self.language | ||
| } | ||
|
|
||
| fn is_comment_sort(&self, id: SortId) -> bool { | ||
| self.comment_sorts.contains(&id) | ||
| } | ||
|
|
||
| fn metavariable_sort(&self) -> SortId { | ||
| self.metavariable_sort | ||
| } | ||
| } | ||
|
|
||
| #[cfg(test)] | ||
| mod tests { | ||
| use crate::language::nodes_from_indices; | ||
|
|
||
| use super::*; | ||
|
|
||
| #[test] | ||
| fn pair_snippet() { | ||
| let snippet = "import kotlin.math.PI"; | ||
| let lang = Kotlin::new(None); | ||
| let snippets = lang.parse_snippet_contexts(snippet); | ||
| let nodes = nodes_from_indices(&snippets); | ||
| println!("{:?}", nodes); | ||
| assert!(!nodes.is_empty()); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters