stark 0.5.0

data/.autotest

@@ -0,0 +1,23 @@
+# -*- ruby -*-
+
+require 'autotest/restart'
+
+# Autotest.add_hook :initialize do |at|
+#   at.extra_files << "../some/external/dependency.rb"
+#
+#   at.libs << ":../some/external"
+#
+#   at.add_exception 'vendor'
+#
+#   at.add_mapping(/dependency.rb/) do |f, _|
+#     at.files_matching(/test_.*rb$/)
+#   end
+#
+#   %w(TestA TestB).each do |klass|
+#     at.extra_class_map[klass] = "test/test_misc.rb"
+#   end
+# end
+
+# Autotest.add_hook :run_command do |at|
+#   system "rake build"
+# end

data/History.txt

@@ -0,0 +1,6 @@
+=== 1.0.0 / 2012-08-20
+
+* 1 major enhancement
+
+  * Birthday!
+

data/Manifest.txt

@@ -0,0 +1,31 @@
+.autotest
+History.txt
+Manifest.txt
+README.txt
+Rakefile
+bin/stark
+lib/stark.rb
+lib/stark/ast.rb
+lib/stark/client.rb
+lib/stark/converters.rb
+lib/stark/field.rb
+lib/stark/log_transport.rb
+lib/stark/parser.rb
+lib/stark/processor.rb
+lib/stark/raw_parser.rb
+lib/stark/ruby.rb
+lib/stark/struct.rb
+lib/stark/thrift.kpeg
+stark.gemspec
+test/ThriftSpec.thrift
+test/gen-rb/profile_constants.rb
+test/gen-rb/profile_types.rb
+test/gen-rb/user_storage.rb
+test/leg.rb
+test/legacy_profile/profile_constants.rb
+test/legacy_profile/profile_types.rb
+test/legacy_profile/user_storage.rb
+test/profile.thrift
+test/test_client.rb
+test/test_parser.rb
+test/test_server.rb

data/README.txt

@@ -0,0 +1,63 @@
+= stark
+
+* http://github.com/evanphx/stark
+
+== DESCRIPTION:
+
+Optimized thrift bindings for ruby
+
+== FEATURES/PROBLEMS:
+
+* Generates much more straightforward code for thrift clients and servers
+  than the default thrift bindings for ruby
+
+== SYNOPSIS:
+
+  $ stark service.thrift > service.rb
+  require 'service'
+
+  Use Service::Client and Service::Processor like the default thrift
+  docs describe them.
+
+== REQUIREMENTS:
+
+* thrift gem
+* .thift files
+
+== INSTALL:
+
+* gem install stark
+
+== DEVELOPERS:
+
+After checking out the source, run:
+
+  $ rake newb
+
+This task will install any missing dependencies, run the tests/specs,
+and generate the RDoc.
+
+== LICENSE:
+
+(The MIT License)
+
+Copyright (c) 2013 Evan Phoenix
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+'Software'), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Rakefile

@@ -0,0 +1,21 @@
+# -*- ruby -*-
+
+require 'rubygems'
+require 'hoe'
+
+Hoe.plugin :gemspec
+Hoe.plugin :git
+
+Hoe.spec 'stark' do
+  developer('Evan Phoenix', 'evan@phx.io')
+
+  dependency "thrift", "~> 0.9.0"
+end
+
+task :parser do
+  sh "kpeg -o lib/stark/raw_parser.rb -s -f lib/stark/thrift.kpeg"
+end
+
+task :test => :parser
+
+# vim: syntax=ruby

data/bin/stark

@@ -0,0 +1,17 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'stark/parser'
+require 'stark/ruby'
+
+data = File.read ARGV.shift
+
+tg = Stark::Parser.new data
+
+unless tg.parse
+  tg.raise_error
+end
+
+ruby = Stark::Ruby.new
+
+tg.result.each { |i| i.accept ruby }

data/lib/stark/ast.rb

@@ -0,0 +1,47 @@
+require 'stark/parser'
+
+class Stark::Parser
+  module AST
+    class Namespace
+      def accept(obj)
+        obj.process_namespace self
+      end
+    end
+
+    class Include
+      def accept(obj)
+        obj.process_include self
+      end
+    end
+
+    class Struct
+      def accept(obj)
+        obj.process_struct self
+      end
+    end
+
+    class Field
+      def accept(obj)
+        obj.process_field self
+      end
+    end
+
+    class Function
+      def accept(obj)
+        obj.process_function self
+      end
+    end
+
+    class Service
+      def accept(obj)
+        obj.process_service self
+      end
+    end
+
+    class Enum
+      def accept(obj)
+        obj.process_enum self
+      end
+    end
+  end
+end

data/lib/stark/client.rb

@@ -0,0 +1,63 @@
+require 'stark'
+
+module Stark
+  class Client
+    def initialize(iprot, oprot)
+      @iprot = iprot
+      @oprot = oprot
+    end
+
+    def handle_exception(mtype)
+      if mtype == Thrift::MessageTypes::EXCEPTION
+        x = Thrift::ApplicationException.new
+        x.read(@iprot)
+        @iprot.read_message_end
+        raise x
+      end
+    end
+
+    def handle_unexpected(rtype)
+      return if rtype == ::Thrift::Types::STOP
+      @iprot.skip(rtype)
+    end
+
+    def handle_bad_list(rtype, size)
+      size.times { @iprot.skip(rtype) }
+    end
+
+    def handle_bad_map(key, value, size)
+      size.times do
+        @iprot.skip(key)
+        @iprot.skip(value)
+      end
+    end
+
+    def hash_cast(obj)
+      return obj if obj.kind_of? Hash
+      return obj.to_h if obj.respond_to? :to_h
+
+      raise TypeError, "Unable to convert #{obj.class} to Hash"
+    end
+
+    def read_generic(type, id, cls)
+      ip = @iprot
+
+      obj = cls.new
+
+      ip.read_struct_begin
+
+      while true
+        _, ftype, fid = ip.read_field_begin
+        break if ftype == ::Thrift::Types::STOP
+
+        obj.set_from_index ftype, fid, ip
+
+        ip.read_field_end
+      end
+
+      ip.read_struct_end
+
+      obj
+    end
+  end
+end

data/lib/stark/converters.rb

@@ -0,0 +1,29 @@
+require 'thrift'
+
+module Stark
+  module Converters
+    module I32
+      module_function
+
+      def type
+        Thrift::Types::I32
+      end
+
+      def read(ip)
+        ip.read_i32
+      end
+    end
+
+    module STRING
+      module_function
+
+      def type
+        Thrift::Types::STRING
+      end
+
+      def read(ip)
+        ip.read_string
+      end
+    end
+  end
+end

data/lib/stark/field.rb

@@ -0,0 +1,21 @@
+require 'stark/converters'
+
+module Stark
+  class Field
+    def initialize(idx, name, converter)
+      @index = idx
+      @name = name
+      @converter = converter
+    end
+
+    attr_reader :index, :name, :converter
+
+    def type
+      @converter.type
+    end
+
+    def read(ip)
+      @converter.read ip
+    end
+  end
+end

data/lib/stark/log_transport.rb

@@ -0,0 +1,20 @@
+require 'thrift'
+
+module Stark
+  class LogTransport < Thrift::BaseTransport
+    def initialize(inner, prefix="log")
+      @prefix = prefix
+      @inner = inner
+    end
+
+    def log(name)
+      puts "#{@prefix}: #{name}"
+    end
+
+    def open?; log :open?; @inner.open? end
+    def read(sz); log :read; @inner.read(sz) end
+    def write(buf); log :write; @inner.write(buf) end
+    def close; log :close; @inner.close end
+    def to_io; @inner.to_io end
+  end
+end

data/lib/stark/parser.rb

@@ -0,0 +1,4 @@
+module Stark
+end
+
+require 'stark/raw_parser'

data/lib/stark/processor.rb

@@ -0,0 +1,50 @@
+module Stark
+  class Processor
+    def initialize(handler)
+      @handler = handler
+    end
+
+    def process(iprot, oprot)
+      name, type, seqid  = iprot.read_message_begin
+      if respond_to?("process_#{name}")
+        send("process_#{name}", seqid, iprot, oprot)
+        true
+      else
+        iprot.skip(::Thrift::Types::STRUCT)
+        iprot.read_message_end
+        x = ::Thrift::ApplicationException.new(Thrift::ApplicationException::UNKNOWN_METHOD, 'Unknown function '+name)
+        oprot.write_message_begin(name, ::Thrift::MessageTypes::EXCEPTION, seqid)
+        x.write(oprot)
+        oprot.write_message_end
+        oprot.trans.flush
+        false
+      end
+    end
+
+    def hash_cast(obj)
+      return obj if obj.kind_of? Hash
+      return obj.to_h if obj.respond_to? :to_h
+
+      raise TypeError, "Unable to convert #{obj.class} to Hash"
+    end
+
+    def read_struct(ip, type, id, cls)
+      obj = cls.new
+
+      ip.read_struct_begin
+
+      while true
+        _, ftype, fid = ip.read_field_begin
+        break if ftype == ::Thrift::Types::STOP
+
+        obj.set_from_index ftype, fid, ip
+
+        ip.read_field_end
+      end
+
+      ip.read_struct_end
+
+      obj
+    end
+  end
+end