Module: Ar2gostruct

Defined in:
lib/ar2gostruct.rb,
lib/ar2gostruct/railtie.rb,
lib/ar2gostruct/version.rb

Defined Under Namespace

Classes: Railtie

Constant Summary collapse

MODEL_DIR = 
ENV["model_dir"] || "app/models"
TYPE_MAP = 
{
  "string"     => "string",
  "text"       => "string",
  "boolean"    => "bool",
  "integer(1)" => "int8",
  "integer(2)" => "int16",
  "integer(3)" => "int32",
  "integer(4)" => "int32",
  "integer(8)" => "int64",
  "float"      => "float64",
  "datetime"   => "time.Time",
  "date"       => "time.Time"
}
VERSION = 
"0.0.3"

Class Method Summary collapse

Class Method Details

.convert! ⇒ Object 



97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
 
# File 'lib/ar2gostruct.rb', line 97

def self.convert!
  annotated = []
  self.get_model_names.each do |m|
    class_name = m.sub(/\.rb$/,'').camelize
    begin
      klass = class_name.split('::').inject(Object){ |klass,part| klass.const_get(part) }
      if klass < ActiveRecord::Base && !klass.abstract_class?
        annotated << class_name
        self.convert_to_gostruct(klass)
      end
    rescue Exception => e
      puts "// Unable to convert #{class_name}: #{e.message}"
    end

  end
end

.convert_to_gostruct(klass) ⇒ Object 



80
81
82
83
84
85
86
87
 
# File 'lib/ar2gostruct.rb', line 80

def self.convert_to_gostruct(klass)
  info = get_schema_info(klass)

  model_file_name = File.join(MODEL_DIR, klass.name.underscore + ".rb")

  puts "// #{model_file_name}"
  puts info
end

.get_model_names ⇒ Object 



89
90
91
92
93
94
95
 
# File 'lib/ar2gostruct.rb', line 89

def self.get_model_names
  models = []
  Dir.chdir(MODEL_DIR) do
    models = Dir["**/*.rb"]
  end
  models
end

.get_schema_info(klass) ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
 
# File 'lib/ar2gostruct.rb', line 33

def self.get_schema_info(klass)
  info = "// Table name: #{klass.table_name}\n"
  info << "type #{klass.table_name.camelize} struct {\n"

  max_size = klass.column_names.collect{|name| name.size}.max + 1
  klass.columns.each do |col|
    tags = []

    # add json tag
    tags << "json:\"#{col.name}\""

    case ENV["orm"]
    when "qbs"
      orm_option = []
      # primary key
      if col.name == klass.primary_key
        orm_option << "pk"
      end
      # not null Constraint
      unless col.null
        orm_option << "notnull"
      end
      # default value
      if col.default
        orm_option << "default:'#{col.default}'"
      end
      if orm_option.present?
        tags << "qbs:\"#{orm_option.join(",")}\""
      end
    end

    col_type = col.type.to_s
    case col_type
    when "integer"
      type = TYPE_MAP["integer(#{col.limit})"] || "int32"
      type = "u#{col_type}" if col.sql_type.match("unsigned").present?
    else
      type = TYPE_MAP[col_type] || "string"
    end

    info << sprintf("\t%-#{max_size}.#{max_size}s%-15.15s'%s'\n", col.name.camelize, type, tags.join(" "))

  end

  info << "}\n\n"
end

.load ⇒ Object 



25
26
27
28
29
30
31
 
# File 'lib/ar2gostruct.rb', line 25

def self.load
  begin
    require "#{Dir.pwd}/config/environment"
  rescue
  end
  Rails.application.eager_load!
end