init

Author: melborne

.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in yagviz.gemspec
+gemspec

LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2012 kyoendo
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

README.md

@@ -0,0 +1,29 @@
+# Yagviz
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'yagviz'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install yagviz
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

lib/yagviz.rb

@@ -0,0 +1,110 @@
+class Yagviz
+  class Node < Struct.new(:id, :attrs)
+    def initialize(id, attrs={})
+      raise ArgumentError, "`id` must not include underscores" if id.match(/_/)
+      super
+    end
+
+    def to_s
+      "#{id}"
+    end
+  end
+
+  class Edge < Struct.new(:id, :attrs)
+    attr_reader :st, :ed, :seq
+    def initialize(id, attrs={})
+      raise ArgumentError, "`attrs` must a hash" unless attrs.is_a?(Hash)
+      st, ed, seq = "#{id}".split('_')
+      @st, @ed = [st, ed].map(&:intern)
+      @seq = seq.to_i
+      super
+    end
+
+    def to_s
+      "#{st} -> #{ed}"
+    end
+
+    def nodes
+      [st, ed]
+    end
+  end
+
+  attr_reader :edges, :nodes
+  def initialize
+    @edges = {}
+    @nodes = {}
+  end
+
+  def node(id, attrs={})
+    raise ArgumentError, '`id` must a symbol' unless id.is_a?(Symbol)
+    Node[id, attrs].tap { |node| @nodes.update(id => node) }
+  end
+
+  def edge(id, attrs={})
+    unless id.is_a?(Symbol) && id.match(/._./)
+      raise ArgumentError, '`id` must a symbol in which node names joined with "_"'
+    end
+    Edge[id, attrs].tap do |edge|
+      @edges.update(id => edge)
+      create_nodes
+    end
+  end
+  
+  def graph(&blk)
+    instance_eval(&blk)
+  end
+
+  def add(*nodes_or_routes)
+    nodes_or_routes.each do |unit|
+      case unit
+      when Hash
+        unit.each do |sts, eds|
+          Array(sts).product(Array(eds))
+                    .each { |st, ed| edge([st,ed].join('_').intern) }
+        end
+      when Symbol
+        node(unit)
+      else
+        raise ArgumentError, 'pass nodes in symbol or edges in hash'
+      end
+    end
+    self
+  end
+
+  def to_s
+    result = []
+    result << "digraph {"
+    
+    nodes.values.each do |node|
+      attrs = build_attrs(node.attrs)
+      result << "  #{node.id}#{attrs};"
+    end
+    edges.values.each do |edge|
+      attrs = build_attrs(edge.attrs)
+      result << "  #{edge}#{attrs};"
+    end
+    
+    result << "}\n"
+    result.join("\n")
+  end
+  
+  private
+  def create_nodes
+    # only create unregistered nodes
+    ids = edges.values.flat_map(&:nodes).uniq - nodes.keys
+    ids.each { |id| node(id) }
+    nodes
+  end
+
+  def build_attrs(attrs)
+    return nil if attrs.empty?
+    '[' + attrs.map { |attr| attr.join("=") }.join(',') + ']'
+  end
+end
+
+if __FILE__ == $0
+  yag = Yagviz.new
+  yag.add :main => [:init, :parse, :cleanup, :printf]
+  puts yag
+end
+

lib/yagviz/version.rb

@@ -0,0 +1,3 @@
+module Yagviz
+  VERSION = "0.0.1"
+end

spec/spec_helper.rb

@@ -0,0 +1,9 @@
+require "yagviz"
+require "rspec"
+
+class String
+  def ~
+    margin = scan(/^ +/).map(&:size).min
+    gsub(/^ {#{margin}}/, '')
+  end
+end