protip 0.20.1 → 0.20.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e9a530ecc12160b36bb16eace8ff4d684ef30eae
-  data.tar.gz: 165e373d6d3931574b1b3891379204af6cd6f7d7
+  metadata.gz: 2250fc4ff6bdcd6d69c30c369da5a6934d8458b9
+  data.tar.gz: 96844a4c73302bcebbea6c429d4cdab57f855a70
 SHA512:
-  metadata.gz: f0a869ac73a101dde2c45735361166f90eb3466ec5b8b81f4befbb17a1607b0c4ff077f10ee270988ddffae022f34eb904ec2ba52efe74f62b5e263a6f931c3b
-  data.tar.gz: 2da3112ab256949d3cad42f7ed804c15b0baa2d8aa12f91a50c06773fc449ff51c3a43d2fa2dcdfc1bcd3c44b5342d4b47890ffeeff17315fb151a53854c1f4b
+  metadata.gz: aa3294f1b6d92f0aa0fccb7fc238f5e07ec63edbf5e4aa4c3fbab2dc318867bc39bdd863c21e598959bbdecce3e8822c703f829e676bdff6ad8d5c5810c47f59
+  data.tar.gz: 8f81ffd115d480d25e831e8b982ca4097a9f44f0273934bcabee3b60fc829e2394ee60eb8d075c5cbe98a390bd58239ae60c45503e119b67ade0ce81e47f3710

data/lib/protip/tasks/compile.rake

@@ -0,0 +1,60 @@
+require 'shellwords'
+namespace :protip do
+  desc 'compile a single .proto file to Ruby'
+  task :compile, [:filename, :proto_path, :ruby_path] do |t, args|
+    proto_path = [args[:proto_path] || 'definitions'].flatten.compact.reject{|path| path == ''}
+    proto_path << File.join(Gem.loaded_specs['protip'].full_gem_path, 'definitions')
+
+    ruby_path = args[:ruby_path] || 'lib'
+
+    filename = args[:filename] || raise(ArgumentError.new 'filename argument is required')
+
+    command = "protoc #{proto_path.map{|p| "--proto_path=#{Shellwords.escape p}"}.join ' '} --ruby_out=#{Shellwords.escape ruby_path} #{Shellwords.escape filename}"
+    puts command # is there a better way to log this?
+    system command
+
+    ## hack around missing options in Ruby, remove when https://github.com/google/protobuf/issues/1198 is resolved
+    package_match = File.read(filename).match(/package "?([a-zA-Z0-9\.]+)"?;/)
+    package = (package_match ? package_match[1] : nil)
+    ruby_file = filename.gsub(/^#{proto_path.first}\/?/, "#{ruby_path}/").gsub(/proto$/, 'rb') # Relies on a relative filename and proto path, which protoc requires anyway at this point
+    raise "cannot find generated Ruby file (#{ruby_file})" unless File.exists?(ruby_file)
+
+    # Push/pop message names as we move through the protobuf file
+    message_name_stack = []
+    first_match = true
+    File.open filename, 'r' do |f|
+      f.each_line do |line|
+        if line.include? '{'
+          match = line.match(/message\s([a-zA-Z]+)/)
+          message_name_stack << (match ? match[1] : nil)
+        end
+
+        # figure out the field name and enum name if this line is like
+        # +protip.messages.EnumValue value = 3 [(protip_enum) = "Foo"]+
+        match = line.match /\s*[a-zA-Z0-9\.]+\s+([a-zA-Z0-9]+).+\[\s*\(\s*protip_enum\s*\)\s*=\s*"([a-zA-Z0-9\.]+)"\s*\]/
+        if match
+          message_name = "#{package ? "#{package}." : ''}#{message_name_stack.compact.join('.')}"
+          field_name = match[1]
+          enum_name = match[2]
+          File.open ruby_file, 'a' do |f|
+            if first_match
+              f.puts <<-RBY
+
+# -- Protip hack until https://github.com/google/protobuf/issues/1198 is resolved
+RBY
+            end
+            first_match = false
+            f.puts <<-RBY
+Google::Protobuf::DescriptorPool.generated_pool.lookup("#{message_name}").lookup("#{field_name}").instance_variable_set(:"@_protip_enum_value", "#{enum_name}")
+RBY
+          end
+        end
+
+        if line.include? '}'
+          message_name_stack.pop
+        end
+      end
+    end
+
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: protip
 version: !ruby/object:Gem::Version
-  version: 0.20.1
+  version: 0.20.2
 platform: ruby
 authors:
 - AngelList
@@ -219,6 +219,7 @@ files:
 - lib/protip/resource/search_methods.rb
 - lib/protip/resource/updateable.rb
 - lib/protip/standard_converter.rb
+- lib/protip/tasks/compile.rake
 - lib/protip/wrapper.rb
 - test/functional/protip/resource_test.rb
 - test/test_helper.rb