lib/shellplay/screen.rb in shellplay-0.1.1 vs lib/shellplay/screen.rb in shellplay-0.1.2

@@ -1,8 +1,12 @@
+require "shell2html"
+
 module Shellplay
   class Screen
 
+    include Shell2html
+
     attr_reader :stdin, :stdout, :stderr, :display, :timespent, :displaycommand, :playprompt, :clearscreen, :customprompt
 
     def initialize
       @displaycommand = true
       @playprompt = true
@@ -34,9 +38,13 @@
         stdin: @stdin,
         stdout: @stdout,
         stderr: @stderr,
         timespent: @timespent
       }
+    end
+
+    def html
+      to_html(@stdout)
     end
 
   end
 end