mermaid 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a1adee7a6d24996d2cfe5bdc3e2bf812fd8491d3a15dd900cc32632363beb6ac
-  data.tar.gz: a5d9f2ee2722d208202031ab780c9586691a7feec371b500ab60cf899fb34d22
+  metadata.gz: 2f45d52ec5ba6ee5b68319696a39f2d63457aef75450754a3aa8db9d00664765
+  data.tar.gz: 4567ecd088bfd31d769fc80a5e256e1a169c74c5ad5c349fbef5b4d42fad857b
 SHA512:
-  metadata.gz: 450ccbce6c9bde4f918fca49c09761950d7f0b6b0f057bee6010a99a85ed43e8e585d6934d1db131b735c6e1c408d75960edab7a961011bf2a4bf19f47ebf2c0
-  data.tar.gz: 2c27efa7b199251b37b8d5d9ce52df250922dcb9124ed1a715b1b25a5c5d4da3ef044c48447a1668074f1c370b0e4aafe1a5a97616f4f64d159130cd04298114
+  metadata.gz: 984eae28a2bbe392c12a23331a8b69db664252b4bc949858a46fc8d9d4025ecbaab3c7aea0a65926a0b9f5ad14737e3066aa2df8236d96e64c377f6e90de6d3e
+  data.tar.gz: 104f52bd0c05ba3b1ad4ad8df6deec4d594bd08255fdc728c7f547b6627872fbd4fb8cf4692d931511fd1e2202f0b1aa0a24449fcf27e7c1e1fa2e1788e57baa

data/CHANGELOG.md

@@ -1,5 +1,20 @@
 ## [Unreleased]
 
+## 0.0.2 (2025-04-01)
+
+
+### Features
+
+* add ClassDiagram generator ([364782e](https://github.com/seuros/mermaid-ruby/commit/364782e86827e2fe5745e7177ab954cb5ddc3a6a))
+* add release flow ([2ed9710](https://github.com/seuros/mermaid-ruby/commit/2ed9710d6f8820a91aa6f19bef4d4388a5b0bc31))
+* fix diagram inital implementation ([fab6bfa](https://github.com/seuros/mermaid-ruby/commit/fab6bfa94b932902e4664d551406c77c4037f950))
+* Implemented #to_mermaid functionality for StateDiagram, ClassDiagram, PieDiagram, FlowchartDiagram, GanttDiagram, GitgraphDiagram, and TimelineDiagram by creating renderer modules ([7ab6486](https://github.com/seuros/mermaid-ruby/commit/7ab64865e1cbf6a41df23aa971f8fad79a8ba274))
+
+
+### Bug Fixes
+
+* correct releaser ([2eee28c](https://github.com/seuros/mermaid-ruby/commit/2eee28cf78d9c948ea9c8a034ba48e3b12aa0c13))
+
 ## [0.1.0] - 2024-01-23
 
 - Draft release

data/README.md

@@ -1,34 +1,138 @@
-# Mermaid
+# Mermaid Ruby
 
-TODO: Delete this and the text below, and describe your gem
+[![Gem Version](https://badge.fury.io/rb/mermaid.svg)](https://badge.fury.io/rb/mermaid)
+[![Build Status](https://github.com/seuros/mermaid-ruby/actions/workflows/ci.yml/badge.svg)](https://github.com/seuros/mermaid-ruby/actions/workflows/ci.yml)
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/mermaid`. To experiment with that code, run `bin/console` for an interactive prompt.
+This gem provides Mermaid syntax generation capabilities for diagram objects created using the [`diagram`](https://github.com/seuros/diagram-ruby) gem.
+
+It achieves this by adding renderer to the `diagram` gem (like `Diagrams::ClassDiagram`, `Diagrams::StateDiagram`, etc.) to add a `#to_mermaid` instance method.
 
 ## Installation
 
-TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
+Add these lines to your application's Gemfile:
+
+```ruby
+gem 'diagram' # The base gem for creating diagram objects
+gem 'mermaid' # This gem, for generating Mermaid syntax
+```
 
-Install the gem and add to the application's Gemfile by executing:
+And then execute:
 
-    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG
+```bash
+$ bundle install
+```
 
-If bundler is not being used to manage dependencies, install the gem by executing:
+Or install it yourself as:
 
-    $ gem install UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG
+```bash
+$ gem install diagram mermaid
+```
 
 ## Usage
 
-TODO: Write usage instructions here
+Simply require the `mermaid` gem after requiring `diagrams`. This will automatically add the `#to_mermaid` method to the diagram classes.
+
+```ruby
+require 'diagrams'
+require 'mermaid' # This applies the patches
+
+# --- Example: State Diagram ---
+state_a = Diagrams::Elements::State.new(id: 'A', label: 'State A')
+state_b = Diagrams::Elements::State.new(id: 'B')
+transition1 = Diagrams::Elements::Transition.new(source_state_id: '*', target_state_id: 'A')
+transition2 = Diagrams::Elements::Transition.new(source_state_id: 'A', target_state_id: 'B', label: 'Event X')
+
+state_diagram = Diagrams::StateDiagram.new(
+  states: [state_a, state_b],
+  transitions: [transition1, transition2]
+)
+
+puts state_diagram.to_mermaid
+# Output:
+# stateDiagram-v2
+#   A : State A
+#   [*] --> A
+#   A --> B : Event X
+
+# --- Example: Class Diagram ---
+class_animal = Diagrams::Elements::ClassEntity.new(name: 'Animal', attributes: ['+String name'])
+class_dog = Diagrams::Elements::ClassEntity.new(name: 'Dog', attributes: ['+String breed'], methods: ['+bark() void'])
+rel_inheritance = Diagrams::Elements::Relationship.new(
+  source_class_name: 'Dog',
+  target_class_name: 'Animal',
+  type: 'inheritance'
+)
+
+class_diagram = Diagrams::ClassDiagram.new(
+  classes: [class_animal, class_dog],
+  relationships: [rel_inheritance]
+)
+
+puts class_diagram.to_mermaid
+# Output:
+# classDiagram
+# class Animal {
+#   +String name
+# }
+# class Dog {
+#   +String breed
+#   +bark() void
+# }
+# Dog <|-- Animal
+
+# --- Example: Flowchart ---
+node1 = Diagrams::Elements::Node.new(id: 'n1', label: 'Start')
+node2 = Diagrams::Elements::Node.new(id: 'n2', label: 'Process')
+edge1 = Diagrams::Elements::Edge.new(source_id: 'n1', target_id: 'n2', label: 'Go')
+
+flowchart = Diagrams::FlowchartDiagram.new(nodes: [node1, node2], edges: [edge1])
+
+puts flowchart.to_mermaid
+# Output:
+# graph TD
+#   n1["Start"]
+#   n2["Process"]
+#   n1 -- "Go" --> n2
+
+# --- Example: Pie Chart ---
+slice1 = Diagrams::Elements::Slice.new(label: 'Work', value: 8)
+slice2 = Diagrams::Elements::Slice.new(label: 'Sleep', value: 8)
+slice3 = Diagrams::Elements::Slice.new(label: 'Play', value: 8)
+
+pie_chart = Diagrams::PieDiagram.new(title: 'Daily Routine', slices: [slice1, slice2, slice3])
+
+puts pie_chart.to_mermaid
+# Output:
+# pie
+# title Daily Routine
+# "Work" : 8
+# "Sleep" : 8
+# "Play" : 8
+
+# --- Example: Gantt Chart ---
+task1 = Diagrams::Elements::Task.new(id: 't1', name: 'Phase 1', start_date: '2024-01-01', end_date: '2024-01-10')
+task2 = Diagrams::Elements::Task.new(id: 't2', name: 'Phase 2', start_date: '2024-01-11', end_date: '2024-01-20')
+
+gantt_chart = Diagrams::GanttDiagram.new(title: 'Project Timeline', tasks: [task1, task2])
+
+puts gantt_chart.to_mermaid
+# Output:
+# gantt
+# title Project Timeline
+# : Phase 1 : t1, 2024-01-01, 2024-01-10
+# : Phase 2 : t2, 2024-01-11, 2024-01-20
+
+```
 
 ## Development
 
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `bundle exec rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
 
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `lib/mermaid/version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/mermaid.
+Bug reports and pull requests are welcome on GitHub at https://github.com/seuros/mermaid-ruby.
 
 ## License
 

data/Rakefile

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require "bundler/gem_tasks"
-require "minitest/test_task"
+require 'bundler/gem_tasks'
+require 'minitest/test_task'
 
 Minitest::TestTask.create
 

data/lib/mermaid/class_diagram_renderer.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module Mermaid
+  # Module containing Mermaid rendering logic for Diagrams::ClassDiagram
+  module ClassDiagramRenderer
+    # @!method to_mermaid
+    #   Generates the Mermaid syntax for the class diagram.
+    #   @return [String] The Mermaid syntax string.
+    def to_mermaid
+      mermaid_string = +"classDiagram\n"
+
+      classes.each do |klass|
+        fragment = format_class_entity(klass) # Call private helper method
+        mermaid_string << fragment.to_s << "\n"
+      end
+
+      relationships.each do |rel|
+        mermaid_string << rel.to_mermaid_fragment.to_s
+      end
+
+      mermaid_string.strip
+    end
+
+    private
+
+    # Helper to format ClassEntity for Mermaid
+    def format_class_entity(klass)
+      diagram = "class #{klass.name}"
+      attributes_list = klass[:attributes] || []
+      methods_list = klass[:methods] || []
+      has_members = attributes_list.any? || methods_list.any?
+      diagram += " {\n" if has_members
+
+      attributes_list.each do |attr_str|
+        diagram += "  #{attr_str}\n"
+      end
+      methods_list.each do |method_str|
+        diagram += "  #{method_str}\n"
+      end
+
+      diagram += "}\n" if has_members
+      diagram
+    end
+  end
+end

data/lib/mermaid/element_renderers/edge_renderer.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Mermaid
+  module ElementRenderers
+    # Module containing Mermaid rendering logic for Diagrams::Elements::Edge
+    module EdgeRenderer
+      # @!method to_mermaid_fragment
+      #   Generates the Mermaid fragment for a flowchart edge.
+      #   @return [String] Mermaid syntax fragment.
+      def to_mermaid_fragment
+        if label && !label.empty?
+          "#{source_id} -- \"#{label}\" --> #{target_id}"
+        else
+          "#{source_id} --> #{target_id}"
+        end
+      end
+    end
+  end
+end

data/lib/mermaid/element_renderers/node_renderer.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module Mermaid
+  module ElementRenderers
+    # Module containing Mermaid rendering logic for Diagrams::Elements::Node
+    module NodeRenderer
+      # @!method to_mermaid_fragment
+      #   Generates the Mermaid fragment for a flowchart node.
+      #   @return [String] Mermaid syntax fragment.
+      def to_mermaid_fragment
+        escaped_label = label.gsub('"', '#quot;')
+        "#{id}[\"#{escaped_label}\"]"
+      end
+    end
+  end
+end

data/lib/mermaid/element_renderers/relationship_renderer.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Mermaid
+  module ElementRenderers
+    # Module containing Mermaid rendering logic for Diagrams::Elements::Relationship
+    module RelationshipRenderer
+      # Maps relationship type strings to Mermaid symbols
+      MERMAID_RELATIONSHIP_MAP = {
+        'inheritance' => '<|--', 'composition' => '*--', 'aggregation' => 'o--',
+        'association' => '-->', 'dependency' => '..>', 'realization' => '<|..',
+        'link' => '--'
+      }.freeze
+
+      # @!method to_mermaid_fragment
+      #   Generates the Mermaid fragment for a relationship.
+      #   @return [String] Mermaid syntax fragment.
+      def to_mermaid_fragment
+        arrow = MERMAID_RELATIONSHIP_MAP[type] || '--'
+        label_str = label ? " : #{label}" : ''
+        "#{source_class_name} #{arrow} #{target_class_name}#{label_str}\n"
+      end
+    end
+  end
+end

data/lib/mermaid/element_renderers/slice_renderer.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module Mermaid
+  module ElementRenderers
+    # Module containing Mermaid rendering logic for Diagrams::Elements::Slice
+    module SliceRenderer
+      # @!method to_mermaid_fragment
+      #   Generates the Mermaid fragment for a pie chart slice.
+      #   @return [String] Mermaid syntax fragment.
+      def to_mermaid_fragment
+        formatted_value = value.to_i == value ? value.to_i : value
+        "\"#{label}\" : #{formatted_value}"
+      end
+    end
+  end
+end

data/lib/mermaid/element_renderers/state_renderer.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Mermaid
+  module ElementRenderers
+    # Module containing Mermaid rendering logic for Diagrams::Elements::State
+    module StateRenderer
+      # @!method to_mermaid_fragment
+      #   Generates the Mermaid fragment for a state definition.
+      #   @return [String] Mermaid syntax fragment.
+      def to_mermaid_fragment
+        label ? "#{id} : #{label}" : id
+      end
+    end
+  end
+end

data/lib/mermaid/element_renderers/task_renderer.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Mermaid
+  module ElementRenderers
+    # Module containing Mermaid rendering logic for Diagrams::Elements::Task
+    module TaskRenderer
+      # @!method to_mermaid_fragment
+      #   Generates the Mermaid fragment for a Gantt task.
+      #   @return [String] Mermaid syntax fragment.
+      def to_mermaid_fragment
+        "#{name} :#{id}, #{start_date}, #{end_date}"
+      end
+    end
+  end
+end

data/lib/mermaid/element_renderers/transition_renderer.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Mermaid
+  module ElementRenderers
+    # Module containing Mermaid rendering logic for Diagrams::Elements::Transition
+    module TransitionRenderer
+      # @!method to_mermaid_fragment
+      #   Generates the Mermaid fragment for a transition.
+      #   @return [String] Mermaid syntax fragment.
+      def to_mermaid_fragment
+        from_node = source_state_id == '*' ? '[*]' : source_state_id
+        to_node = target_state_id == '*' ? '[*]' : target_state_id
+        label ? "#{from_node} --> #{to_node} : #{label}" : "#{from_node} --> #{to_node}"
+      end
+    end
+  end
+end

data/lib/mermaid/flowchart_diagram_renderer.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module Mermaid
+  # Module containing Mermaid rendering logic for Diagrams::FlowchartDiagram
+  module FlowchartDiagramRenderer
+    # @!method to_mermaid
+    #   Generates the Mermaid syntax for the flowchart diagram.
+    #   @return [String] The Mermaid syntax string.
+    def to_mermaid
+      mermaid_string = +"graph TD\n"
+
+      nodes.each do |node|
+        mermaid_string << "  #{node.to_mermaid_fragment}\n" # Ensure newline after node
+      end
+
+      edges.each do |edge|
+        mermaid_string << "  #{edge.to_mermaid_fragment}\n" # Ensure newline after edge
+      end
+
+      mermaid_string # Return the full string including final newline
+    end
+  end
+end

data/lib/mermaid/gantt_diagram_renderer.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module Mermaid
+  # Module containing Mermaid rendering logic for Diagrams::GanttDiagram
+  module GanttDiagramRenderer
+    # @!method to_mermaid
+    #   Generates the Mermaid syntax for the Gantt chart.
+    #   @return [String] The Mermaid syntax string.
+    def to_mermaid
+      mermaid_string = +"gantt\n"
+      mermaid_string << "title #{@title}\n" if @title && !@title.empty?
+      tasks.each { |task| mermaid_string << task.to_mermaid_fragment << "\n" }
+      mermaid_string.strip
+    end
+  end
+end

data/lib/mermaid/gitgraph_diagram_renderer.rb

@@ -0,0 +1,113 @@
+# frozen_string_literal: true
+
+module Mermaid
+  # Module containing Mermaid rendering logic for Diagrams::GitgraphDiagram
+  module GitgraphDiagramRenderer
+    # @!method to_mermaid
+    #   Generates the Mermaid syntax for the gitgraph diagram.
+    #   @return [String] The Mermaid syntax string.
+    def to_mermaid
+      mermaid_string = +"gitGraph\n"
+      processed_commits = Set.new # Track the current branch in Mermaid output (defaults to 'master')
+
+      # Determine initial branch (default to 'main') and add checkout
+      first_commit_id = commit_order.first
+      # Default to 'main' if no commits or first commit doesn't specify branch
+      # Default to 'main' if no commits exist or first commit doesn't specify branch
+      initial_branch = (first_commit_id ? commits[first_commit_id]&.branch_name : nil) || 'master' # Default to 'master'
+      mermaid_string << "  checkout #{initial_branch}\n"
+      current_mermaid_branch = initial_branch
+
+      commit_order.each do |commit_id|
+        commit = commits[commit_id]
+        next unless commit # Skip if commit data is missing
+
+        # --- Determine Branch Context ---
+        commit.branch_name
+
+        # --- Handle Branch Creation & Checkout ---
+        # Determine the branch this commit *should* be on based on its branch_name attribute
+        target_branch_name = commit.branch_name
+
+        # Check if a new branch *starts* at this commit's parent
+        parent_commit_id = commit.parent_ids.first
+        new_branch_starting_here = nil
+        # Only consider branching if the target is NOT master and IS different from current
+        if parent_commit_id && target_branch_name != 'master' && target_branch_name != current_mermaid_branch
+          # Check if a branch object exists that starts at the parent and matches the target name
+          new_branch_starting_here = branches.values.find do |b|
+            b.start_commit_id == parent_commit_id && b.name == target_branch_name
+          end
+        end
+
+        if new_branch_starting_here
+          # Ensure we are on the parent branch before creating the new one
+          parent_commit = commits[parent_commit_id]
+          parent_branch_name = parent_commit&.branch_name
+          if parent_branch_name && parent_branch_name != current_mermaid_branch
+            mermaid_string << "  checkout #{parent_branch_name}\n"
+            current_mermaid_branch = parent_branch_name
+          end
+          mermaid_string << "  branch #{target_branch_name}\n"
+          # No checkout here, the next check handles it
+        end
+
+        # Now, handle checkout if needed for the current commit
+        if target_branch_name != current_mermaid_branch
+          # Always checkout if the branch differs from the last known Mermaid branch
+          mermaid_string << "  checkout #{target_branch_name}\n"
+          current_mermaid_branch = target_branch_name
+        end
+
+        # --- Handle Merge ---
+        if commit.type == :MERGE && commit.parent_ids.length == 2 && !processed_commits.include?(commit_id)
+          # Identify both parents
+          parent1_id = commit.parent_ids[0]
+          parent2_id = commit.parent_ids[1]
+          parent1_commit = commits[parent1_id]
+          parent2_commit = commits[parent2_id]
+
+          # Find the parent commit whose branch is NOT the current branch
+          source_commit = if parent1_commit&.branch_name != current_mermaid_branch
+                            parent1_commit
+                          elsif parent2_commit&.branch_name != current_mermaid_branch
+                            parent2_commit
+                          end
+          source_branch_name = source_commit&.branch_name
+
+          # Ensure source branch is valid and different from target
+          if source_branch_name && source_branch_name != current_mermaid_branch
+            mermaid_string << "  merge #{source_branch_name}"
+            mermaid_string << " id: \"#{commit.id}\""
+            mermaid_string << " tag: \"#{commit.tag}\"" if commit.tag
+            mermaid_string << "\n"
+            processed_commits.add(commit_id)
+          else
+            # If source branch can't be determined, generate a regular commit as fallback
+            warn "Could not determine source branch for merge commit #{commit_id}. Generating standard commit."
+            generate_commit_line(mermaid_string, commit, processed_commits) unless processed_commits.include?(commit_id)
+          end
+
+        # --- Handle Standard Commit ---
+        elsif !processed_commits.include?(commit_id)
+          generate_commit_line(mermaid_string, commit, processed_commits)
+        end
+      end
+
+      mermaid_string.strip
+    end
+
+    private
+
+    # Helper to generate a standard commit line
+    def generate_commit_line(mermaid_string, commit, processed_commits)
+      mermaid_string << '  commit'
+      mermaid_string << " id: \"#{commit.id}\""
+      mermaid_string << " tag: \"#{commit.tag}\"" if commit.tag
+      mermaid_string << " type: #{commit.type.to_s.upcase}" unless commit.type == :NORMAL
+      # Mermaid doesn't directly support messages in the command, they appear on the graph
+      mermaid_string << "\n"
+      processed_commits.add(commit.id)
+    end
+  end
+end

data/lib/mermaid/pie_diagram_renderer.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module Mermaid
+  # Module containing Mermaid rendering logic for Diagrams::PieDiagram
+  module PieDiagramRenderer
+    # @!method to_mermaid
+    #   Generates the Mermaid syntax for the pie chart.
+    #   @return [String] The Mermaid syntax string.
+    def to_mermaid
+      mermaid_string = +"pie\n"
+      mermaid_string << "title #{@title}\n" if @title && !@title.empty?
+      slices.each { |slice| mermaid_string << slice.to_mermaid_fragment << "\n" }
+      mermaid_string.strip
+    end
+  end
+end

data/lib/mermaid/state_diagram_renderer.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module Mermaid
+  # Module containing Mermaid rendering logic for Diagrams::StateDiagram
+  module StateDiagramRenderer
+    # @!method to_mermaid
+    #   Generates the Mermaid syntax for the state diagram.
+    #   @return [String] The Mermaid syntax string.
+    def to_mermaid
+      mermaid_string = +"stateDiagram-v2\n" # Use mutable string
+
+      # Define states with labels first
+      states.each do |state|
+        mermaid_string << "  #{state.to_mermaid_fragment}\n" if state.label
+      end
+
+      # Add transitions
+      transitions.each do |transition|
+        mermaid_string << "  #{transition.to_mermaid_fragment}\n"
+      end
+
+      mermaid_string.strip # Remove potential trailing newline
+    end
+
+    # Patch validation to allow '*' for start/end states used by Mermaid
+    private
+
+    def validate_elements!
+      # Temporarily filter out transitions involving '*' before calling original validation
+      original_transitions = @transitions
+      @transitions = original_transitions.reject do |t|
+        t.source_state_id == '*' || t.target_state_id == '*'
+      end
+
+      # Call original validation using super
+      super # Calls Diagrams::StateDiagram#validate_elements!
+    ensure
+      # Restore original transitions array regardless of validation outcome
+      @transitions = original_transitions
+    end
+  end
+end

data/lib/mermaid/timeline_diagram_renderer.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Mermaid
+  # Module containing Mermaid rendering logic for Diagrams::TimelineDiagram
+  module TimelineDiagramRenderer
+    # @!method to_mermaid
+    #   Generates the Mermaid syntax for the timeline diagram.
+    #   @return [String] The Mermaid syntax string.
+    def to_mermaid
+      mermaid_string = +"timeline\n"
+      mermaid_string << "title #{@title}\n" if @title && !@title.empty?
+
+      sections.each do |section|
+        # Don't output the default section title if it's the only one and has no specific title
+        unless sections.size == 1 && section.title == Diagrams::TimelineDiagram::DEFAULT_SECTION_TITLE
+          mermaid_string << "section #{section.title}\n"
+        end
+
+        section.periods.each do |period|
+          # Format: Period Label : Event 1 : Event 2 ...
+          event_descriptions = period.events.map(&:description).join(' : ')
+          mermaid_string << "  #{period.label} : #{event_descriptions}\n"
+        end
+      end
+
+      mermaid_string.strip
+    end
+  end
+end

data/lib/mermaid/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Mermaid
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 end

data/lib/mermaid.rb

@@ -1,17 +1,27 @@
 # frozen_string_literal: true
 
 require_relative 'mermaid/version'
-require 'zeitwerk'
-require 'dry-types'
-require 'dry-struct'
+require 'diagrams'
 
-loader = Zeitwerk::Loader.for_gem
-loader.inflector.inflect(
-  'er_diagram' => 'ERDiagram',
-  'xy_diagram' => 'XYDiagram'
-)
+loader = Zeitwerk::Loader.for_gem # Remove Zeitwerk setup
 loader.setup
 
-module Mermaid
-
+module Diagrams
+  StateDiagram.prepend(Mermaid::StateDiagramRenderer)
+  ClassDiagram.prepend(Mermaid::ClassDiagramRenderer)
+  FlowchartDiagram.prepend(Mermaid::FlowchartDiagramRenderer)
+  PieDiagram.prepend(Mermaid::PieDiagramRenderer)
+  GanttDiagram.prepend(Mermaid::GanttDiagramRenderer)
+  GitgraphDiagram.prepend(Mermaid::GitgraphDiagramRenderer)
+  TimelineDiagram.prepend(Mermaid::TimelineDiagramRenderer) # Add prepend for Timeline
+  module Elements
+    State.prepend(Mermaid::ElementRenderers::StateRenderer)
+    Transition.prepend(Mermaid::ElementRenderers::TransitionRenderer)
+    Relationship.prepend(Mermaid::ElementRenderers::RelationshipRenderer)
+    Node.prepend(Mermaid::ElementRenderers::NodeRenderer)
+    Edge.prepend(Mermaid::ElementRenderers::EdgeRenderer)
+    Slice.prepend(Mermaid::ElementRenderers::SliceRenderer)
+    Task.prepend(Mermaid::ElementRenderers::TaskRenderer)
+    # Add GitCommit, GitBranch, Timeline* prepends if/when renderers are created
+  end
 end