root/lang/ruby/scaflute/scaflute.rb @ 30165

require 'rexml/document'
require 'erb'

# DBFlute が生成したスキーマ情報を元に、
# Teeda 用の CRUD ページを生成する scaffold スクリプト（予定）。

module Main
  def self.run()
    tables = Schema.read_xml('templates/project-schema-sample.xml')
    tables.each() do |table|
      $table = table
      $table.javaPackage = 'com.sample'
      ERB.exec_file('templates/dxo.erb', "result/java/#{table.javaClass}Dxo.java")
      ERB.exec_file('templates/page.erb', "result/java/Abstract#{table.javaClass}Page.java")
      ERB.exec_file('templates/list.erb', "result/html/#{table.javaProperty}List.html")
      ERB.exec_file('templates/list_page.erb', "result/java/#{table.javaClass}ListPage.java")
      ERB.exec_file('templates/edit.erb', "result/html/#{table.javaProperty}Edit.html")
      ERB.exec_file('templates/edit_page.erb', "result/java/#{table.javaClass}EditPage.java")
      ERB.exec_file('templates/confirm.erb', "result/html/#{table.javaProperty}Confirm.html")
      ERB.exec_file('templates/confirm_page.erb', "result/java/#{table.javaClass}ConfirmPage.java")
    end
  end
end

class BsTable
  attr_accessor :name,
                :schema,
                :type,
                :columns
end

class BsColumn
  attr_accessor :autoIncrement,
                :comment,
                :dbType,
                :default,
                :javaType,
                :name,
                :size,
                :primaryKey,
                :required,
                :type
end

class Table < BsTable
  attr_accessor :javaClass,
                :javaProperty,
                :javaPackage
end

class Column < BsColumn
  attr_accessor :javaProperty,
                :htmlMaxLength
end

class ERB
  def self.exec_file(file_name, output = nil)
    File.open(file_name, 'r') do |f|
      str = f.read()
      erb = ERB.new(str, nil, '-')
      if output.nil?
        erb.run()
      else
        File.open(output, 'w+') do |f|
          f.puts(erb.result)
        end
      end
    end
  end
end

class String
  # 文字列を CamelCase に変換
  # FOO_BAR -> fooBar
  def camelize()
    str = self.downcase()
    str = str.gsub(/_+([a-z]?)/) do
      $1.upcase()
    end
    return str
  end

  # 最初の文字を大文字に変換
  # fooBar -> FooBar
  def upcase_first()
    str = String.new(self)
    str[0] = str[0].chr().upcase()
    return str
  end
end

module Schema
  def self.read_xml(file_name)
    result = []
    read_document(file_name).root.elements.each() do |element|
      result << read_table(element)
    end
    return result
  end

private

  def self.read_document(file_name)
    doc = nil
    File.open(file_name, 'r') do |f|
      doc = REXML::Document.new(f)
    end
    return doc
  end

  def self.read_table(element)
    table = Table.new()
    table.name = element.attributes['name']
    table.schema = element.attributes['schema']
    table.type = element.attributes['type']
    table.columns = []
    table.javaProperty = table.name.camelize()
    table.javaClass = table.javaProperty.upcase_first()

    element.elements.each() do |element|
      next unless element.name == 'column'
      table.columns << read_column(element)
    end
    return table
  end

  def self.read_column(element)
    column = Column.new()
    column.autoIncrement = element.attributes['autoIncrement']
    column.comment = element.attributes['comment']
    column.dbType = element.attributes['dbType']
    column.default = element.attributes['default']
    column.javaType = element.attributes['javaType']
    column.name = element.attributes['name']
    column.size = element.attributes['size']
    column.primaryKey = element.attributes['primaryKey']
    column.required = element.attributes['required']
    column.type = element.attributes['type']
    column.javaProperty = column.name.camelize()
    column.htmlMaxLength = read_maxlength(column)
    return column
  end

  def self.read_maxlength(column)
    case column.javaType
    when 'java.util.Date'
      return '10'
    when 'java.math.BigDecimal'
      if column.size =~ /(\d+), (\d+)/
        return ($1.to_i() + $2.to_i()).to_s()
      end
    when 'Long'
      return '19'
    when 'Integer'
      return '10'
    end
    return column.size
  end
end

Main.run()