Extract visualization to superclass

lib/mosaik/command.rb

@@ -31,6 +31,25 @@ class Graph < Command
       def validate
         raise OptionError, "unknown renderer: #{options[:renderer]}" unless options[:renderer].in? ["dot", "fdp", "sfdp", "neato"]
       end
+
+      protected
+
+      def visualize
+        # Write graph to file
+        File.write(options[:file], graph.to_csv)
+
+        info "Dependency graph written to #{options[:file]}"
+
+        return unless options[:visualize]
+
+        file = File.basename(options[:file], ".*")
+
+        # Write visualization to file
+        debug graph.to_dot(options)
+        graph.to_svg(file, options)
+
+        info "Dependency graph written to #{file}.gv and rendered to #{file}.svg"
+      end
     end
   end
 end

lib/mosaik/commands/evaluate.rb

@@ -46,20 +46,11 @@ def call
           debug "Components: #{cluster.vertices.map(&:id).join(', ')}"
         end
 
-        file = "#{File.basename(options[:file], '.*')}-evaluation"
+        # Change file name
+        options[:file] = "#{File.basename(options[:file], '.*')}-evaluation"
 
         # Write graph to file
-        File.write("#{file}.csv", graph.to_csv)
-
-        info "Dependency graph written to #{options[:file]}"
-
-        return unless options[:visualize]
-
-        # Write visualization to file
-        debug graph.to_dot(options)
-        graph.to_png(file, options)
-
-        info "Evaluated microservice candidate graph written to #{file}.gv and rendered to #{file}.png"
+        visualize
       end
 
       private

lib/mosaik/commands/extract.rb

@@ -45,20 +45,8 @@ def call
           .tap(&:validate)
           .call
 
-        # Write graph to file
-        File.write(options[:file], graph.to_csv)
-
-        info "Dependency graph written to #{options[:file]}"
-
-        return unless options[:visualize]
-
-        file = File.basename(options[:file], ".*")
-
-        # Write visualization to file
-        debug graph.to_dot(options)
-        graph.to_svg(file, options)
-
-        info "Dependency graph written to #{file}.gv and rendered to #{file}.svg"
+        # Write graph to file(s)
+        visualize
       end
 
       private

lib/mosaik/commands/identify.rb

@@ -67,20 +67,11 @@ def call
           end
         end
 
-        file = "#{File.basename(options[:file], '.*')}-candidates"
+        # Change file name
+        options[:file] = "#{File.basename(options[:file], '.*')}-candidates"
 
         # Write graph to file
-        File.write("#{file}.csv", graph.to_csv)
-
-        info "Dependency graph written to #{options[:file]}"
-
-        return unless options[:visualize]
-
-        # Write visualization to file
-        debug graph.to_dot(options)
-        graph.to_svg(file, options)
-
-        info "Microservice candidate graph written to #{file}.gv and rendered to #{file}.svg"
+        visualize
       end
 
       private