goon_model_gen 0.1.0

data/lib/goon_model_gen/generator.rb

@@ -0,0 +1,111 @@
+require "goon_model_gen"
+
+require "goon_model_gen/templates/dsl"
+require "goon_model_gen/golang"
+
+module GoonModelGen
+  class Generator
+    include Templates::DSL
+
+    attr_reader :file, :packages
+    attr_reader :templates_dir
+
+    attr_accessor :thor
+    attr_accessor :gofmt_disabled  # false/true
+    attr_accessor :version_comment # false/true
+    attr_accessor :force, :skip    # false/true
+    attr_accessor :overwrite_custom_file   # false/true
+
+    DEFAULT_TEMPLATES_DIR = File.expand_path('../templates', __FILE__)
+
+    # @param file [Golang::File]
+    # @param templates_dir [String]
+    def initialize(file, packages, templates_dir: DEFAULT_TEMPLATES_DIR, thor: nil)
+      @file = file
+      @packages = packages
+      @templates_dir = templates_dir
+      @thor = thor
+    end
+
+    COLORS = {
+      blue:  "\e[34m",
+      clear: "\e[0m",
+    }
+
+    def run(variables = {})
+      output_path = File.join(Golang.gopath, 'src', file.package.path, file.name)
+
+      if file.custom_suffix && File.exist?(output_path) && !overwrite_custom_file
+        $stderr.puts("%sKEEP%s %s" % [COLORS[:blue], COLORS[:clear], output_path])
+        return
+      end
+
+      content = execute(variables)
+
+      options = {skip: skip, force: force}
+      thor.create_file(output_path, content, options)
+    end
+
+    def execute(variables = {})
+      variables.each do |key, val|
+        define_singleton_method(key){ val }
+      end
+
+      texts = file.sentences.sort_by(&:template_path).map do |sentence|
+        template_path = File.join(templates_dir, sentence.template_path)
+
+        # local variables used in tempaltes
+        type = sentence.type
+        package = type.package
+
+        erb = ERB.new(File.read(template_path), nil, "-")
+        erb.filename = template_path
+        erb.result(binding).strip
+      end
+
+      r = [
+        header_comments,
+        "package %s" % file.package.name,
+        partitioned_imports(except: [file.package.path]),
+        texts.join("\n\n"),
+      ].join("\n\n").strip << "\n"
+
+      r = gofmt(r) unless gofmt_disabled
+      r
+    end
+
+    def header_comments
+      r = []
+      if user_editable?
+        r << "You can edit this file. goa_model_gen doesn't overwrite this file."
+      else
+        r << "DO NOT EDIT this file."
+      end
+
+      if version_comment
+        r << "This code generated by goon_model_gen-#{GoonModelGen::VERSION}"
+      end
+
+      return r.map{|s| "// #{s}" }.join("\n")
+    end
+
+
+    def gofmt(content)
+      # https://docs.ruby-lang.org/ja/2.5.0/class/IO.html#S_POPEN
+      r = IO.popen(["gofmt"], "r+", err: :out) do |io|
+        io.puts(content)
+        io.close_write
+        io.read
+      end
+      return r unless r.empty?
+      raise "gofmt returned empty output:\n#{content}"
+    end
+
+    # @param cfg [Config]
+    def load_config(cfg)
+      [:gofmt_disabled, :version_comment].each do |key|
+        self.send("#{key}=", cfg.send(key))
+      end
+    end
+  end
+end

data/lib/goon_model_gen/golang.rb

@@ -0,0 +1,22 @@
+require "goon_model_gen"
+
+require "goon_model_gen/golang/package"
+require "goon_model_gen/golang/builtin"
+
+module GoonModelGen
+  module Golang
+    class << self
+
+      def gopath
+        # Instead of $GOPATH
+        `go env GOPATH`.strip
+      end
+
+      # @return [Package]
+      def builtin_package
+        Buildin.package
+      end
+
+    end
+  end
+end

data/lib/goon_model_gen/golang/builtin.rb

@@ -0,0 +1,31 @@
+require "goon_model_gen"
+
+require "goon_model_gen/golang/predeclared_type"
+require "goon_model_gen/golang/package"
+
+module GoonModelGen
+  module Golang
+    class Builtin < PredeclaredType
+      TYPE_NAMES =
+        %w[bool byte complex128 complex64
+           error float32 float64 int int16 int32 int64 int8
+           rune string uint uint16 uint32 uint64 uint8 uintptr]
+
+      class << self
+        def package
+          @package ||= Package.new(nil).tap do |pkg|
+            instances.each do |i|
+              i.package = pkg
+              pkg.types.push(i)
+            end
+          end
+        end
+
+        def instances
+          @instances ||= TYPE_NAMES.map{|name| self.new(name) }
+        end
+      end
+
+    end
+  end
+end

data/lib/goon_model_gen/golang/datastore_supported.rb

@@ -0,0 +1,48 @@
+require "goon_model_gen"
+
+require "goon_model_gen/golang/predeclared_type"
+require "goon_model_gen/golang/package"
+require "goon_model_gen/golang/packages"
+require "goon_model_gen/golang/builtin"
+
+module GoonModelGen
+  module Golang
+    class DatastoreSupported < PredeclaredType
+
+      class << self
+
+        def datastore
+          @datastore ||= Package.new('google.golang.org/appengine/datastore').tap do |pkg|
+            pkg.add(self.new('ByteString'))
+            pkg.add(self.new('Key'))
+          end
+        end
+
+        def time
+          @time ||= Package.new('time').tap do |pkg|
+            pkg.add(self.new('Time'))
+          end
+        end
+
+        def appengine
+          @appengine ||= Package.new('google.golang.org/appengine').tap do |pkg|
+            pkg.add(self.new('BlobKey'))
+            pkg.add(self.new('GeoPoint'))
+          end
+        end
+
+
+        def packages
+          @packages ||= Packages.new.tap do |pkgs|
+            pkgs << Builtin.package
+            pkgs << datastore
+            pkgs << time
+            pkgs << appengine
+          end
+        end
+
+      end
+
+    end
+  end
+end

data/lib/goon_model_gen/golang/enum.rb

@@ -0,0 +1,26 @@
+require "goon_model_gen"
+
+require "goon_model_gen/golang/type"
+
+module GoonModelGen
+  module Golang
+    class Enum < Type
+      attr_reader :base_type_name, :map
+      attr_reader :base_type
+
+      # @param name [String]
+      # @param base_type_name [String]
+      # @param map [Hash<Object,String>]
+      def initialize(name, base_type_name, map)
+        super(name)
+        @base_type_name = base_type_name
+        @map = map
+      end
+
+      # @param pkgs [Packages]
+      def resolve(pkgs)
+        @base_type = pkgs.type_for(base_type_name) || raise("#{base_type_name.inspect} not found")
+      end
+    end
+  end
+end

data/lib/goon_model_gen/golang/field.rb

@@ -0,0 +1,41 @@
+require "goon_model_gen"
+
+module GoonModelGen
+  module Golang
+    class Field
+      attr_reader :name, :type_name
+      attr_reader :tags # Hash<string,Array[string]> ex. for datastore, validate, json, etc...
+      attr_reader :goon_id # true/false
+      attr_reader :type
+      attr_accessor :struct
+      attr_accessor :prepare_method
+
+      def initialize(name, type_name, tags, goon_id: false)
+        @name, @type_name = name, type_name
+        @tags = tags || {}
+        @goon_id = goon_id
+      end
+
+      # @param pkgs [Packages]
+      def resolve(pkgs)
+        @type = pkgs.type_for(type_name) || raise("#{type_name.inspect} not found for #{struct.qualified_name}.#{name}")
+      end
+
+      def tags_string
+        tags.keys.sort.map do |key|
+          val = tags[key]
+          vals = val.is_a?(Array) ? val.join(',') : val.to_s
+          vals.empty? ? nil : "#{key}:\"#{vals}\""
+        end.compact.join(' ')
+      end
+
+      # @param pkg [Package]
+      # @return [string]
+      def definition_in(pkg)
+        type_exp =
+          (type.package.path == pkg.path) ? type.name : type.qualified_name
+        "#{ name } #{ type_exp } `#{ tags_string }`"
+      end
+    end
+  end
+end

data/lib/goon_model_gen/golang/file.rb

@@ -0,0 +1,27 @@
+require "goon_model_gen"
+
+require "goon_model_gen/golang/sentence"
+
+module GoonModelGen
+  module Golang
+    class File
+      attr_reader :package, :name
+      attr_reader :sentences
+      attr_accessor :custom_suffix # false/true
+
+      # @param package [Package]
+      # @param name [string]
+      def initialize(package, name)
+        @package = package
+        @name = name
+        @sentences = []
+      end
+
+      def new_sentence(template_path, type)
+        Sentence.new(template_path, type).tap do |s|
+          sentences.push(s)
+        end
+      end
+    end
+  end
+end

data/lib/goon_model_gen/golang/modifier.rb

@@ -0,0 +1,65 @@
+require "goon_model_gen"
+
+require "goon_model_gen/golang/type"
+
+module GoonModelGen
+  module Golang
+    class Modifier < Type
+      attr_reader :prefix, :target_name
+      attr_reader :target
+      def initialize(prefix, target)
+        @prefix = prefix
+        case target
+        when Type
+          @target = target
+          @target_name = target.name
+        else
+          @target = nil
+          @target_name = target.to_s
+        end
+      end
+
+      def package
+        target.package
+      end
+
+      def name
+        prefix + target.name
+      end
+
+      def qualified_name
+        prefix + target.qualified_name
+      end
+
+      # @param pkgs [Packages]
+      def resolve(pkgs)
+        return if target
+        @target = pkgs.type_for(target_name) || raise("#{target_name.inspect} not found")
+      end
+
+      PATTERN = /\A\*|\A\[\d*\]/
+
+      class << self
+        # @param s [String]
+        # @return [Proc]
+        def parse(s)
+          parts = parse_expression(s)
+          t = yield(parts.shift)
+          return nil unless t
+          parts.each do |part|
+            t = new(part, t)
+          end
+          t
+        end
+
+        # @param s [String]
+        # @return [Array<String>]
+        def parse_expression(s)
+          m = s.match(PATTERN)
+          m.nil? ? [s] : parse_expression(s.sub(PATTERN, '')) + [m[0]]
+        end
+
+      end
+    end
+  end
+end

data/lib/goon_model_gen/golang/named_slice.rb

@@ -0,0 +1,24 @@
+require "goon_model_gen"
+
+require "goon_model_gen/golang/type"
+
+module GoonModelGen
+  module Golang
+    class NamedSlice < Type
+      attr_reader :base_type_name
+      attr_reader :base_type
+
+      # @param name [String]
+      # @param base_type_name [String]
+      def initialize(name, base_type_name)
+        super(name)
+        @base_type_name = base_type_name
+      end
+
+      # @param pkgs [Packages]
+      def resolve(pkgs)
+        @base_type = pkgs.type_for(base_type_name) || raise("#{base_type_name.inspect} not found")
+      end
+    end
+  end
+end

data/lib/goon_model_gen/golang/package.rb

@@ -0,0 +1,83 @@
+require "goon_model_gen"
+
+require "goon_model_gen/golang/struct"
+require "goon_model_gen/golang/enum"
+require "goon_model_gen/golang/named_slice"
+require "goon_model_gen/golang/file"
+
+module GoonModelGen
+  module Golang
+    class Package
+      attr_reader :path
+      attr_reader :types
+      attr_reader :files
+
+      # @param path [String]
+      def initialize(path)
+        @path = path
+        @types = []
+        @files = []
+      end
+
+      def basename
+        @basename ||= path ? ::File.basename(path, '.*') : nil
+      end
+
+      def name
+        @name ||= basename ? basename.gsub(/[\-\_]/, '').downcase : nil
+      end
+
+      def add(type)
+        types.push(type)
+        type.package = self
+      end
+
+      def new_struct(name)
+        Struct.new(name).tap{|s| add(s) }
+      end
+
+      # @param name [string]
+      # @param base_type [String]
+      # @param map [Hash<Object,Hash>] elements of enum from YAML
+      # @return [Enum]
+      def new_enum(name, base_type, map)
+        Enum.new(name, base_type, map).tap{|s| add(s) }
+      end
+
+      # @param name [string]
+      # @param base_type_name [String]
+      # @return [Slice]
+      def new_named_slice(name, base_type_name)
+        NamedSlice.new(name, "[]#{base_type_name}").tap{|s| add(s) }
+      end
+
+      # @param name [string]
+      # @return [File]
+      def find_or_new_file(name)
+        files.detect{|f| f.name == name} || new_file(name)
+      end
+
+      # @param name [string]
+      # @return [File]
+      def new_file(name)
+        File.new(self, name).tap do |f|
+          files.push(f)
+        end
+      end
+
+      # @return [Hash<String,Type>]
+      def name_to_type_map
+        @name_to_type_map ||= types.each_with_object({}) do |t,d|
+          d[t.name] = t
+        end
+      end
+
+      # @param name [string]
+      # @param [Type]
+      def lookup(name)
+        name_to_type_map[name]
+      end
+
+    end
+  end
+end