taurus 0.1.0

data/lib/taurus/xml/document.rb

@@ -0,0 +1,202 @@
+# frozen_string_literal: true
+
+require "ffi"
+
+class Taurus::XML::Document
+  attr_reader :c_ptr, :wrapper_cache
+
+  # @api private
+  # Internal flag container shared between the Document instance and its
+  # GC finalizer. Using a one-element Array because Procs close over
+  # variables by reference — mutating freed[0] is visible from both
+  # the explicit `free` path and the finalizer. This eliminates the
+  # double-free that FFI::AutoPointer's release proc caused when
+  # `Document#free` was called explicitly and then GC ran.
+  Freed = Struct.new(:state)  # state: :alive | :freed
+
+  def initialize(c_ptr = nil, freed = Freed.new(:alive))
+    @c_ptr = c_ptr
+    @freed = freed
+    # Per-document weak-ref cache for Node wrappers, keyed on c_ptr
+    # address. Eliminates re-allocation when the same node is accessed
+    # repeatedly (e.g. via children, siblings, multiple xpath calls).
+    # Dies with the Document — no stale entries pointing at freed memory.
+    @wrapper_cache = ObjectSpace::WeakMap.new
+  end
+
+  def self.parse(xml_or_io)
+    xml = xml_or_io.respond_to?(:read) ? xml_or_io.read : xml_or_io.to_s
+    if xml.empty?
+      raise Taurus::XML::ParseError, "empty input"
+    end
+    status_ptr = ::FFI::MemoryPointer.new(:int)
+    raw = Taurus::XML::FFI.taurus_parse_string(xml, xml.bytesize, status_ptr)
+    if raw.null?
+      status = status_ptr.read_int
+      raise Taurus::XML::ParseError,
+        "taurus_parse_string failed (status=#{status})"
+    end
+    wrap(raw)
+  end
+
+  def self.parse_file(path)
+    status_ptr = ::FFI::MemoryPointer.new(:int)
+    raw = Taurus::XML::FFI.taurus_parse_file(path, status_ptr)
+    if raw.null?
+      status = status_ptr.read_int
+      raise Taurus::XML::ParseError,
+        "taurus_parse_file failed (status=#{status})"
+    end
+    wrap(raw)
+  end
+
+  # Convert a raw TaurusDocument pointer into a Ruby Document with safe
+  # GC lifetime management. The finalizer captures the raw address
+  # integer (not the Document or Pointer object — those would prevent
+  # GC) and shares a one-shot flag with the instance so explicit
+  # `#free` and the GC finalizer can never both call
+  # `taurus_document_free` on the same address.
+  def self.wrap(raw_address)
+    addr = raw_address.is_a?(::FFI::Pointer) ? raw_address.address : raw_address
+    ptr = ::FFI::Pointer.new(addr)
+    freed = Freed.new(:alive)
+    doc = new(ptr, freed)
+    ObjectSpace.define_finalizer(doc, finalizer(addr, freed))
+    doc
+  end
+
+  def self.finalizer(address, freed)
+    proc do
+      next if freed.state == :freed
+      freed.state = :freed
+      Taurus::XML::FFI.taurus_document_free(::FFI::Pointer.new(address))
+    end
+  end
+  private_class_method :finalizer
+
+  def root
+    raise Taurus::XML::UseAfterFreeError if @freed.state == :freed
+    return nil if @c_ptr.nil?
+    ptr = Taurus::XML::FFI.taurus_document_root(@c_ptr)
+    return nil if ptr.null?
+    Taurus::XML::Element.new(ptr, self)
+  end
+
+  def create_element(name)
+    ptr = Taurus::XML::FFI.taurus_element_create(@c_ptr, name)
+    raise Taurus::XML::Error, "taurus_element_create failed" if ptr.null?
+    Taurus::XML::Element.new(ptr, self)
+  end
+
+  def create_text_node(content)
+    ptr = Taurus::XML::FFI.taurus_text_node_create(@c_ptr, content.to_s)
+    raise Taurus::XML::Error, "taurus_text_node_create failed" if ptr.null?
+    Taurus::XML::Text.new(ptr, self)
+  end
+
+  def create_comment(content)
+    ptr = Taurus::XML::FFI.taurus_comment_node_create(@c_ptr, content.to_s)
+    raise Taurus::XML::Error, "taurus_comment_node_create failed" if ptr.null?
+    Taurus::XML::Comment.new(ptr, self)
+  end
+
+  def create_cdata(content)
+    ptr = Taurus::XML::FFI.taurus_cdata_node_create(@c_ptr, content.to_s)
+    raise Taurus::XML::Error, "taurus_cdata_node_create failed" if ptr.null?
+    Taurus::XML::CDATA.new(ptr, self)
+  end
+
+  def create_processing_instruction(target, data = "")
+    ptr = Taurus::XML::FFI.taurus_pi_node_create(@c_ptr, target.to_s, data.to_s)
+    raise Taurus::XML::Error, "taurus_pi_node_create failed" if ptr.null?
+    Taurus::XML::ProcessingInstruction.new(ptr, self)
+  end
+
+  def fragment(markup)
+    Taurus::XML::DocumentFragment.parse(markup, self)
+  end
+
+  def dup
+    raw = Taurus::XML::FFI.taurus_document_copy(@c_ptr)
+    raise Taurus::XML::Error, "taurus_document_copy failed" if raw.null?
+    self.class.wrap(raw)
+  end
+  alias_method :clone, :dup
+
+  def doctype
+    ptr = Taurus::XML::FFI.taurus_document_internal_subset(@c_ptr)
+    return nil if ptr.null?
+    Taurus::XML::DocType.new(ptr, self)
+  end
+  alias_method :internal_subset, :doctype
+
+  def to_xml(indent: 0, no_decl: false, encoding: nil)
+    raise Taurus::XML::UseAfterFreeError if @freed.state == :freed
+    return "" if @c_ptr.nil?
+    opts, enc_ptr = build_serialize_options(indent: indent, no_decl: no_decl, encoding: encoding)
+    str_ptr = Taurus::XML::FFI.taurus_document_serialize(@c_ptr, opts.pointer)
+    return "" if str_ptr.null?
+    str_ptr.read_string.tap { |s| Taurus::XML::FFI.taurus_free_string(str_ptr) }
+  end
+  alias_method :to_s, :to_xml
+  alias_method :serialize, :to_xml
+
+  def save(path, **opts)
+    opts_struct, enc_ptr = build_serialize_options(
+      indent: opts.fetch(:indent, 0),
+      no_decl: opts.fetch(:no_decl, false),
+      encoding: opts[:encoding])
+    status = Taurus::XML::FFI.taurus_document_save_file(@c_ptr, path, opts_struct.pointer)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    self
+  end
+
+  def canonicalize(version = Taurus::XML::FFI::C14N_1_0,
+                   inclusive_namespaces = nil,
+                   with_comments: false,
+                   exclusive: false,
+                   mode: nil)
+    raise Taurus::XML::UseAfterFreeError if @freed.state == :freed
+    return "" if @c_ptr.nil?
+    resolved_mode = mode || (exclusive ? Taurus::XML::FFI::C14N_MODE_EXCLUSIVE
+                                       : Taurus::XML::FFI::C14N_MODE_CANONICAL)
+    ns_ptr, _anchor = Taurus::XML.c14n_build_ns_pointer(inclusive_namespaces)
+    flags = with_comments ? 1 : 0
+    str_ptr = Taurus::XML::FFI.taurus_c14n_canonicalize_ex(
+      @c_ptr, version, resolved_mode, ns_ptr, flags)
+    return "" if str_ptr.null?
+    str_ptr.read_string.tap { Taurus::XML::FFI.taurus_free_string(str_ptr) }
+  end
+  alias_method :c14n, :canonicalize
+
+  def free
+    return if @freed.state == :freed
+    @freed.state = :freed
+    Taurus::XML::FFI.taurus_document_free(@c_ptr) unless @c_ptr.nil?
+    @c_ptr = nil
+  end
+
+  def name; "document"; end
+  def document; self; end
+  def encoding
+    return nil if @c_ptr.nil?
+    Taurus::XML::FFI.taurus_document_encoding(@c_ptr)
+  end
+
+  private
+
+  def build_serialize_options(indent:, no_decl:, encoding:)
+    opts = Taurus::XML::FFI::SerializeOptions.new
+    opts[:indent] = indent.to_i
+    opts[:xml_declaration] = no_decl ? 0 : 1
+    enc_ptr = nil
+    if encoding
+      enc_ptr = ::FFI::MemoryPointer.from_string(encoding.to_s)
+      opts[:encoding] = enc_ptr
+    end
+    [opts, enc_ptr]
+  end
+
+  include Taurus::XML::Searchable
+end

data/lib/taurus/xml/document_fragment.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require "ffi"
+
+# Wraps the synthetic "#document-fragment" element returned by
+# taurus_parse_fragment. Children of this fragment are the parsed nodes;
+# the fragment itself isn't part of any document tree but borrows its
+# document's lifetime.
+class Taurus::XML::DocumentFragment
+  attr_reader :document, :c_ptr
+
+  def initialize(document, c_ptr)
+    @document = document
+    @c_ptr = c_ptr
+  end
+
+  def self.parse(xml, document)
+    status_ptr = ::FFI::MemoryPointer.new(:int)
+    raw = Taurus::XML::FFI.taurus_parse_fragment(
+      xml.to_s, xml.to_s.bytesize, document.c_ptr, status_ptr)
+    if raw.null?
+      raise Taurus::XML::ParseError,
+        "taurus_parse_fragment failed (status=#{status_ptr.read_int})"
+    end
+    new(document, raw)
+  end
+
+  def children
+    count = Taurus::XML::FFI.taurus_element_child_count(@c_ptr)
+    nodes = []
+    ptr = Taurus::XML::FFI.taurus_node_first_child(@c_ptr)
+    until ptr.nil? || ptr.null?
+      nodes << Taurus::XML::Node.wrap(ptr, @document)
+      ptr = Taurus::XML::FFI.taurus_node_next_sibling(ptr)
+    end
+    Taurus::XML::NodeSet.new(@document, nodes)
+  end
+
+  def name
+    "#document-fragment"
+  end
+end

data/lib/taurus/xml/element.rb

@@ -0,0 +1,278 @@
+# frozen_string_literal: true
+
+class Taurus::XML::Element < Taurus::XML::Node
+  def name
+    Taurus::XML::FFI.taurus_element_name(@c_ptr)
+  end
+  alias_method :node_name, :name
+
+  def name=(new_name)
+    status = Taurus::XML::FFI.taurus_element_set_name(@c_ptr, new_name)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    new_name
+  end
+  alias_method :node_name=, :name=
+
+  def content
+    Taurus::XML::FFI.taurus_element_text(@c_ptr)
+  end
+
+  def content=(new_content)
+    status = Taurus::XML::FFI.taurus_element_set_text(@c_ptr, new_content.to_s)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    new_content
+  end
+
+  def [](key)
+    Taurus::XML::FFI.taurus_element_attribute(@c_ptr, key.to_s)
+  end
+  alias_method :attr, :[]
+  alias_method :get_attribute, :[]
+
+  def []=(key, value)
+    status = Taurus::XML::FFI.taurus_element_set_attribute(@c_ptr, key.to_s, value.to_s)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    value
+  end
+  alias_method :set_attribute, :[]=
+
+  def key?(name)
+    !Taurus::XML::FFI.taurus_element_attribute(@c_ptr, name.to_s).nil?
+  end
+  alias_method :has_attribute?, :key?
+
+  def remove_attribute(name)
+    status = Taurus::XML::FFI.taurus_element_remove_attribute(@c_ptr, name.to_s)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    self
+  end
+  alias_method :delete, :remove_attribute
+
+  def keys
+    count = Taurus::XML::FFI.taurus_element_attribute_count(@c_ptr)
+    count.times.map { |i| Taurus::XML::FFI.taurus_element_attribute_name_at(@c_ptr, i) }
+  end
+
+  def values
+    count = Taurus::XML::FFI.taurus_element_attribute_count(@c_ptr)
+    count.times.map { |i| Taurus::XML::FFI.taurus_element_attribute_value_at(@c_ptr, i) }
+  end
+
+  def attributes
+    count = Taurus::XML::FFI.taurus_element_attribute_count(@c_ptr)
+    result = {}
+    count.times do |i|
+      name = Taurus::XML::FFI.taurus_element_attribute_name_at(@c_ptr, i)
+      value = Taurus::XML::FFI.taurus_element_attribute_value_at(@c_ptr, i)
+      result[name] = Taurus::XML::Attr.new(name, value, self)
+    end
+    result
+  end
+
+  def attribute_nodes
+    count = Taurus::XML::FFI.taurus_element_attribute_count(@c_ptr)
+    count.times.map do |i|
+      name = Taurus::XML::FFI.taurus_element_attribute_name_at(@c_ptr, i)
+      value = Taurus::XML::FFI.taurus_element_attribute_value_at(@c_ptr, i)
+      Taurus::XML::Attr.new(name, value, self)
+    end
+  end
+
+  def add_child(node)
+    status = Taurus::XML::FFI.taurus_element_append_child(@c_ptr, node.c_ptr)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    node
+  end
+  alias_method :<<, :add_child
+
+  def prepend_child(node)
+    status = Taurus::XML::FFI.taurus_element_prepend_child(@c_ptr, node.c_ptr)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    node
+  end
+
+  def add_next_sibling(node)
+    status = Taurus::XML::FFI.taurus_element_insert_after(@c_ptr, node.c_ptr)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    node
+  end
+
+  def add_previous_sibling(node)
+    status = Taurus::XML::FFI.taurus_element_insert_before(@c_ptr, node.c_ptr)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    node
+  end
+
+  def remove_child(node)
+    status = Taurus::XML::FFI.taurus_element_remove_child(@c_ptr, node.c_ptr)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    node
+  end
+
+  def children=(node_or_nodes)
+    # Remove existing children, then attach the new ones in source order.
+    Taurus::XML::FFI.taurus_element_remove_children(@c_ptr)
+    Array(node_or_nodes).each { |n| add_child(n) }
+  end
+
+  # Replace this element with +new_node+ in the parent's child list.
+  # +new_node+ must belong to the same document. Returns +new_node+.
+  def replace(new_node)
+    parent = self.parent
+    raise Taurus::XML::Error, "cannot replace a node with no parent" unless parent
+    add_next_sibling(new_node)
+    parent.remove_child(self)
+    new_node
+  end
+
+  # Like #replace but returns self for chaining.
+  def swap(new_node)
+    replace(new_node)
+    self
+  end
+
+  # Wrap this element in a new element parsed from +markup+ or a dup of
+  # +node+. The wrapper takes this element's place in the tree, and this
+  # element becomes its only child. Returns self for chaining.
+  def wrap(node_or_markup)
+    wrapper =
+      case node_or_markup
+      when Taurus::XML::Element then node_or_markup.dup
+      when String
+        frag_doc = Taurus::XML::Document.parse(node_or_markup)
+        frag_doc.root or raise Taurus::XML::Error, "wrap markup has no root element"
+      else
+        raise ArgumentError, "wrap expects a String or Element, got #{node_or_markup.class}"
+      end
+
+    parent = self.parent
+    raise Taurus::XML::Error, "cannot wrap a node with no parent" unless parent
+
+    # Insert wrapper at self's position, then move self into wrapper.
+    # add_child moves self (unlinks from old parent first), so no explicit
+    # remove_child needed — and trying to remove after the move corrupts
+    # the C tree (libtaurus silently handles non-child args badly).
+    add_next_sibling(wrapper)
+    wrapper.add_child(self)
+    self
+  end
+
+  def dup
+    copy_ptr = Taurus::XML::FFI.taurus_element_copy(@c_ptr, @document.c_ptr)
+    raise Taurus::XML::Error, "taurus_element_copy failed" if copy_ptr.null?
+    Taurus::XML::Element.new(copy_ptr, @document)
+  end
+  alias_method :clone, :dup
+
+  def add_child(node_or_markup)
+    case node_or_markup
+    when Taurus::XML::Node
+      status = Taurus::XML::FFI.taurus_element_append_child(@c_ptr, node_or_markup.c_ptr)
+      raise Taurus::XML::Error,
+        Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+      node_or_markup
+    when String
+      frag = Taurus::XML::DocumentFragment.parse(node_or_markup, @document)
+      added = []
+      frag.children.each do |n|
+        status = Taurus::XML::FFI.taurus_element_append_child(@c_ptr, n.c_ptr)
+        raise Taurus::XML::Error,
+          Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+        added << n
+      end
+      Taurus::XML::NodeSet.new(@document, added)
+    else
+      raise ArgumentError, "add_child expects a Node or String, got #{node_or_markup.class}"
+    end
+  end
+
+  def namespace
+    uri = Taurus::XML::FFI.taurus_element_namespace(@c_ptr)
+    return nil if uri.nil? || uri.empty?
+    Taurus::XML::Namespace.new(self, uri)
+  end
+
+  def namespace_definitions
+    count = Taurus::XML::FFI.taurus_element_namespace_count(@c_ptr)
+    count.times.map do |i|
+      prefix = Taurus::XML::FFI.taurus_element_namespace_decl_prefix(@c_ptr, i)
+      uri = Taurus::XML::FFI.taurus_element_namespace_decl_uri(@c_ptr, i)
+      Taurus::XML::Namespace.new(self, uri, prefix: prefix)
+    end
+  end
+
+  def namespaces
+    scopes = {}
+    node = self
+    while node.is_a?(Taurus::XML::Element)
+      node.namespace_definitions.each do |ns|
+        key = ns.prefix ? "xmlns:#{ns.prefix}" : "xmlns"
+        scopes[key] ||= ns.href
+      end
+      node = node.parent
+    end
+    scopes
+  end
+
+  def to_xml(indent: 0, no_decl: false, encoding: nil)
+    opts = Taurus::XML::FFI::SerializeOptions.new
+    opts[:indent] = indent.to_i
+    opts[:xml_declaration] = no_decl ? 0 : 1
+    enc_ptr = nil
+    if encoding
+      enc_ptr = ::FFI::MemoryPointer.from_string(encoding.to_s)
+      opts[:encoding] = enc_ptr
+    end
+    str_ptr = Taurus::XML::FFI.taurus_element_serialize(@c_ptr, opts.pointer)
+    return "" if str_ptr.null?
+    str_ptr.read_string.tap { Taurus::XML::FFI.taurus_free_string(str_ptr) }
+  end
+
+  def canonicalize(version = Taurus::XML::FFI::C14N_1_0,
+                   inclusive_namespaces = nil,
+                   with_comments: false,
+                   exclusive: false,
+                   mode: nil)
+    resolved_mode = mode || (exclusive ? Taurus::XML::FFI::C14N_MODE_EXCLUSIVE
+                                       : Taurus::XML::FFI::C14N_MODE_CANONICAL)
+    ns_ptr, _anchor = Taurus::XML.c14n_build_ns_pointer(inclusive_namespaces)
+    flags = with_comments ? 1 : 0
+    str_ptr = Taurus::XML::FFI.taurus_c14n_canonicalize_subtree_ex(
+      @c_ptr, version, resolved_mode, ns_ptr, flags)
+    return "" if str_ptr.null?
+    str_ptr.read_string.tap { Taurus::XML::FFI.taurus_free_string(str_ptr) }
+  end
+  alias_method :c14n, :canonicalize
+
+  def add_namespace_definition(prefix, href)
+    status = Taurus::XML::FFI.taurus_element_add_namespace_definition(
+      @c_ptr, prefix.to_s, href.to_s)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    Taurus::XML::Namespace.new(self, href.to_s, prefix: prefix.nil? ? nil : prefix.to_s)
+  end
+  alias_method :add_namespace, :add_namespace_definition
+
+  def default_namespace=(href)
+    status = Taurus::XML::FFI.taurus_element_set_default_namespace(@c_ptr, href.to_s)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    href
+  end
+
+  def remove_namespace_definition(prefix)
+    status = Taurus::XML::FFI.taurus_element_remove_namespace_definition(@c_ptr, prefix.to_s)
+    raise Taurus::XML::Error,
+      Taurus::XML::FFI.taurus_status_string(status) unless status == Taurus::XML::FFI::TAURUS_OK
+    self
+  end
+end