weapon 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: dd05b8383fdf6a56aee99e98e7c4e6aadfd800e4
-  data.tar.gz: caf2a88d30150d5b98fd8824fdb5af1b90ba5e90
+  metadata.gz: 8621ad57230f376e88746ffb2ca40f8957f7d483
+  data.tar.gz: 969cb6e67bdc34eec0263004c0887896edc189f8
 SHA512:
-  metadata.gz: 40ed25df93a88f72b4f1f14ae96a59de044d39fd23563b7943c8aa05dcaf9accaf2117b58bfb18529d2db826ebf522674417e504038ec5f6a322771f04fed121
-  data.tar.gz: 026e9afc4441a43fbef02c8ca2c9f73de6c12866b5a034927420daf0ff1ca680f265e658f538d1cd203f2586040d938059380e25e5034b49ded79ca06eb86f3a
+  metadata.gz: 5641986317808d41699ffa477bd7e34c5edbeaedb5cd8dd5524f3f3649ab3f923fabd85c6a62bff9cc1a79b3c7d3ce7359162d6d0e20f3b15ec8ae03703f276d
+  data.tar.gz: 015e1e722924516d70c47c74aaeaa305f33e3a087993daf3cdd7ebe5edb4fa473f4984d0c694d8423105b62682df12bd2f5765d01b6ebd701ce0d101e86ac0af

data/lib/generators/weapon/activeadmin/activeadmin_generator.rb

@@ -1,36 +1,54 @@
 
 class Weapon::ActiveadminGenerator < Rails::Generators::NamedBase # :nodoc:
-  desc 'This generator creates a example email setting and add exception_notification for production'
-
-  def create_example_email
-    application(nil, env: "production") do
-      config.action_mailer.raise_delivery_errors = true
-      config.action_mailer.default_url_options = {:host => '120.25.61.54:8100'}
-      config.action_mailer.delivery_method = :smtp
-      config.action_mailer.smtp_settings = {
-        :address => "smtp.163.com",
-        :port    => 25,
-        :user_name => "ledcloud@163.com",
-        :password => "ffgexobhlolfjhij",
-        :enable_starttls_auto => true,
-        :authentication   => "plain",
-        :domain => "163.com"
-      }
-    end
-  end
+  desc 'This generator creates activeadmin files'
 
-  def add_exception_notify
-    gem_group :production do
-      gem "rspec-rails"
-    end
+  class_option :only, type: :string, desc: 'only generate model for listed model'
+  class_option :except, type: :string, desc: 'when generate model, exclude listed model'
+
+  def generate_activeadmin
+
+    prefix = 'app/admin'
+
+    if args[0].nil? || args[0] == "all"
+      names = Dir["app/models/*.rb"].map {|x| File.basename(x, '.rb')}
+
+      if options[:only]
+        names = names & options[:only].split(',')
+      end
+
+      if options[:except]
+        names = names - options[:except].split(',')
+      end
+
+      names.each do |name|
+        next unless name.camelize.constantize.respond_to? 'column_names'
+        column_names = name.camelize.constantize.column_names
+        infos = column_names.map {|column| "expose :#{column}"}
+      end
+    else
+      name = args[0]
+      return unless name.camelize.constantize.respond_to? 'column_names'
+      column_names = name.camelize.constantize.column_names
+
+      if options[:only]
+        column_names = column_names & options[:only].split(',')
+      end
+
+      if options[:except]
+        column_names = column_names - options[:except].split(',')
+      end
+
+      ap column_names
+      a = Magicfile.new
+      infos = ['xx']
+      a.append_string_lines(infos)
+      a.to_file "#{prefix}/name.rb"
 
-    application(nil, env: "production") do
-      config.middleware.use ExceptionNotification::Rack,
-        :email => {
-          :email_prefix => "[weapon] ",
-          :sender_address => "weapon <ledcloud@163.com>",
-          :exception_recipients => %w{dilin.life@gmail.com}
-        }
     end
+
+
+
   end
+
+
 end

data/lib/generators/weapon/grape/grape_generator.rb

@@ -1,20 +1,19 @@
 require 'magicfile'
 require 'fileutils'
-
+require 'thor'
 
 class Weapon::GrapeGenerator < Rails::Generators::Base # :nodoc:
-  desc 'This generator creates grape entity for model'
+  desc 'This generator creates basic grape files for models'
+  include Thor::Actions
 
-  def generate_grape_entity
+  source_root File.expand_path("../templates", __FILE__)
+  class_option :only, type: :string, desc: 'only generate api for listed model'
+  class_option :except, type: :string, desc: 'when generate api, exclude listed model'
 
-    prefix_dirs = Dir.glob('app/api/*/')
-    if prefix_dirs.count == 0
-      prefix = 'app/api/v1'
-    elsif prefix_dirs.count == 1
-      prefix = prefix_dirs[0]
-    else
-      prefix = ask('there are multi directory in app/api directory, please input path like app/api/v1')
-    end
+  def generate_grape_files
+
+    prefix = 'app/api/v1'
+    FileUtils.mkdir_p prefix
 
     if args[0].nil? || args[0] == "all"
       names = Dir["app/models/*.rb"].map {|x| File.basename(x, '.rb')}
@@ -22,28 +21,44 @@ class Weapon::GrapeGenerator < Rails::Generators::Base # :nodoc:
       names = args
     end
 
-    ap names
-
-    names.each do |name|
-      entity_dir = File.join(prefix, "entities")
-      entity_path = File.join(entity_dir, "#{name}.rb")
-
-      ap "dir is #{entity_dir}"
-      ap "path is #{entity_path}"
+    if options[:only]
+      names = names & options[:only].split(',')
+    end
 
-      next unless name.camelize.constantize.respond_to? 'column_names'
-      column_names = name.camelize.constantize.column_names
-      infos = column_names.map {|column| "expose :#{column}"}
+    if options[:except]
+      names = names - options[:except].split(',')
+    end
 
-      FileUtils.mkdir_p entity_dir
+    module_array = prefix.split('/')[1..-1]
+    names.each do |name|
       a = Magicfile.new
-      module_array = prefix.split('/')[1..-1].append('entities')
       a.append_modules(module_array)
-      a.append_class(name.camelize, 'Grape::Entity')
+      a.append_class(name.camelize.pluralize, 'Grape::API')
+      infos = ["version 'v1', using: :path", "format :json", "default_format: json"]
       a.append_string_lines(infos)
-      a.to_file entity_path
+      a.to_file "#{prefix}/#{name.pluralize}.rb"
     end
 
+    b = Magicfile.new
+    b.append_modules(module_array)
+    b.append_class('Root', 'Grape::API')
+    infos = ["helpers Api::V1::Helpers", "version 'v1', using: :path", "cascade false", "format :json", "default_format :json\n", "use Api::V1::Auth::Middleware\n"]
+    infos += names.map {|x| "mount Api::V1::#{x.camelize.pluralize}"}
+    infos += ["\n", "before do", "  header 'Access-Control-Allow-Origin'", "end\n"]
+    b.append_string_lines(infos)
+    b.to_file "#{prefix}/root.rb"
+
+    copy_file 'dispatch.rb', "app/api/dispatch.rb"
+    copy_file 'errors.rb', "#{prefix}/errors.rb"
+    copy_file 'helpers.rb', "#{prefix}/helpers.rb"
+    copy_file 'named_params.rb', "#{prefix}/named_params.rb"
+    FileUtils.mkdir_p "#{prefix}/auth"
+    copy_file 'auth/utils.rb', "#{prefix}/auth/utils.rb"
+    copy_file 'auth/middleware.rb', "#{prefix}/auth/middleware.rb"
+    copy_file 'auth/authenticator.rb', "#{prefix}/auth/authenticator.rb"
+
+    route "mount Api::Dispatch => '/api'"
+
   end
 
 end

data/lib/generators/weapon/grape/templates/auth/authenticator.rb

@@ -0,0 +1,86 @@
+module Api
+  module V1
+    module Auth
+      class Authenticator
+
+	      def initialize(request, params)
+	        @request = request
+	        @params  = params
+	      end
+
+
+	      def authenticate!
+	        check_user!
+	        check_tonce!
+	        check_signature!
+	        user
+	      end
+
+	      def user
+	        @user ||= User.where(access_key: @params[:access_key]).first
+	      end
+
+	      def check_user!
+	        raise InvalidAccessKeyError, @params[:access_key] unless user
+	        raise ExpiredAccessKeyError, @params[:access_key] if user.auth_expired?
+	      end
+
+	      def check_signature!
+					Rails.logger.warn "payload is #{payload}"
+					Rails.logger.warn "uploaded signature is #{@params[:signature]}"
+					Rails.logger.warn "generated signature is #{Utils.hmac_signature(user.secret_key, payload)}"
+	        if @params[:signature] != Utils.hmac_signature(user.secret_key, payload)
+	          Rails.logger.warn "Api v1 auth failed: signature doesn't match. token: #{user.access_key} payload: #{payload}"
+	          raise Api::V1::IncorrectSignatureError, @params[:signature]
+	        end
+	      end
+
+	      def check_tonce!
+	        key = "api:v1:tonce:#{user.access_key}:#{tonce}"
+	        if Utils.cache.read(key)
+	          Rails.logger.warn "API v1 auth failed: used tonce.access_key: #{user.access_key} payload: #{payload} tonce: #{tonce}"
+	          raise Api::V1::TonceUsedError.new(user.access_key, tonce)
+	        end
+	        Utils.cache.write key, tonce, 61 # forget after 61 seconds
+
+	        now = Time.now.to_i
+	        if tonce < now-30 || tonce > now+30# within 30 seconds
+	          Rails.logger.warn "Api v1 auth failed: invalid tonce. access_key: #{user.access_key} payload: #{payload} tonce: #{tonce} current timestamp: #{now}"
+	          raise Api::V1::InvalidTonceError.new(tonce, now)
+	        end
+	      end
+
+	      def tonce
+	        @tonce ||= @params[:tonce].to_i
+	      end
+
+	      def payload
+	        "#{canonical_verb}|#{Api::V1::Root::PREFIX}#{canonical_uri}|#{canonical_query}"
+	      end
+
+	      def canonical_verb
+	        @request.request_method
+	      end
+
+	      def canonical_uri
+	        @request.path_info
+	      end
+
+	      def canonical_query
+	        hash = @params.select {|k,v| !%w(route_info signature format).include?(k) }
+					ap hash
+	        URI.unescape(hash.to_param)
+	      end
+
+	      def endpoint
+	        @request.env['api.endpoint']
+	      end
+
+	      def route_scopes
+	        endpoint.options[:route_options][:scopes]
+	      end
+
+			end
+  	end
+	end
+end

data/lib/generators/weapon/grape/templates/auth/middleware.rb

@@ -0,0 +1,35 @@
+module Api
+  module V1
+		module Auth
+			class Middleware < ::Grape::Middleware::Base
+
+	    	def before
+					#raise AuthorizationError
+	        if provided?
+	          auth = Api::V1::Auth::Authenticator.new(request, params)
+						ap "do auth"
+						@env['user'] = auth.authenticate!
+	        end
+	      end
+
+	      def provided?
+					ap params
+					ap params["access_key"]
+					ap params["tonce"]
+					ap params["signature"]
+	        params["access_key"] && params["tonce"] && params["signature"]
+	      end
+
+	      def request
+	        @request ||= ::Grape::Request.new(env)
+	      end
+
+	      def params
+	        @params ||= request.params
+	      end
+
+	    end
+    end
+
+  end
+end

data/lib/generators/weapon/grape/templates/auth/utils.rb

@@ -0,0 +1,28 @@
+module Api
+  module V1
+    module Auth
+	    module Utils
+	      class <<self
+
+	        def cache
+	          # Simply use rack-attack cache wrapper
+	          @cache ||= Rack::Attack::Cache.new
+	        end
+
+	        def urlsafe_string_40
+	          # 30 is picked so generated string length is 40
+	          SecureRandom.urlsafe_base64(30).tr('_-', 'xx')
+	        end
+
+	        alias :generate_access_key :urlsafe_string_40
+	        alias :generate_secret_key :urlsafe_string_40
+
+	        def hmac_signature(secret_key, payload)
+	          OpenSSL::HMAC.hexdigest 'SHA256', secret_key, payload
+	        end
+
+	      end
+	    end
+    end
+  end
+end

data/lib/generators/weapon/grape/templates/dispatch.rb

@@ -0,0 +1,13 @@
+
+module Api
+  class Dispatch < Grape::API
+
+    mount Api::V1::Root
+
+    format :json
+    default_format :json
+
+    content_type    :json, "application/json;charset=utf-8"
+
+  end
+end

data/lib/generators/weapon/grape/templates/errors.rb

@@ -0,0 +1,81 @@
+module Api
+  module V1
+
+	  module ExceptionHandlers
+
+	    def self.included(base)
+	      base.instance_eval do
+	        rescue_from Grape::Exceptions::ValidationErrors do |e|
+	          Rack::Response.new({
+	            success: false,
+							message: e.message,
+							data: {}
+	          }.to_json, e.status)
+	        end
+	      end
+	    end
+
+	  end
+
+	  class Error < Grape::Exceptions::Base
+	    attr :code, :text
+
+	    # code: api error code defined by Peatio, errors originated from
+	    # subclasses of Error have code start from 2000.
+	    # text: human readable error message
+	    # status: http status code
+	    def initialize(opts={})
+	      @text    = opts[:text]   || ''
+
+	      @status  = 200
+	      @message = {success: false, message: @text, data: {}}
+	    end
+	  end
+
+	  class AuthorizationError < Error
+	    def initialize
+	      super code: 2001, text: 'Authorization failed', status: 401
+	    end
+	  end
+
+
+
+	  class IncorrectSignatureError < Error
+	    def initialize(signature)
+	      super code: 2005, text: "Signature #{signature} is incorrect.", status: 401
+	    end
+	  end
+
+	  class TonceUsedError < Error
+	    def initialize(access_key, tonce)
+	      super code: 2006, text: "The tonce #{tonce} has already been used by access key #{access_key}.", status: 401
+	    end
+	  end
+
+	  class InvalidTonceError < Error
+	    def initialize(tonce, now)
+	      super code: 2007, text: "The tonce #{tonce} is invalid, current timestamp is #{now}.", status: 401
+	    end
+	  end
+
+	  class InvalidAccessKeyError < Error
+	    def initialize(access_key)
+	      super code: 2008, text: "The access key #{access_key} does not exist.", status: 401
+	    end
+	  end
+
+	  class DisabledAccessKeyError < Error
+	    def initialize(access_key)
+	      super code: 2009, text: "The access key #{access_key} is disabled.", status: 401
+	    end
+	  end
+
+	  class ExpiredAccessKeyError < Error
+	    def initialize(access_key)
+	      super code: 2010, text: "The access key #{access_key} has expired.", status: 401
+	    end
+	  end
+
+
+  end
+end

data/lib/generators/weapon/grape/templates/helpers.rb

@@ -0,0 +1,24 @@
+module Api
+  module V1
+    module Helpers
+      def render_success(data = {}, msg = nil)
+        status 200
+        { success: true, message: msg.to_s, data: data }
+      end
+
+      def render_fail(msg = nil, data = {})
+        status 200
+        { success: false, message: msg.to_s, data: data }
+      end
+
+      def authenticate!
+        current_user or raise AuthorizationError
+      end
+
+      def current_user
+        @user ||= env['user']
+      end
+
+    end
+  end
+end

data/lib/generators/weapon/grape/templates/named_params.rb

@@ -0,0 +1,14 @@
+module Api
+  module V1
+    module NamedParams
+      extend ::Grape::API::Helpers
+
+      params :auth do
+        requires :access_key, type: String,  desc: "Access key."
+        requires :tonce,      type: Integer, desc: "Tonce is an integer represents the milliseconds elapsed since Unix epoch."
+        requires :signature,  type: String,  desc: "The signature of your request payload, generated using your secret key."
+      end
+
+    end
+  end
+end

data/lib/generators/weapon/grape_entity/grape_entity_generator.rb

@@ -0,0 +1,55 @@
+require 'magicfile'
+require 'fileutils'
+
+
+class Weapon::GrapeEntityGenerator < Rails::Generators::Base # :nodoc:
+  desc 'This generator creates grape entity for model'
+
+  class_option :only, type: :string, desc: 'only generate api for listed model'
+  class_option :except, type: :string, desc: 'when generate api, exclude listed model'
+
+  def generate_grape_entity
+
+    prefix_dirs = Dir.glob('app/api/*/')
+    if prefix_dirs.count == 0
+      prefix = 'app/api/v1'
+    elsif prefix_dirs.count == 1
+      prefix = prefix_dirs[0]
+    else
+      prefix = ask('there are multi directory in app/api directory, please input path like app/api/v1')
+    end
+
+    if args[0].nil? || args[0] == "all"
+      names = Dir["app/models/*.rb"].map {|x| File.basename(x, '.rb')}
+    else
+      names = args
+    end
+
+    if options[:only]
+      names = names & options[:only].split(',')
+    end
+
+    if options[:except]
+      names = names - options[:except].split(',')
+    end
+
+    names.each do |name|
+      entity_dir = File.join(prefix, "entities")
+      entity_path = File.join(entity_dir, "#{name}.rb")
+
+      next unless name.camelize.constantize.respond_to? 'column_names'
+      column_names = name.camelize.constantize.column_names
+      infos = column_names.map {|column| "expose :#{column}"}
+
+      FileUtils.mkdir_p entity_dir
+      a = Magicfile.new
+      module_array = prefix.split('/')[1..-1].append('entities')
+      a.append_modules(module_array)
+      a.append_class(name.camelize, 'Grape::Entity')
+      a.append_string_lines(infos)
+      a.to_file entity_path
+    end
+
+  end
+
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: weapon
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - seaify
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-01-26 00:00:00.000000000 Z
+date: 2016-01-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -92,6 +92,14 @@ files:
 - lib/generators/weapon/activeadmin/activeadmin_generator.rb
 - lib/generators/weapon/email/email_generator.rb
 - lib/generators/weapon/grape/grape_generator.rb
+- lib/generators/weapon/grape/templates/auth/authenticator.rb
+- lib/generators/weapon/grape/templates/auth/middleware.rb
+- lib/generators/weapon/grape/templates/auth/utils.rb
+- lib/generators/weapon/grape/templates/dispatch.rb
+- lib/generators/weapon/grape/templates/errors.rb
+- lib/generators/weapon/grape/templates/helpers.rb
+- lib/generators/weapon/grape/templates/named_params.rb
+- lib/generators/weapon/grape_entity/grape_entity_generator.rb
 - lib/generators/weapon/i18n/i18n_generator.rb
 - lib/support/create_gem/basic.bin
 - lib/support/create_gem/basic.gemspec