goon_model_gen 0.1.0

data/lib/goon_model_gen/source/loader.rb

@@ -0,0 +1,71 @@
+require "goon_model_gen"
+
+require "erb"
+require "yaml"
+
+require "goon_model_gen/source/context"
+require "goon_model_gen/source/file"
+
+module GoonModelGen
+  module Source
+    class Loader
+      # @param filepaths [Array<String>]
+      # @return [Context]
+      def process(filepaths)
+        context = Context.new
+        filepaths.each do |filepath|
+          load_source_yaml(context, filepath)
+        end
+        return context
+      end
+
+      # @param context [Context]
+      # @param path [string]
+      def load_source_yaml(context, path)
+        erb = ERB.new(::File.read(path), nil, "-")
+        erb.filename = path
+        txt = erb.result
+        raw = YAML.load(txt)
+
+        f = File.new(path).tap do |f|
+          f.context = context
+          context.files.push(f)
+        end
+
+        (raw['types'] || {}).each do |name, t|
+          source_type =
+            if t['fields'].is_a?(Hash)
+              load_struct(f, name, t).tap do |s|
+                if g = t['goon']
+                  s.id_name = g['id_name']
+                  s.id_type = g['id_type']
+                end
+              end
+            elsif t['enum_map'].is_a?(Hash) && t['base']
+              f.new_enum(name, t['base'], t['enum_map'])
+            elsif t['slice_of'].is_a?(String)
+              f.new_named_slice(name, t['slice_of'])
+            else
+              raise "Unsupported type definition named '#{name}': #{t.inspect}"
+            end
+          source_type.generators = t['generates']
+        end
+      end
+
+      # @param f [File]
+      # @param name [String]
+      # @param t [Hash<String,Hash>|Hash<String,String>] definition of struct from YAML
+      # @return [Struct]
+      def load_struct(f, name, t)
+        f.new_struct(name).tap do |s|
+          s.ref_name = t['ref_name']
+          t['fields'].each do |field_name, attrs|
+            attrs = {'type' => attrs} unless attrs.is_a?(Hash)
+            s.new_field(field_name, attrs)
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/goon_model_gen/source/named_slice.rb

@@ -0,0 +1,18 @@
+require "goon_model_gen"
+
+require "goon_model_gen/source/type"
+
+module GoonModelGen
+  module Source
+    class NamedSlice < Type
+      attr_reader :base_type_name
+
+      # @param name [String]
+      # @param base_type [String]
+      def initialize(name, base_type_name)
+        super(name)
+        @base_type_name = base_type_name
+      end
+    end
+  end
+end

data/lib/goon_model_gen/source/struct.rb

@@ -0,0 +1,26 @@
+require "goon_model_gen"
+
+require "goon_model_gen/source/type"
+require "goon_model_gen/source/field"
+
+module GoonModelGen
+  module Source
+    class Struct < Type
+      attr_accessor :id_name, :id_type
+      attr_accessor :ref_name
+
+      def fields
+        @fields ||= []
+      end
+
+      # @param name [string]
+      # @param attrs [Hash<String,Object>]
+      def new_field(name, attrs)
+        Field.new(name, attrs).tap do |f|
+          f.context = self.context
+          self.fields.push(f)
+        end
+      end
+    end
+  end
+end

data/lib/goon_model_gen/source/type.rb

@@ -0,0 +1,21 @@
+require "goon_model_gen"
+
+require "goon_model_gen/source/contextual"
+
+module GoonModelGen
+  module Source
+    class Type
+      include Contextual
+
+      attr_reader :name
+      attr_accessor :package
+      attr_accessor :generators # key: method_name, value: true|false|string(file_suffix)
+
+      # @param name [string]
+      def initialize(name)
+        @name = name
+      end
+
+    end
+  end
+end

data/lib/goon_model_gen/templates/dsl.rb

@@ -0,0 +1,60 @@
+require "goon_model_gen"
+
+require "active_support/core_ext/string"
+
+module GoonModelGen
+  module Templates
+    module DSL
+      def dependencies
+        @dependencies ||= {}
+      end
+
+      def import(alias_or_package, package_or_nil = nil)
+        package_path = package_or_nil || alias_or_package
+        new_alias = package_or_nil ? alias_or_package.to_s : nil
+        package_path = package_path.path if package_path.respond_to?(:path)
+        return if package_path.blank?
+        if dependencies.key?(package_path)
+          old_alias = dependencies[package_path]
+          raise "Conflict alias #{old_alias.inspect} and #{new_alias.inspect}" if old_alias != new_alias
+        end
+        dependencies[package_path] ||= new_alias
+      end
+
+      def partitioned_imports(except: [])
+        pkg_paths = dependencies.keys - except
+        import_contents = partition(pkg_paths).map do |group|
+          group.map do |path|
+            ailas_name = dependencies[path]
+            ailas_name ? "\t#{ailas_name} \"#{path}\"" : "\t\"#{path}\""
+          end.join("\n")
+        end
+        import_contents.empty? ? '' : "import (\n%s\n)\n" % import_contents.join("\n\n")
+      end
+
+      PARTITION_PATTERNS = [
+        /\A[^\.\/]+(?:\/.+)?\z/,
+        /\Agopkg\.in\//,
+        /\Agolang\.org\//,
+        /\Agoogle\.golang\.org\//,
+        /\Agithub\.com\//,
+      ].freeze
+
+      def partition(paths)
+        groups = paths.group_by do |path|
+          PARTITION_PATTERNS.index{|ptn| ptn =~ path} || PARTITION_PATTERNS.length
+        end
+        groups.keys.sort.map{|k| groups[k].sort }
+      end
+
+      def user_editable(value: true)
+        @user_editable = value
+      end
+
+      def user_editable?
+        @user_editable
+      end
+
+    end
+  end
+end

data/lib/goon_model_gen/templates/model/enum/01_base.go.erb

@@ -0,0 +1,7 @@
+type <%= type.name %> <%= type.base_type.qualified_name %>
+
+const (
+<%-   type.map.each do |key, name| -%>
+  <%= name.camelize %> <%= type.name %> = <%= key.inspect %>
+<%-   end -%>
+)

data/lib/goon_model_gen/templates/model/enum/02_All.go.erb

@@ -0,0 +1,5 @@
+var All<%= type.name %> = []<%= type.name %>{
+<%-   type.map.each do |key, name| -%>
+  <%= name.camelize %>,
+<%-   end -%>
+}

data/lib/goon_model_gen/templates/model/enum/02_Map.go.erb

@@ -0,0 +1,5 @@
+var <%= type.name %>Map = map[<%= type.name %>]string{
+<%-   type.map.each do |key, name| -%>
+  <%= key.inspect %>: <%= name.inspect %>,
+<%-   end -%>
+}

data/lib/goon_model_gen/templates/model/enum/03_String.go.erb

@@ -0,0 +1,3 @@
+func (t <%= type.name %>) String() string {
+  return <%= type.name %>Map[t]
+}

data/lib/goon_model_gen/templates/model/enum/04_NameToValue.go.erb

@@ -0,0 +1,5 @@
+var <%= type.name %>NameToValue = map[string]<%= type.name %>{
+<%-   type.map.each do |key, name| -%>
+  <%= name.inspect %>: <%= key.inspect %>,
+<%-   end -%>
+}

data/lib/goon_model_gen/templates/model/enum/05_StringTo.go.erb

@@ -0,0 +1,19 @@
+<%-
+err_type_name = "InvalidEnumName"
+err_type = packages.type_for(err_type_name)
+if err_type
+  import err_type.package
+-%>
+
+func StringTo<%= type.name %>(s string) (<%= type.name %>, error) {
+  val, ok := <%= type.name %>NameToValue[s]
+  if ok {
+    return val, nil
+  } else {
+    err := &<%= err_type.qualified_name %>{Name: s}
+    return <%= type.name %>(0), err
+  }
+}
+<%-
+end
+-%>

data/lib/goon_model_gen/templates/model/enum/06_ToString.go.erb

@@ -0,0 +1,3 @@
+func <%= type.name %>ToString(v <%= type.name %>) string {
+  return v.String()
+}

data/lib/goon_model_gen/templates/model/enum/07_Valid.go.erb

@@ -0,0 +1,4 @@
+func (t <%= type.name %>) Valid() bool {
+	_, ok := <%= type.name %>Map[t]
+	return ok
+}

data/lib/goon_model_gen/templates/model/goon/01_struct.go.erb

@@ -0,0 +1,6 @@
+type <%= type.name %> struct {
+<%- type.fields.each do |field| -%>
+<%-  import field.type.package -%>
+	<%= field.definition_in(type.package) %>
+<%- end -%>
+}

data/lib/goon_model_gen/templates/model/goon/02_Assign.go.erb

@@ -0,0 +1,16 @@
+<%-
+ref_type_name = "*Ref#{type.id_field.type.name.camelize}"
+ref_type = packages.type_for(ref_type_name)
+if ref_type
+  import ref_type.package
+-%>
+func (m *<%= type.name %>) Assign(r <%= ref_type.qualified_name %>) error {
+	r.ID = m.ID
+<%-   if type.ref_name.present? -%>
+	r.Name = m.<%= type.ref_name %>
+<%-   end -%>
+	return nil
+}
+<%-
+end
+-%>

data/lib/goon_model_gen/templates/model/goon/03_PrepareToCreate.go.erb

@@ -0,0 +1,16 @@
+func (m *<%= type.name %>) PrepareToCreate() error {
+  m.PrepareFields()
+<%- if type.fields.any?{|f| f.name == "CreatedAt"} -%>
+<%-   import "time" -%>
+	if m.CreatedAt.IsZero() {
+		m.CreatedAt = time.Now()
+	}
+<%- end -%>
+<%- if type.fields.any?{|f| f.name == "UpdatedAt"} -%>
+<%-   import "time" -%>
+	if m.UpdatedAt.IsZero() {
+		m.UpdatedAt = time.Now()
+	}
+<%- end -%>
+	return nil
+}

data/lib/goon_model_gen/templates/model/goon/04_PrepareToUpdate.go.erb

@@ -0,0 +1,8 @@
+func (m *<%= type.name %>) PrepareToUpdate() error {
+	m.PrepareFields()
+<%- if type.fields.any?{|f| f.name == "UpdatedAt"} -%>
+<%-   import "time" -%>
+	m.UpdatedAt = time.Now()
+<%- end -%>
+	return nil
+}

data/lib/goon_model_gen/templates/model/goon/05_PrepareFields.go.erb

@@ -0,0 +1,8 @@
+func (m *<%= type.name %>) PrepareFields() error {
+<%- type.fields.each do |field| -%>
+<%-   if field.prepare_method.present? -%>
+  m.<%= field.name %> = m.<%= field.name %>.<%= field.prepare_method %>()
+<%-   end -%>
+<%- end -%>
+	return nil
+}

data/lib/goon_model_gen/templates/model/goon/06_IsPersisted.go.erb

@@ -0,0 +1,3 @@
+func (m *<%= type.name %>) IsPersisted() bool {
+	return m.ID != <%= type.id_field.type.qualified_name %>(0)
+}

data/lib/goon_model_gen/templates/model/slice/01_base.go.erb

@@ -0,0 +1 @@
+type <%= type.name %> []<%= type.qualified_name %>

data/lib/goon_model_gen/templates/model/struct/01_struct.go.erb

@@ -0,0 +1,6 @@
+type <%= type.name %> struct {
+<%- type.fields.each do |field| -%>
+<%-   import field.type.package -%>
+	<%= field.definition_in(type.package) %>
+<%- end -%>
+}

data/lib/goon_model_gen/version.rb

@@ -0,0 +1,3 @@
+module GoonModelGen
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,189 @@
+--- !ruby/object:Gem::Specification
+name: goon_model_gen
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- akm
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2019-05-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-stack_explorer
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Generate model source files with datastore in Golang
+email:
+- akm2000@gmail.com
+executables:
+- goon_model_gen
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- exe/goon_model_gen
+- goon_model_gen.gemspec
+- lib/goon_model_gen.rb
+- lib/goon_model_gen/cli.rb
+- lib/goon_model_gen/config.rb
+- lib/goon_model_gen/generator.rb
+- lib/goon_model_gen/golang.rb
+- lib/goon_model_gen/golang/builtin.rb
+- lib/goon_model_gen/golang/datastore_supported.rb
+- lib/goon_model_gen/golang/enum.rb
+- lib/goon_model_gen/golang/field.rb
+- lib/goon_model_gen/golang/file.rb
+- lib/goon_model_gen/golang/modifier.rb
+- lib/goon_model_gen/golang/named_slice.rb
+- lib/goon_model_gen/golang/package.rb
+- lib/goon_model_gen/golang/packages.rb
+- lib/goon_model_gen/golang/predeclared_type.rb
+- lib/goon_model_gen/golang/sentence.rb
+- lib/goon_model_gen/golang/struct.rb
+- lib/goon_model_gen/golang/type.rb
+- lib/goon_model_gen/model_builder.rb
+- lib/goon_model_gen/source/context.rb
+- lib/goon_model_gen/source/contextual.rb
+- lib/goon_model_gen/source/enum.rb
+- lib/goon_model_gen/source/field.rb
+- lib/goon_model_gen/source/file.rb
+- lib/goon_model_gen/source/loader.rb
+- lib/goon_model_gen/source/named_slice.rb
+- lib/goon_model_gen/source/struct.rb
+- lib/goon_model_gen/source/type.rb
+- lib/goon_model_gen/templates/dsl.rb
+- lib/goon_model_gen/templates/model/enum/01_base.go.erb
+- lib/goon_model_gen/templates/model/enum/02_All.go.erb
+- lib/goon_model_gen/templates/model/enum/02_Map.go.erb
+- lib/goon_model_gen/templates/model/enum/03_String.go.erb
+- lib/goon_model_gen/templates/model/enum/04_NameToValue.go.erb
+- lib/goon_model_gen/templates/model/enum/05_StringTo.go.erb
+- lib/goon_model_gen/templates/model/enum/06_ToString.go.erb
+- lib/goon_model_gen/templates/model/enum/07_Valid.go.erb
+- lib/goon_model_gen/templates/model/goon/01_struct.go.erb
+- lib/goon_model_gen/templates/model/goon/02_Assign.go.erb
+- lib/goon_model_gen/templates/model/goon/03_PrepareToCreate.go.erb
+- lib/goon_model_gen/templates/model/goon/04_PrepareToUpdate.go.erb
+- lib/goon_model_gen/templates/model/goon/05_PrepareFields.go.erb
+- lib/goon_model_gen/templates/model/goon/06_IsPersisted.go.erb
+- lib/goon_model_gen/templates/model/slice/01_base.go.erb
+- lib/goon_model_gen/templates/model/struct/01_struct.go.erb
+- lib/goon_model_gen/version.rb
+homepage: https://github.com/akm/goon_model_gen
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/akm/goon_model_gen
+  source_code_uri: https://github.com/akm/goon_model_gen
+  changelog_uri: https://github.com/akm/goon_model_gen
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Generate model source files with datastore in Golang
+test_files: []