tyteen4a03 · April 4, 2020 13:04
diff --git a/gistfile1.txt b/gistfile1.txt
 require 'cgi'
 require 'shellwords'

 class ConvertToPDF
  PDF_OPTIONS = {
    page_size: 'A4'
  }.freeze

  def initialize(params = {})
    if !params.key?(:from) || params[:from].nil?
      raise ArgumentError.new 'where is the codebase you want to convert to PDF?'
    elsif !valid_directory?(params[:from])
      raise LoadError.new "#{params[:from]} not found"
    elsif !params.key?(:to) || params[:to].nil?
      raise ArgumentError.new 'where should I save the generated pdf file?'
    else
      @from, @to, @except = params[:from], params[:to], params[:except].to_s

      if File.exist?(@except) && invalid_blacklist?
        raise LoadError.new "#{@except} is not a valid blacklist YAML file"
      end

      save
    end
  end

  private

  def save
    File.write(@to + '.html', pdf)
    #pdf.to_file(@to)
  end

  def pdf
    style = Rouge::Themes::Base16.mode(:light).render(scope: '')
    n = "        

    .rouge-table {
      margin-bottom: 2.5rem;
    }"
    html ||= "<html><head>
      <style type=\"text/css\">
      @page {
        size: A4;
        margin: 0;
      }
      @media print {
        html, body {
          width: 210mm;
          height: 297mm;
        }
      }

      @media print, screen {
        #{style}

        .filename {
          font-size: 1.5rem;
          font-family: Helvetica Neue, sans-serif;
        }
  
        table {
          margin-bottom: 2rem;
        }
  
        table pre {
          font-size: 1rem;
        }
      }
      </style>
    </head><body>"

    #style = 'font-size: 16px; font-family: Helvetica, sans-serif;'

    read_files.each do |file|
      html += "<strong class=\"filename\">File: #{file.first}</strong></br></br>"
      html += prepare_line_breaks(syntax_highlight(file)).to_s
    end

    html += '</body></html>'

    html
    #@kit = PDFKit.new(html, page_size: 'A4')
    #@kit
  end

  def syntax_highlight(file)
    file_type = File.extname(file.first)[1..-1]
    file_lexer = Rouge::Lexer.find(file_type)
    return CGI.escapeHTML(file.last) unless file_lexer

    formatter = Rouge::Formatters::HTML.new
    #formatter = Rouge::Formatters::HTMLInline.new(theme)
    #formatter = Rouge::Formatters::HTMLPygments.new(formatter, class: 'line-%i')
    formatter = Rouge::Formatters::HTMLTable.new(formatter, start_line: 1)
    code_data = file.last.encode('UTF-8', 'binary', invalid: :replace, undef: :replace, replace: '')
    formatter.format(file_lexer.lex(code_data))
  end

  def invalid_blacklist?
    return true if FileTest.directory?(@except)

    @blacklist = YAML.load_file(@except)

    [email protected]?(:directories) || [email protected]?(:files)
  end

  def in_directory_blacklist?(item_path)
    @blacklist[:directories].include?(item_path.gsub("#{@from}/", '')) if @blacklist
  end

  def in_file_blacklist?(item_path)
    if @blacklist
      @blacklist[:files].include?(item_path.split('/').last) || @blacklist[:files].include?(item_path.gsub("#{@from}/", ''))
    end
  end

  def valid_directory?(dir)
    File.exist?(dir) && FileTest.directory?(dir)
  end

  def valid_file?(file)
    File.exist?(file) && FileTest.file?(file)
  end

  def read_files(path = nil)
    @files ||= []
    path   ||= @from

    Dir.foreach(path) do |item|
      item_path = "#{path}/#{item}"

      if valid_directory?(item_path) && !%w[. ..].include?(item) && !in_directory_blacklist?(item_path)
        read_files(item_path)
      elsif valid_file?(item_path) && !in_file_blacklist?(item_path)
        @files << [item_path, process_file(item_path)]
      end
    end

    @files
  end

  def process_file(file)
    puts "Reading file #{file}"

    content = ''
    File.open(file, 'r') do |f|
      if `file #{file.shellescape}` !~ /text/
        content << '[binary]'
      else
        f.each_line { |line_content| content << line_content }
      end
    end
    content
  end

  def prepare_line_breaks(content)
    content.gsub(/\n/, '<br>')
  end

  def add_space(height)
    "<div style='margin-bottom: #{height}px'>&nbsp;</div>"
  end
 end
	require 'cgi'
	require 'shellwords'

	class ConvertToPDF
	PDF_OPTIONS = {
	page_size: 'A4'
	}.freeze

	def initialize(params = {})
	if !params.key?(:from) \|\| params[:from].nil?
	raise ArgumentError.new 'where is the codebase you want to convert to PDF?'
	elsif !valid_directory?(params[:from])
	raise LoadError.new "#{params[:from]} not found"
	elsif !params.key?(:to) \|\| params[:to].nil?
	raise ArgumentError.new 'where should I save the generated pdf file?'
	else
	@from, @to, @except = params[:from], params[:to], params[:except].to_s

	if File.exist?(@except) && invalid_blacklist?
	raise LoadError.new "#{@except} is not a valid blacklist YAML file"
	end

	save
	end
	end

	private

	def save
	File.write(@to + '.html', pdf)
	#pdf.to_file(@to)
	end

	def pdf
	style = Rouge::Themes::Base16.mode(:light).render(scope: '')
	n = "

	.rouge-table {
	margin-bottom: 2.5rem;
	}"
	html \|\|= "<html><head>
	<style type=\"text/css\">
	@page {
	size: A4;
	margin: 0;
	}
	@media print {
	html, body {
	width: 210mm;
	height: 297mm;
	}
	}

	@media print, screen {
	#{style}

	.filename {
	font-size: 1.5rem;
	font-family: Helvetica Neue, sans-serif;
	}

	table {
	margin-bottom: 2rem;
	}

	table pre {
	font-size: 1rem;
	}
	}
	</style>
	</head><body>"

	#style = 'font-size: 16px; font-family: Helvetica, sans-serif;'

	read_files.each do \|file\|
	html += "<strong class=\"filename\">File: #{file.first}</strong></br></br>"
	html += prepare_line_breaks(syntax_highlight(file)).to_s
	end

	html += '</body></html>'

	html
	#@kit = PDFKit.new(html, page_size: 'A4')
	#@kit
	end

	def syntax_highlight(file)
	file_type = File.extname(file.first)[1..-1]
	file_lexer = Rouge::Lexer.find(file_type)
	return CGI.escapeHTML(file.last) unless file_lexer

	formatter = Rouge::Formatters::HTML.new
	#formatter = Rouge::Formatters::HTMLInline.new(theme)
	#formatter = Rouge::Formatters::HTMLPygments.new(formatter, class: 'line-%i')
	formatter = Rouge::Formatters::HTMLTable.new(formatter, start_line: 1)
	code_data = file.last.encode('UTF-8', 'binary', invalid: :replace, undef: :replace, replace: '')
	formatter.format(file_lexer.lex(code_data))
	end

	def invalid_blacklist?
	return true if FileTest.directory?(@except)

	@blacklist = YAML.load_file(@except)

	[email protected]?(:directories) \|\| [email protected]?(:files)
	end

	def in_directory_blacklist?(item_path)
	@blacklist[:directories].include?(item_path.gsub("#{@from}/", '')) if @blacklist
	end

	def in_file_blacklist?(item_path)
	if @blacklist
	@blacklist[:files].include?(item_path.split('/').last) \|\| @blacklist[:files].include?(item_path.gsub("#{@from}/", ''))
	end
	end

	def valid_directory?(dir)
	File.exist?(dir) && FileTest.directory?(dir)
	end

	def valid_file?(file)
	File.exist?(file) && FileTest.file?(file)
	end

	def read_files(path = nil)
	@files \|\|= []
	path \|\|= @from

	Dir.foreach(path) do \|item\|
	item_path = "#{path}/#{item}"

	if valid_directory?(item_path) && !%w[. ..].include?(item) && !in_directory_blacklist?(item_path)
	read_files(item_path)
	elsif valid_file?(item_path) && !in_file_blacklist?(item_path)
	@files << [item_path, process_file(item_path)]
	end
	end

	@files
	end

	def process_file(file)
	puts "Reading file #{file}"

	content = ''
	File.open(file, 'r') do \|f\|
	if `file #{file.shellescape}` !~ /text/
	content << '[binary]'
	else
	f.each_line { \|line_content\| content << line_content }
	end
	end
	content
	end

	def prepare_line_breaks(content)
	content.gsub(/\n/, '<br>')
	end

	def add_space(height)
	"<div style='margin-bottom: #{height}px'> </div>"
	end
	end