lutaml 0.9.43 → 0.10.0

data/lib/lutaml/uml_repository/queries/inheritance_query.rb

@@ -0,0 +1,259 @@
+# frozen_string_literal: true
+
+require_relative "base_query"
+require_relative "../../uml/qualified_name"
+
+module Lutaml
+  module UmlRepository
+    module Queries
+      # Query service for inheritance operations.
+      #
+      # Provides methods to navigate class inheritance hierarchies using the
+      # inheritance_graph index, which maps parent qualified names to arrays
+      # of child qualified names.
+      #
+      # @example Getting a class's parent
+      #   query = InheritanceQuery.new(document, indexes)
+      #   parent = query.supertype("ModelRoot::Child")
+      #
+      # @example Getting all ancestors
+      #   ancestors = query.ancestors("ModelRoot::GrandChild")
+      #   # => [Parent, GrandParent, ...]
+      #
+      # @example Getting descendants
+      #   descendants = query.descendants("ModelRoot::Parent", max_depth: 2)
+      class InheritanceQuery < BaseQuery
+        # Get the direct parent class (supertype).
+        #
+        # @param class_or_qname [Lutaml::Uml::Class, String] The class object
+        #   or qualified name string
+        # @return [Lutaml::Uml::Class, nil] The parent class,
+        # or nil if no parent
+        # @example
+        #   parent = query.supertype("ModelRoot::Child")
+        #   # Or
+        #   parent = query.supertype(child_class)
+        def supertype(class_or_qname) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity
+          klass = resolve_class(class_or_qname)
+          return nil unless klass
+          return nil unless klass.respond_to?(:generalization)
+          return nil unless klass.generalization
+
+          parent_name = extract_parent_name(klass.generalization)
+          return nil unless parent_name
+          # avoid self-references
+          return nil if parent_name == klass.name
+
+          # Try to find in qualified_names index
+          qname_string = resolve_qname(class_or_qname)
+          return nil unless qname_string
+
+          qname = Lutaml::Uml::QualifiedName.new(qname_string)
+          package_path = qname.package_path.to_s
+
+          # Try to resolve parent qualified name
+          parent_qname = resolve_parent_qualified_name(parent_name,
+                                                       package_path)
+          return nil unless parent_qname
+
+          indexes[:qualified_names][parent_qname]
+        end
+        alias_method :find_parent, :supertype
+
+        # Get direct child classes (subtypes).
+        #
+        # @param class_or_qname [Lutaml::Uml::Class, String] The class object
+        #   or qualified name string
+        # @param recursive [Boolean] Whether to include all descendants
+        #   (default: false)
+        # @return [Array] Array of child class objects
+        # @example Direct children only
+        #   children = query.subtypes("ModelRoot::Parent", recursive: false)
+        #
+        # @example All descendants
+        #   all_descendants = query.subtypes(
+        #   "ModelRoot::Parent", recursive: true)
+        def subtypes(class_or_qname, recursive: false)
+          qname_string = resolve_qname(class_or_qname)
+          return [] unless qname_string
+
+          if recursive
+            descendants(class_or_qname)
+          else
+            direct_subtypes(qname_string)
+          end
+        end
+        alias_method :find_children, :subtypes
+
+        # Get all ancestor classes up to the root.
+        #
+        # Returns ancestors in order from immediate parent to root.
+        #
+        # @param class_or_qname [Lutaml::Uml::Class, String] The class object
+        #   or qualified name string
+        # @return [Array] Array of ancestor class objects, ordered from nearest
+        #   to furthest
+        # @example
+        #   ancestors = query.ancestors("ModelRoot::GrandChild")
+        #   # => [Parent, GrandParent]
+        def ancestors(class_or_qname)
+          result = []
+          current = class_or_qname
+
+          loop do
+            parent = supertype(current)
+            break unless parent
+
+            result << parent
+            current = parent
+          end
+
+          result
+        end
+
+        # Get all ancestor classes up to the root by class XMI ID.
+        #
+        # @param class_xmi_id [String] The XMI ID of the class
+        # @return [Array] Array of ancestor class objects
+        def find_ancestors(class_xmi_id)
+          qualified_name, _klass = find_class_by_id(class_xmi_id)
+          return [] unless qualified_name
+
+          ancestors(qualified_name)
+        end
+
+        # Get all descendant classes.
+        #
+        # @param class_or_qname [Lutaml::Uml::Class, String] The class object
+        #   or qualified name string
+        # @param max_depth [Integer, nil] Maximum depth to traverse (nil for
+        #   unlimited)
+        # @return [Array] Array of descendant class objects
+        # @example
+        #   descendants = query.descendants("ModelRoot::Parent", max_depth: 2)
+        def descendants(class_or_qname, max_depth: nil)
+          qname_string = resolve_qname(class_or_qname)
+          return [] unless qname_string
+
+          collect_descendants(qname_string, max_depth, 0)
+        end
+
+        # Resolve a class or qualified name to a class object.
+        #
+        # @param class_or_qname [Lutaml::Uml::Class, String] The class object
+        #   or qualified name string
+        # @return [Lutaml::Uml::Class, nil] The class object,
+        # or nil if not found
+        def resolve_class(class_or_qname)
+          if class_or_qname.is_a?(String)
+            indexes[:qualified_names][class_or_qname]
+          else
+            class_or_qname
+          end
+        end
+
+        # Resolve a class or qualified name to a qualified name string.
+        #
+        # @param class_or_qname [Lutaml::Uml::Class, String] The class object
+        #   or qualified name string
+        # @return [String, nil] The qualified name string, or nil if not found
+        def resolve_qname(class_or_qname)
+          if class_or_qname.is_a?(String) &&
+              indexes[:qualified_names].key?(class_or_qname)
+            return class_or_qname
+          end
+
+          # Search for the class in the index
+          qname, _klass = indexes[:qualified_names].find do |_name, entity|
+            entity == class_or_qname
+          end
+
+          qname.nil? ? nil : qname
+        end
+
+        private
+
+        # Get direct subtypes of a class
+        #
+        # @param qname_string [String] Qualified name of the parent class
+        # @return [Array] Array of child class objects
+        def direct_subtypes(qname_string)
+          child_qnames = indexes[:inheritance_graph][qname_string]
+          return [] unless child_qnames
+
+          child_qnames.filter_map do |child_qname|
+            indexes[:qualified_names][child_qname]
+          end
+        end
+
+        # Recursively collect descendants
+        #
+        # @param qname_string [String] Qualified name of the parent class
+        # @param max_depth [Integer, nil] Maximum depth to traverse
+        # @param current_depth [Integer] Current depth
+        # @return [Array] Array of descendant class objects
+        def collect_descendants(qname_string, max_depth, current_depth) # rubocop:disable Metrics/MethodLength
+          return [] if max_depth && current_depth >= max_depth
+
+          children = direct_subtypes(qname_string)
+          result = children.dup
+
+          children.each do |child|
+            child_qname = resolve_qname(child)
+            next unless child_qname
+
+            grandchildren = collect_descendants(child_qname, max_depth,
+                                                current_depth + 1)
+            result.concat(grandchildren)
+          end
+
+          result
+        end
+
+        # Extract parent name from generalization object
+        #
+        # @param generalization [Lutaml::Uml::Generalization]
+        # Generalization object
+        # @return [String, nil] Parent class name
+        def extract_parent_name(generalization) # rubocop:disable Metrics/CyclomaticComplexity,Metrics/MethodLength
+          return nil unless generalization
+
+          # Check for general attribute (could be a string or object)
+          if generalization.respond_to?(:general)
+            parent = generalization.general
+            return parent.name if parent.respond_to?(:name)
+            return parent.to_s if parent
+          end
+
+          # Check for name attribute directly
+          if generalization.respond_to?(:name) && generalization.name
+            return generalization.name
+          end
+
+          nil
+        end
+
+        # Resolve a class name to its qualified name
+        #
+        # @param name [String] Class name to resolve
+        # @param current_package_path [String] Current package context
+        # @return [String, nil] Resolved qualified name
+        def resolve_parent_qualified_name(name, current_package_path)
+          # If name contains "::", it might already be qualified
+          return name if indexes[:qualified_names].key?(name)
+
+          # Try in current package
+          local_qname = "#{current_package_path}::#{name}"
+          return local_qname if indexes[:qualified_names].key?(local_qname)
+
+          # Try to find in all qualified names (simple name match)
+          indexes[:qualified_names].each_key do |qname|
+            return qname if qname.end_with?("::#{name}")
+          end
+
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/lutaml/uml_repository/queries/package_query.rb

@@ -0,0 +1,239 @@
+# frozen_string_literal: true
+
+require_relative "base_query"
+require_relative "../../uml/package_path"
+
+module Lutaml
+  module UmlRepository
+    module Queries
+      # Query service for package operations.
+      #
+      # Provides methods to find, list, and navigate package hierarchies
+      # using the package_paths index for efficient lookups.
+      #
+      # @example Finding a package by path
+      #   query = PackageQuery.new(document, indexes)
+      #   package = query.find_by_path("ModelRoot::i-UR::urf")
+      #
+      # @example Listing child packages
+      #   packages = query.list("ModelRoot::i-UR", recursive: false)
+      #
+      # @example Building a package tree
+      #   tree = query.tree("ModelRoot", max_depth: 2)
+      class PackageQuery < BaseQuery
+        # Find a single package by its path.
+        #
+        # Supports multiple search strategies:
+        # 1. Exact full path match (e.g., "ModelRoot::i-UR::uro")
+        # 2. Simple name match (e.g., "uro")
+        # 3. Partial path match (e.g., "i-UR::uro")
+        #
+        # @param path_string [String] The package path (e.g., "ModelRoot::i-UR")
+        # @return [Lutaml::Uml::Package, Lutaml::Uml::Document, nil] The package
+        #   or document, or nil if not found
+        # @example Exact path
+        #   package = query.find_by_path("ModelRoot::i-UR::urf")
+        #
+        # @example Simple name
+        #   package = query.find_by_path("urf")
+        #
+        # @example Partial path
+        #   package = query.find_by_path("i-UR::urf")
+        def find_by_path(path_string) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity
+          return nil if path_string.nil? || path_string.empty?
+
+          # Strategy 1: Try exact match first (most common case)
+          exact_match = indexes[:package_paths][path_string.to_s]
+          return exact_match if exact_match
+
+          # Strategy 2 & 3: Search for simple name or partial path match
+          search_segments = path_string.to_s.split("::")
+          matches = []
+
+          indexes[:package_paths].each do |full_path, package|
+            full_segments = full_path.to_s.split("::")
+
+            # Simple name match: last segment matches
+            if search_segments.length == 1
+              matches << package if full_segments.last == search_segments.first
+            # Partial path match: ends with search path
+            elsif full_segments.last(search_segments.length) == search_segments
+              matches << package
+            end
+          end
+
+          # Return single match, or nil if no match or ambiguous
+          matches.length == 1 ? matches.first : nil
+        end
+
+        # List packages at a specific path.
+        #
+        # @param parent_path_string [String] The parent package path
+        # @param recursive [Boolean] Whether to include nested packages
+        #   recursively (default: false)
+        # @return [Array<Lutaml::Uml::Package>] Array of packages
+        # @example Non-recursive listing
+        #   packages = query.list("ModelRoot::i-UR", recursive: false)
+        #   # Returns only direct children of i-UR
+        #
+        # @example Recursive listing
+        #   packages = query.list("ModelRoot", recursive: true)
+        #   # Returns all packages under ModelRoot
+        def list(parent_path_string, recursive: false) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity
+          return [] if parent_path_string.nil? || parent_path_string.empty?
+
+          parent_path = Lutaml::Uml::PackagePath.new(parent_path_string.to_s)
+          results = []
+
+          indexes[:package_paths].each do |path_string, package|
+            path = Lutaml::Uml::PackagePath.new(path_string)
+
+            # Skip the parent itself
+            next if path == parent_path
+
+            # Check if this path is under the parent
+            next unless path.starts_with?(parent_path)
+
+            if recursive
+              # Include all descendants
+              results << package
+            elsif path.depth == parent_path.depth + 1
+              # Include only direct children (depth = parent_depth + 1)
+              results << package
+            end
+          end
+
+          results
+        end
+
+        # Build a hierarchical tree structure of packages.
+        #
+        # @param root_path_string [String] The root package path to start from
+        # @param max_depth [Integer, nil] Maximum depth to traverse (nil for
+        #   unlimited)
+        # @return [Hash, nil] Tree structure with package information, or nil
+        #   if root not found
+        # @example
+        #   tree = query.tree("ModelRoot::i-UR", max_depth: 2)
+        #   # => {
+        #   #   name: "i-UR",
+        #   #   path: "ModelRoot::i-UR",
+        #   #   classes_count: 0,
+        #   #   diagrams_count: 0,
+        #   #   children: [
+        #   #     {
+        #   #       name: "urf",
+        #   #       path: "ModelRoot::i-UR::urf",
+        #   #       classes_count: 15,
+        #   #       diagrams_count: 2,
+        #   #       children: []
+        #   #     }
+        #   #   ]
+        #   # }
+        def tree(root_path_string, max_depth: nil)
+          return nil if root_path_string.nil? || root_path_string.empty?
+
+          root_package = find_by_path(root_path_string.to_s)
+          return nil unless root_package
+
+          build_tree_node(root_path_string.to_s, root_package, max_depth, 0)
+        end
+
+        private
+
+        # Build a tree node for a package
+        #
+        # @param path_string [String] The package path
+        # @param package [Lutaml::Uml::Package, Lutaml::Uml::Document] The
+        #   package or document object
+        # @param max_depth [Integer, nil] Maximum depth to traverse
+        # @param current_depth [Integer] Current depth in the tree
+        # @return [Hash] Tree node structure
+        def build_tree_node(path_string, package, max_depth, current_depth) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
+          path = Lutaml::Uml::PackagePath.new(path_string)
+
+          node = {
+            name: path.segments.last,
+            path: path_string,
+            classes_count: count_classes(package),
+            diagrams_count: count_diagrams(package),
+            children: [],
+          }
+
+          # Stop if we've reached max depth
+          if max_depth.nil? || current_depth < max_depth
+            # Find direct children
+            children = list(path_string, recursive: false)
+            children.each do |child_package|
+              child_path = find_package_path(child_package, path_string)
+              next unless child_path
+
+              child_node = build_tree_node(
+                child_path,
+                child_package,
+                max_depth,
+                current_depth + 1,
+              )
+              node[:children] << child_node
+            end
+          end
+
+          node
+        end
+
+        # Count classes in a package
+        #
+        # @param package [Lutaml::Uml::Package, Lutaml::Uml::Document]
+        # The package
+        # @return [Integer] Number of classes
+        def count_classes(package) # rubocop:disable Metrics/CyclomaticComplexity,Metrics/MethodLength
+          count = 0
+          count += if package.respond_to?(:classes) && package.classes
+                     package.classes.size
+                   end
+          count += if package.respond_to?(:data_types) && package.data_types
+                     package.data_types.size
+                   end
+          count += if package.respond_to?(:enums) && package.enums
+                     package.enums.size
+                   end
+          count
+        end
+
+        # Count diagrams in a package
+        #
+        # @param package [Lutaml::Uml::Package, Lutaml::Uml::Document]
+        # The package
+        # @return [Integer] Number of diagrams
+        def count_diagrams(package)
+          return 0 unless package.respond_to?(:diagrams) && package.diagrams
+
+          package.diagrams.size
+        end
+
+        # Find the path of a package under a parent path
+        #
+        # @param package [Lutaml::Uml::Package] The package to find
+        # @param parent_path_string [String] The parent package path
+        # @return [String, nil] The package path, or nil if not found
+        def find_package_path(package, parent_path_string) # rubocop:disable Metrics/MethodLength
+          parent_path = Lutaml::Uml::PackagePath.new(parent_path_string)
+
+          # Search for this package in the index
+          indexes[:package_paths].each do |path_string, indexed_package|
+            next unless indexed_package == package
+
+            path = Lutaml::Uml::PackagePath.new(path_string)
+            # Check if it's a direct child of parent_path
+            if path.starts_with?(parent_path) &&
+                path.depth == parent_path.depth + 1
+              return path_string
+            end
+          end
+
+          nil
+        end
+      end
+    end
+  end
+end