yuml 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 508fdebd5e20dd067d21bc3172fb3e4d9c3c0f17
-  data.tar.gz: 12bb3693b5dfadcf8fa497edb035d20afe6913d9
+  metadata.gz: d26bbbcd5645713f0d95dd6e418fd6fe1acc617f
+  data.tar.gz: a17b1a05f2602551c70f5f94c2d49e472d70b4c7
 SHA512:
-  metadata.gz: 6ff3c8a87b37206505f1ae58b1ad54f6eaaff440b48988218f0ad215f1ba6f693f3ec06f2e8b23f4fa37bb7b27bea600579ea7cbbb0322008a0991dbc59ba843
-  data.tar.gz: ff31268654ebcfa26f32a8208fb75a090bec511f2c4f662745fc3ec7ace043e612d83b93a029652d9b5948116ecde73ea746989c90aa9a6536d8ff16d98e9ce9
+  metadata.gz: 81a3897f59e805c8bc2ba41c3e1f4dc1f5acde251084c541a9324273bab5871a4a9ea51a30e7c7083d8ab347232910ab99d29a7158e89ae1aed7d9000525dc24
+  data.tar.gz: f5ffcb9409a70ce29144d6968b1cbf9613f00a5d91567f31c4e318586fcffcc27201c9f7c4680dd9194aa6eaaff04d0b3c8e24f5eb2b445a05c4dc22b8acefda

data/lib/yuml.rb

@@ -1,13 +1,82 @@
 # Fetches UML from yUML
+require 'net/http'
 require_relative 'yuml/class'
 require_relative 'yuml/relationship'
 
 module YUML
-  def generate_class
-    y = YUML::Class.new
-    yield y
-    y
+  extend self
+
+  ESCAPE_CHARACTERS = {
+    '{' => "\u23A8",
+    '}' => "\u23AC"
+  }
+
+  ESCAPE_COMMA = "\u201A"
+
+  def generate(options)
+    options = { file: '/tmp/yuml.pdf' }.merge(options)
+    classes.clear
+    yield self
+    fetch_uml options[:file]
+  end
+
+  def class(&block)
+    yuml_class = YUML::Class.new
+    block.arity < 1 ? yuml_class.instance_eval(&block) : block.call(yuml_class)
+    classes << yuml_class
+    yuml_class
+  end
+
+  private
+
+  def classes
+    @classes ||= []
+  end
+
+  def yuml
+    uml = classes.map(&:to_s).join(',')
+    relationships = classes.map(&:relationships).compact
+    uml << ',' << relationships.join(',') unless relationships.empty?
+    puts uml
+    URI.encode(uml, encodings)
+  end
+
+  def fetch_uml(file)
+    uri = URI("http://yuml.me/diagram/class/#{yuml}.pdf")
+    response = Net::HTTP.get_response(uri)
+    File.write(file, response.body)
+  end
+
+  def encodings
+    "#{ESCAPE_CHARACTERS.values.join}#{ESCAPE_COMMA}[](){}+->|,=;* "
+  end
+end
+
+YUML.generate(file: 'tmp.pdf') do |uml|
+  doc = uml.class do
+    name 'Document'
+  end
+
+  pic = uml.class do
+    name 'Picture'
+    public_methods :foo
+  end
+
+  doc.has_a(pic)
+end
+
+YUML.generate(file: 'tmp2.pdf') do |uml|
+  doc = uml.class do
+    name 'Document'
+  end
+
+  pic = uml.class do
+    name 'Picture'
+    public_variables :fizz
+    private_variables :buzz
+    public_methods(foo: [:bar])
+    private_methods(baz: ['arg = nil', 'param = {}', '*args'], lam: [:star])
   end
 
-  module_function :generate_class
+  doc.has_a(pic)
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: yuml
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Derek Stride