textbringer-tree-sitter 1.0.0

data/lib/textbringer/tree_sitter/node_maps/pascal.rb

@@ -0,0 +1,102 @@
+# frozen_string_literal: true
+
+module Textbringer
+  module TreeSitter
+    module NodeMaps
+      # Pascal の tree-sitter ノードタイプは kXxx 形式が多い
+      PASCAL_FEATURES = {
+        comment: %i[comment],
+        string: %i[string],
+        keyword: %i[
+          kBegin
+          kEnd
+          kIf
+          kThen
+          kElse
+          kFor
+          kTo
+          kDownto
+          kWhile
+          kRepeat
+          kUntil
+          kDo
+          kCase
+          kOf
+          kWith
+          kFunction
+          kProcedure
+          kProgram
+          kUnit
+          kInterface
+          kImplementation
+          kVar
+          kConst
+          kType
+          kArray
+          kRecord
+          kSet
+          kFile
+          kClass
+          kObject
+          kConstructor
+          kDestructor
+          kProperty
+          kInherited
+          kPrivate
+          kProtected
+          kPublic
+          kPublished
+          kVirtual
+          kOverride
+          kAbstract
+          kStatic
+          kForward
+          kExternal
+          kUses
+          kIn
+          kNil
+          kNot
+          kAnd
+          kOr
+          kXor
+          kDiv
+          kMod
+          kShl
+          kShr
+          kAs
+          kIs
+          kTry
+          kExcept
+          kFinally
+          kRaise
+          kOn
+          kGoto
+          kLabel
+          kExit
+          kBreak
+          kContinue
+        ],
+        number: %i[
+          integer
+          real
+          hex
+        ],
+        constant: %i[],
+        function_name: %i[],
+        variable: %i[identifier],
+        type: %i[type_identifier],
+        operator: %i[],
+        punctuation: %i[],
+        builtin: %i[
+          kTrue
+          kFalse
+        ],
+        property: %i[]
+      }.freeze
+
+      PASCAL = PASCAL_FEATURES.flat_map { |face, nodes|
+        nodes.map { |node| [node, face] }
+      }.to_h.freeze
+    end
+  end
+end

data/lib/textbringer/tree_sitter/node_maps/php.rb

@@ -0,0 +1,100 @@
+# frozen_string_literal: true
+
+module Textbringer
+  module TreeSitter
+    module NodeMaps
+      PHP_FEATURES = {
+        comment: %i[comment],
+        string: %i[
+          string
+          encapsed_string
+          heredoc
+          nowdoc
+          shell_command_expression
+        ],
+        keyword: %i[
+          if
+          else
+          elseif
+          endif
+          for
+          foreach
+          endforeach
+          while
+          endwhile
+          do
+          switch
+          case
+          default
+          endswitch
+          break
+          continue
+          return
+          throw
+          try
+          catch
+          finally
+          function
+          fn
+          class
+          interface
+          trait
+          extends
+          implements
+          namespace
+          use
+          as
+          const
+          public
+          private
+          protected
+          static
+          final
+          abstract
+          readonly
+          new
+          clone
+          instanceof
+          insteadof
+          global
+          echo
+          print
+          include
+          include_once
+          require
+          require_once
+          goto
+          yield
+          match
+          enum
+        ],
+        number: %i[
+          integer
+          float
+        ],
+        constant: %i[],
+        function_name: %i[],
+        variable: %i[
+          variable_name
+          name
+        ],
+        type: %i[
+          primitive_type
+          named_type
+        ],
+        operator: %i[],
+        punctuation: %i[],
+        builtin: %i[
+          true
+          false
+          null
+        ],
+        property: %i[]
+      }.freeze
+
+      PHP = PHP_FEATURES.flat_map { |face, nodes|
+        nodes.map { |node| [node, face] }
+      }.to_h.freeze
+    end
+  end
+end

data/lib/textbringer/tree_sitter/node_maps/python.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+module Textbringer
+  module TreeSitter
+    module NodeMaps
+      PYTHON_FEATURES = {
+        comment: %i[comment],
+        string: %i[
+          string
+          string_content
+          concatenated_string
+        ],
+        keyword: %i[
+          if
+          elif
+          else
+          for
+          while
+          break
+          continue
+          return
+          pass
+          raise
+          try
+          except
+          finally
+          with
+          as
+          def
+          class
+          lambda
+          import
+          from
+          global
+          nonlocal
+          assert
+          yield
+          del
+          in
+          not
+          and
+          or
+          is
+          async
+          await
+          match
+          case
+        ],
+        number: %i[
+          integer
+          float
+        ],
+        constant: %i[],
+        function_name: %i[],
+        variable: %i[identifier],
+        type: %i[type],
+        operator: %i[],
+        punctuation: %i[],
+        builtin: %i[
+          true
+          false
+          none
+        ],
+        property: %i[attribute]
+      }.freeze
+
+      PYTHON = PYTHON_FEATURES.flat_map { |face, nodes|
+        nodes.map { |node| [node, face] }
+      }.to_h.freeze
+    end
+  end
+end

data/lib/textbringer/tree_sitter/node_maps/ruby.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+module Textbringer
+  module TreeSitter
+    module NodeMaps
+      # Feature-based ノードマッピング
+      # Emacs の treesit-font-lock-rules 風に feature ごとに分類
+      RUBY_FEATURES = {
+        comment: %i[comment],
+        string: %i[
+          string_content
+          heredoc_content
+          simple_symbol
+          escape_sequence
+        ],
+        keyword: %i[
+          def
+          end
+          class
+          module
+          if
+          else
+          elsif
+          unless
+          case
+          when
+          then
+          while
+          until
+          for
+          do
+          break
+          next
+          redo
+          retry
+          return
+          yield
+          begin
+          rescue
+          raise
+          ensure
+          and
+          or
+          not
+          in
+          alias
+          undef
+          defined?
+          self
+          super
+          __FILE__
+          __LINE__
+          __ENCODING__
+        ],
+        number: %i[integer float complex rational],
+        constant: %i[constant],
+        function_name: %i[method_name],
+        variable: %i[
+          identifier
+          instance_variable
+          class_variable
+          global_variable
+        ],
+        type: %i[],
+        operator: %i[
+          binary
+          unary
+          assignment
+          operator
+        ],
+        punctuation: %i[],
+        builtin: %i[nil true false],
+        property: %i[]
+      }.freeze
+
+      # Feature → Face の展開
+      RUBY = RUBY_FEATURES.flat_map { |face, nodes|
+        nodes.map { |node| [node, face] }
+      }.to_h.freeze
+    end
+  end
+end

data/lib/textbringer/tree_sitter/node_maps/rust.rb

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+module Textbringer
+  module TreeSitter
+    module NodeMaps
+      RUST_FEATURES = {
+        comment: %i[
+          line_comment
+          block_comment
+        ],
+        string: %i[
+          string_literal
+          char_literal
+          raw_string_literal
+        ],
+        keyword: %i[
+          if
+          else
+          for
+          while
+          loop
+          break
+          continue
+          return
+          fn
+          let
+          mut
+          const
+          static
+          struct
+          enum
+          union
+          impl
+          trait
+          type
+          mod
+          use
+          pub
+          crate
+          self
+          super
+          as
+          in
+          ref
+          move
+          dyn
+          async
+          await
+          match
+          where
+          unsafe
+          extern
+          default
+        ],
+        number: %i[
+          integer_literal
+          float_literal
+        ],
+        constant: %i[],
+        function_name: %i[],
+        variable: %i[identifier],
+        type: %i[
+          type_identifier
+          primitive_type
+          generic_type
+        ],
+        operator: %i[],
+        punctuation: %i[],
+        builtin: %i[
+          true
+          false
+        ],
+        property: %i[field_identifier]
+      }.freeze
+
+      RUST = RUST_FEATURES.flat_map { |face, nodes|
+        nodes.map { |node| [node, face] }
+      }.to_h.freeze
+    end
+  end
+end

data/lib/textbringer/tree_sitter/node_maps/yaml.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module Textbringer
+  module TreeSitter
+    module NodeMaps
+      # YAML_LANG - Ruby の YAML モジュールと名前衝突を避けるため
+      YAML_FEATURES = {
+        comment: %i[comment],
+        string: %i[
+          string_scalar
+          double_quote_scalar
+          single_quote_scalar
+          block_scalar
+        ],
+        keyword: %i[
+          anchor
+          alias
+          tag
+        ],
+        number: %i[
+          integer_scalar
+          float_scalar
+        ],
+        constant: %i[],
+        function_name: %i[],
+        variable: %i[],
+        type: %i[],
+        operator: %i[],
+        punctuation: %i[],
+        builtin: %i[
+          boolean_scalar
+          null_scalar
+        ],
+        property: %i[
+          block_mapping_pair
+          flow_pair
+        ]
+      }.freeze
+
+      YAML_LANG = YAML_FEATURES.flat_map { |face, nodes|
+        nodes.map { |node| [node, face] }
+      }.to_h.freeze
+    end
+  end
+end

data/lib/textbringer/tree_sitter/node_maps.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require_relative "node_maps/ruby"
+require_relative "node_maps/hcl"
+require_relative "node_maps/bash"
+require_relative "node_maps/c"
+require_relative "node_maps/csharp"
+require_relative "node_maps/cobol"
+require_relative "node_maps/groovy"
+require_relative "node_maps/haml"
+require_relative "node_maps/html"
+require_relative "node_maps/java"
+require_relative "node_maps/javascript"
+require_relative "node_maps/json"
+require_relative "node_maps/pascal"
+require_relative "node_maps/php"
+require_relative "node_maps/python"
+require_relative "node_maps/rust"
+require_relative "node_maps/yaml"
+
+module Textbringer
+  module TreeSitter
+    module NodeMaps
+      class << self
+        def for(language)
+          # カスタムマップが登録されていれば、デフォルトとマージして返す
+          if @custom_maps&.key?(language)
+            default_map = default_maps[language]
+            if default_map
+              default_map.merge(@custom_maps[language])
+            else
+              @custom_maps[language]
+            end
+          else
+            default_maps[language]
+          end
+        end
+
+        def register(language, node_map)
+          @custom_maps ||= {}
+          @custom_maps[language] = node_map
+        end
+
+        def clear_custom_maps
+          @custom_maps = {}
+        end
+
+        def available_languages
+          (default_maps.keys + (@custom_maps&.keys || [])).uniq
+        end
+
+        private
+
+        def default_maps
+          {
+            ruby: RUBY,
+            hcl: HCL,
+            bash: BASH,
+            c: C,
+            csharp: CSHARP,
+            cobol: COBOL,
+            groovy: GROOVY,
+            haml: HAML,
+            html: HTML,
+            java: JAVA,
+            javascript: JAVASCRIPT,
+            json: JSON_LANG,
+            pascal: PASCAL,
+            php: PHP,
+            python: PYTHON,
+            rust: RUST,
+            yaml: YAML_LANG
+          }
+        end
+      end
+    end
+  end
+end

data/lib/textbringer/tree_sitter/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module Textbringer
+  module TreeSitter
+    VERSION = "1.0.0"
+  end
+end