lib/graph.rb in aws_security_viz-0.1.2 vs lib/graph.rb in aws_security_viz-0.1.3.pre.alpha.pre.63

@@ -11,12 +11,10 @@
     log("node: #{name}")
     uniquely_add(@ops, :node, name) {
       [:node, name]
     }
   end
-  #TODO : Dedup edge labels ?
-  #TODO : Remove env dependencies
 
   def add_edge(from, to, opts)
     log("edge: #{from} -> #{to}")
     add_node(from)
     add_node(to)
@@ -40,8 +38,8 @@
     target << yield
   end
 
 
   def log(msg)
-    puts msg if ENV["DEBUG"]
+    puts msg if @config.debug?
   end
 end