lib/review/epubmaker.rb in review-5.7.0 vs lib/review/epubmaker.rb in review-5.8.0

@@ -15,10 +15,11 @@
 require 'review/latexbuilder'
 require 'review/version'
 require 'review/htmltoc'
 require 'review/htmlbuilder'
 require 'review/img_math'
+require 'review/img_graph'
 
 require 'rexml/document'
 require 'rexml/streamlistener'
 require 'review/call_hook'
 require 'review/epubmaker/producer'
@@ -39,10 +40,11 @@
       @producer = nil
       @htmltoc = nil
       @buildlogtxt = 'build-log.txt'
       @logger = ReVIEW.logger
       @img_math = nil
+      @img_graph = nil
       @basedir = nil
     end
 
     def self.execute(*args)
       self.new.execute(*args)
@@ -280,15 +282,16 @@
       @postcount = 0
 
       @manifeststr = ''
       @ncxstr = ''
       @tocdesc = []
+      @img_graph = ReVIEW::ImgGraph.new(@config, 'html', path_name: '_review_graph')
 
       basedir = File.dirname(yamlfile)
       base_path = Pathname.new(basedir)
       book = ReVIEW::Book::Base.new(basedir, config: @config)
-      @converter = ReVIEW::Converter.new(book, ReVIEW::HTMLBuilder.new(img_math: @img_math))
+      @converter = ReVIEW::Converter.new(book, ReVIEW::HTMLBuilder.new(img_math: @img_math, img_graph: @img_graph))
       @compile_errors = nil
 
       book.parts.each do |part|
         if part.name.present?
           if part.file?
@@ -308,16 +311,24 @@
         part.chapters.each do |chap|
           build_chap(chap, base_path, basetmpdir, false)
         end
       end
       check_compile_status
+
+      begin
+        @img_graph.make_mermaid_images
+      rescue ApplicationError => e
+        error! e.message
+      end
+      @img_graph.cleanup_graphimg
     end
 
     def build_part(part, basetmpdir, htmlfile)
       debug("Create #{htmlfile} from a template.")
       File.open(File.join(basetmpdir, htmlfile), 'w') do |f|
         @part_number = part.number
         @part_title = part.name.strip
+        @title = @part_title
         @body = ReVIEW::Template.generate(path: template_name(localfile: '_part_body.html.erb', systemfile: 'html/_part_body.html.erb'), binding: binding)
         @language = @producer.config['language']
         @stylesheets = @producer.config['stylesheet']
         f.write ReVIEW::Template.generate(path: template_name, binding: binding)
       end