ruby_uml_class 0.2.0 → 0.3.0

data/lib/create_uml_class.rb

@@ -1,297 +1,322 @@
-require "tempfile"
-
-CStruct = Struct.new(:type,
-                     :name,
-                     :block_count,
-                     :var_list,
-                     :method_list,
-                     :inherit_list,
-                     :composition_list)
-
-def print_uml(out, out_list)
-  out_list.each do |o_list|
-    if o_list.type == :class_start
-      # nop
-    elsif o_list.type == :module_start
-      out.push "namespace #{o_list.name} {"
-    elsif o_list.type == :class_end
-      pp o_list if o_list.name == ""
-      out.push "class #{o_list.name} {"
-      # インスタンス変数の出力
-      o_list.var_list.uniq.each do |iv|
-        out.push iv
-      end
-      # メソッドの出力
-      o_list.method_list.each do |ml|
-        out.push ml
-      end
-      out.push "}"
-      # 継承リストの出力
-      o_list.inherit_list.each do |ih|
-        out.push "#{o_list.name} --|> #{ih}"
-      end
-      # compo
-      o_list.composition_list.uniq.each do |co|
-        out.push "#{o_list.name} *-- #{co}"
-      end
-    elsif o_list.type == :module_end
-      # インスタンス変数がある場合はモジュール名と同じクラスを定義
-      if o_list.var_list.size != 0 or
-         o_list.method_list.size != 0 or
-         o_list.inherit_list.size != 0 or
-         o_list.composition_list.size != 0
-        pp o_list if o_list.name == ""
-        out.push "class #{o_list.name} {"
-        # インスタンス変数の出力
-        o_list.var_list.uniq.each do |iv|
-          out.push iv
-        end
-        # メソッドの出力
-        o_list.method_list.each do |ml|
-          out.push ml
-        end
-        out.push "}"
-        # 継承リストの出力
-        o_list.inherit_list.each do |ih|
-          out.push "#{o_list.name} --|> #{ih}"
-        end
-        # compo
-        o_list.composition_list.uniq.each do |co|
-          out.push "#{o_list.name} *-- #{co}"
-        end
-      end
-      out.push "}"
-    else
-      # error
-      puts "error!"
-    end
-  end
-  return out
-end
-
-def delete_here_doc(buf)
-  new_buf = []
-  here_doc = false
-  here_word = ""
-  buf.each_line do |line|
-    if line =~ /(<<|<<~|<<-)[A-Z]+/
-      here_doc = true
-      here_word = line.match(/(<<|<<~|<<-)[A-Z]+/).to_s.gsub(/[<~-]/, "")
-    end
-    if here_word != "" and line =~ Regexp.new("^\s*#{here_word}$")
-      here_word = ""
-      here_doc = false
-      pp line
-    end
-    if here_doc == false
-      new_buf.push line
-    else
-      pp line
-    end
-  end
-  return new_buf.join("")
-end
-
-def create_uml_class(in_dir, out_file)
-  out = []
-  out.push "@startuml"
-
-  puts "in_dir = #{in_dir}"
-  main_composition_list = []
-  main_method_list = []
-  global_var = []
-
-  Dir.glob("#{in_dir}/**/*.{rb,ru}") do |f|
-    puts f
-    buf = ""
-    Tempfile.create("rufo") do |tmp_file|
-      FileUtils.cp(f, tmp_file.path)
-      open("|rufo #{tmp_file.path}") do |f|
-        if f.read =~ /error/
-          puts "rufo error #{f}"
-          return
-        else
-          buf = File.binread tmp_file.path
-        end
-      end
-    end
-
-    # コメント削除
-    buf.gsub!(/(([\/\"\'].*?[\/\"\'])|([^\/\"\'\)\s]*#.+?$))/) do |m|
-      if m[0] == "#" and m[0] != "{"
-        #puts "comment #{m}"
-        # コメント
-        ""
-      else
-        #puts "not comment #{m}"
-        # コメント以外
-        m
-      end
-    end
-    # ヒアドキュメント削除
-    buf = delete_here_doc(buf)
-
-    out_list = []
-    cstruct_list = []
-    block_count = 0
-    method_type = :public
-    class_name = ""
-    # ソースを解析
-    buf.each_line do |line|
-      next if line =~ /^[\r\n]*$/  # 空行は対象外
-
-      # ブロックの開始/終了
-      indent_num = line.match(/^[ ]+/).to_s.size / 2
-      if block_count == indent_num
-        # 変化なし
-      elsif block_count > indent_num
-        # ブロックの終了
-        block_count = indent_num
-      else
-        # ブロックの開始
-        block_count = indent_num
-      end
-
-      #line.gsub!(/\".+\"/, "\"delete_string\"") # 文字列を削除
-      if line =~ /^\s*class\s/
-        unless line =~ /<</ # 特異クラスではない
-          work = line.gsub(/class\s/, "")
-          class_name = work.split("<")[0].to_s.chomp.match(/[A-Z][A-Za-z0-9_:]+/).to_s
-          base_name = work.split("<")[1].to_s.chomp.match(/[A-Z][A-Za-z0-9_:]+/).to_s
-          class_name.gsub!(/::/, ".")
-          if out_list.size != 0 and out_list[-1].type == :class_start # classが連続している
-            class_name = out_list[-1].name + "." + class_name
-            out_list[-1].name = class_name
-            cstruct_list[-1].name = class_name
-          else
-            out_list.push CStruct.new(:class_start, class_name, block_count, [], [], [], [])
-            cstruct_list.push CStruct.new(:class_end, class_name, block_count, [], [], [], [])
-          end
-          #pp line if class_name == ""
-          if base_name != ""
-            #base_name.gsub!(/::/, ".")
-            cstruct_list[-1].inherit_list.push base_name
-          end
-        end
-        next unless line =~ /end\s*$/ # 1行で終了しない場合
-      elsif line =~ /^\s*module\s/
-        module_name = line.split(" ")[1].to_s.chomp
-        module_name.gsub!(/^[:]+/, "")
-        module_name.gsub!(/::/, ".")
-        out_list.push CStruct.new(:module_start, module_name, block_count, [], [], [], [])
-        cstruct_list.push CStruct.new(:module_end, module_name, block_count, [], [], [], [])
-        next unless line =~ /end\s*$/ # 1行で終了しない場合
-      end
-
-      if line =~ /^\s*private$/
-        method_type = :private
-      elsif line =~ /^\s*protected$/
-        method_type = :protected
-      elsif line =~ /^\s*public$/
-        method_type = :public
-      end
-
-      if line =~ /^\s*def\s/
-        # 関数名を取り出す
-        method = line.chomp.gsub(/\s*def\s*/, "")
-        unless method =~ /\(/
-          # 関数名にカッコをつける
-          sp = method.split(" ")
-          if sp.size > 1
-            method = sp[0].to_s + "(" + sp[1..-1].to_s + ")"
-          else
-            method = method + "()"
-          end
-        end
-        if cstruct_list.size != 0
-          method_list = cstruct_list[-1].method_list
-          case method_type
-          when :public
-            method_list.push "+ #{method}"
-          when :private
-            method_list.push "- #{method}"
-          when :protected
-            method_list.push "# #{method}"
-          end
-        else
-          main_method_list.push "+ #{method}"
-        end
-      end
-
-      # composition_list
-      line.match(/(([\/\"\')].*?\.new.*?[\/\"\'])|(?![\/\"\'])([a-zA-Z0-9_]+\.new))/) do |m|
-        if m.to_s[0] != "/" and m.to_s[0] != "\"" and m.to_s[0] != "'"
-          name = m.to_s.gsub(/\.new/, "").match(/[A-Z][A-Za-z0-9_]+/).to_s
-          if name != ""
-            if cstruct_list.size != 0
-              cstruct_list[-1].composition_list.push name
-            else
-              main_composition_list.push "main *-- #{name}"
-            end
-          end
-        end
-      end
-
-      # インスタンス変数
-      if line =~ /\s*@\S+/
-        if cstruct_list.size != 0
-          line.match(/@[a-zA-Z0-9_]+/) { |m|
-            instance_var = cstruct_list[-1].var_list
-            val = m.to_s.gsub(/@/, "")
-            case method_type
-            when :public
-              instance_var.push "+ #{val}"
-            when :private
-              instance_var.push "- #{val}"
-            when :protected
-              instance_var.push "# #{val}"
-            end
-          }
-        end
-      end
-
-      # 外部変数
-      line.match(/\$[a-zA-Z0-9_]+/) { |m|
-        global_var.push "+ #{m.to_s}"
-      }
-
-      # クラスの終了
-      if cstruct_list.size != 0
-        if block_count == cstruct_list[-1].block_count # block_countが一致
-          #puts "end of #{cstruct_list[-1].name}"
-          out_list.push cstruct_list[-1]
-          cstruct_list.slice!(-1) # 最後の要素を削除
-        end
-      end
-      #puts "#{block_count} #{line.chomp}"
-    end
-    if block_count != 0
-      # エラー
-      puts f
-      return ""
-    end
-
-    # UMLの出力
-    out = print_uml(out, out_list)
-  end
-
-  if main_method_list.size != 0 or
-     main_composition_list.size != 0 or
-     main_method_list.size != 0
-    out.push "class main {"
-    main_method_list.each do |mml|
-      out.push mml
-    end
-    # グローバル変数の出力
-    global_var.uniq.each do |gv|
-      out.push gv
-    end
-    out.push "}"
-    main_composition_list.uniq.each do |mcl|
-      out.push mcl
-    end
-  end
-
-  out.push "@enduml"
-  return out.join("\n")
-end
+require "tempfile"
+require "facter"
+
+CStruct = Struct.new(:type,
+                     :name,
+                     :block_count,
+                     :var_list,
+                     :method_list,
+                     :inherit_list,
+                     :composition_list)
+
+def get_rufo_path
+  ENV["PATH"].split(";").each do |path|
+    rufo_path = "#{path}\\rufo"
+    if File.exists? rufo_path
+      return rufo_path
+    end
+  end
+  return ""
+end
+
+def print_uml(out, out_list)
+  out_list.each do |o_list|
+    if o_list.type == :class_start
+      # nop
+    elsif o_list.type == :module_start
+      out.push "namespace #{o_list.name} {"
+    elsif o_list.type == :class_end
+      pp o_list if o_list.name == ""
+      out.push "class #{o_list.name} {"
+      # インスタンス変数の出力
+      o_list.var_list.uniq.each do |iv|
+        out.push iv
+      end
+      # メソッドの出力
+      o_list.method_list.each do |ml|
+        out.push ml
+      end
+      out.push "}"
+      # 継承リストの出力
+      o_list.inherit_list.each do |ih|
+        out.push "#{o_list.name} --|> #{ih}"
+      end
+      # compo
+      o_list.composition_list.uniq.each do |co|
+        out.push "#{o_list.name} *-- #{co}"
+      end
+    elsif o_list.type == :module_end
+      # インスタンス変数がある場合はモジュール名と同じクラスを定義
+      if o_list.var_list.size != 0 or
+         o_list.method_list.size != 0 or
+         o_list.inherit_list.size != 0 or
+         o_list.composition_list.size != 0
+        pp o_list if o_list.name == ""
+        out.push "class #{o_list.name} {"
+        # インスタンス変数の出力
+        o_list.var_list.uniq.each do |iv|
+          out.push iv
+        end
+        # メソッドの出力
+        o_list.method_list.each do |ml|
+          out.push ml
+        end
+        out.push "}"
+        # 継承リストの出力
+        o_list.inherit_list.each do |ih|
+          out.push "#{o_list.name} --|> #{ih}"
+        end
+        # compo
+        o_list.composition_list.uniq.each do |co|
+          out.push "#{o_list.name} *-- #{co}"
+        end
+      end
+      out.push "}"
+    else
+      # error
+      puts "error!"
+    end
+  end
+  return out
+end
+
+def delete_here_doc(buf)
+  new_buf = []
+  here_doc = false
+  here_word = ""
+  buf.each_line do |line|
+    if line =~ /(<<|<<~|<<-)[A-Z]+/
+      here_doc = true
+      here_word = line.match(/(<<|<<~|<<-)[A-Z]+/).to_s.gsub(/[<~-]/, "")
+    end
+    if here_word != "" and line =~ Regexp.new("^\s*#{here_word}$")
+      here_word = ""
+      here_doc = false
+      pp line
+    end
+    if here_doc == false
+      new_buf.push line
+    else
+      pp line
+    end
+  end
+  return new_buf.join("")
+end
+
+def create_uml_class(in_dir, out_file)
+  out = []
+  out.push "@startuml"
+
+  puts "in_dir = #{in_dir}"
+  main_composition_list = []
+  main_method_list = []
+  global_var = []
+
+  Dir.glob("#{in_dir}/**/*.{rb,ru}") do |f|
+    puts f
+    buf = ""
+    Tempfile.create("rufo") do |tmp_file|
+      FileUtils.cp(f, tmp_file.path)
+      kernel = Facter.value(:kernel)
+      if kernel == "windows"
+        open("|rubyw #{get_rufo_path} #{tmp_file.path}") do |f|
+          if f.read =~ /error/
+            puts "rufo error #{f}"
+            return
+          else
+            buf = File.binread tmp_file.path
+          end
+        end
+      else
+        open("|rufo #{tmp_file.path}") do |f|
+          if f.read =~ /error/
+            puts "rufo error #{f}"
+            return
+          else
+            buf = File.binread tmp_file.path
+          end
+        end
+      end
+    end
+
+    # コメント削除
+    buf.gsub!(/(([\/\"\'].*?[\/\"\'])|([^\/\"\'\)\s]*#.+?$))/) do |m|
+      if m[0] == "#" and m[0] != "{"
+        #puts "comment #{m}"
+        # コメント
+        ""
+      else
+        #puts "not comment #{m}"
+        # コメント以外
+        m
+      end
+    end
+    # ヒアドキュメント削除
+    buf = delete_here_doc(buf)
+
+    out_list = []
+    cstruct_list = []
+    block_count = 0
+    method_type = :public
+    class_name = ""
+    # ソースを解析
+    buf.each_line do |line|
+      next if line =~ /^[\r\n]*$/  # 空行は対象外
+
+      # ブロックの開始/終了
+      indent_num = line.match(/^[ ]+/).to_s.size / 2
+      if block_count == indent_num
+        # 変化なし
+      elsif block_count > indent_num
+        # ブロックの終了
+        block_count = indent_num
+      else
+        # ブロックの開始
+        block_count = indent_num
+      end
+
+      #line.gsub!(/\".+\"/, "\"delete_string\"") # 文字列を削除
+      if line =~ /^\s*class\s/
+        unless line =~ /<</ # 特異クラスではない
+          work = line.gsub(/class\s/, "")
+          class_name = work.split("<")[0].to_s.chomp.match(/[A-Z][A-Za-z0-9_:]+/).to_s
+          base_name = work.split("<")[1].to_s.chomp.match(/[A-Z][A-Za-z0-9_:]+/).to_s
+          class_name.gsub!(/::/, ".")
+          class_name.gsub!(/;/, "")
+          if out_list.size != 0 and out_list[-1].type == :class_start # classが連続している
+            class_name = out_list[-1].name + "." + class_name
+            out_list[-1].name = class_name
+            cstruct_list[-1].name = class_name
+          else
+            out_list.push CStruct.new(:class_start, class_name, block_count, [], [], [], [])
+            cstruct_list.push CStruct.new(:class_end, class_name, block_count, [], [], [], [])
+          end
+          #pp line if class_name == ""
+          if base_name != ""
+            #base_name.gsub!(/::/, ".")
+            cstruct_list[-1].inherit_list.push base_name
+          end
+        end
+        next unless line =~ /end\s*$/ # 1行で終了しない場合
+      elsif line =~ /^\s*module\s/
+        module_name = line.split(" ")[1].to_s.chomp
+        module_name.gsub!(/^[:]+/, "")
+        module_name.gsub!(/::/, ".")
+        module_name.gsub!(/;/, "")
+        out_list.push CStruct.new(:module_start, module_name, block_count, [], [], [], [])
+        cstruct_list.push CStruct.new(:module_end, module_name, block_count, [], [], [], [])
+        next unless line =~ /end\s*$/ # 1行で終了しない場合
+      end
+
+      if line =~ /^\s*private$/
+        method_type = :private
+      elsif line =~ /^\s*protected$/
+        method_type = :protected
+      elsif line =~ /^\s*public$/
+        method_type = :public
+      end
+
+      if line =~ /^\s*def\s/
+        # 関数名を取り出す
+        method = line.chomp.gsub(/\s*def\s*/, "")
+        unless method =~ /\(/
+          # 関数名にカッコをつける
+          sp = method.split(" ")
+          if sp.size > 1
+            method = sp[0].to_s + "(" + sp[1..-1].to_s + ")"
+          else
+            method = method + "()"
+          end
+        end
+        if cstruct_list.size != 0
+          method_list = cstruct_list[-1].method_list
+          case method_type
+          when :public
+            method_list.push "+ #{method}"
+          when :private
+            method_list.push "- #{method}"
+          when :protected
+            method_list.push "# #{method}"
+          end
+        else
+          main_method_list.push "+ #{method}"
+        end
+      end
+
+      # composition_list
+      line.match(/(([\/\"\')].*?\.new.*?[\/\"\'])|(?![\/\"\'])([a-zA-Z0-9_]+\.new))/) do |m|
+        if m.to_s[0] != "/" and m.to_s[0] != "\"" and m.to_s[0] != "'"
+          name = m.to_s.gsub(/\.new/, "").match(/[A-Z][A-Za-z0-9_]+/).to_s
+          if name != ""
+            if cstruct_list.size != 0
+              cstruct_list[-1].composition_list.push name
+            else
+              main_composition_list.push "main *-- #{name}"
+            end
+          end
+        end
+      end
+
+      # インスタンス変数
+      if line =~ /\s*@\S+/
+        if cstruct_list.size != 0
+          line.match(/@[a-zA-Z0-9_]+/) { |m|
+            instance_var = cstruct_list[-1].var_list
+            val = m.to_s.gsub(/@/, "")
+            case method_type
+            when :public
+              instance_var.push "+ #{val}"
+            when :private
+              instance_var.push "- #{val}"
+            when :protected
+              instance_var.push "# #{val}"
+            end
+          }
+        end
+      end
+
+      # 外部変数
+      line.match(/\$[a-zA-Z0-9_]+/) { |m|
+        global_var.push "+ #{m.to_s}"
+      }
+
+      # クラスの終了
+      if cstruct_list.size != 0
+        if block_count == cstruct_list[-1].block_count # block_countが一致
+          #puts "end of #{cstruct_list[-1].name}"
+          out_list.push cstruct_list[-1]
+          cstruct_list.slice!(-1) # 最後の要素を削除
+        end
+      end
+      #puts "#{block_count} #{line.chomp}"
+    end
+    if block_count != 0
+      # エラー
+      puts f
+      return ""
+    end
+
+    # UMLの出力
+    out = print_uml(out, out_list)
+  end
+
+  if main_method_list.size != 0 or
+     main_composition_list.size != 0 or
+     main_method_list.size != 0
+    out.push "class main {"
+    main_method_list.each do |mml|
+      out.push mml
+    end
+    # グローバル変数の出力
+    global_var.uniq.each do |gv|
+      out.push gv
+    end
+    out.push "}"
+    main_composition_list.uniq.each do |mcl|
+      out.push mcl
+    end
+  end
+
+  out.push "@enduml"
+  return out.join("\n")
+end

data/lib/history/history.json

@@ -1,3 +1,3 @@
-[
-
+[
+
 ]