bitcoinrb-grpc 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 424e2253993bc766b1f087578da27e42998111280126bc699605150708740a7c
-  data.tar.gz: c3248a38a651e57f57b1ae8043b6f112fd7af75fe5ad6a0ce9d03ae837d9673a
+  metadata.gz: 3fcac84338b9a2eb74ee9d72b276a1ff6ea8cebd2c35052d3a857a650b53e877
+  data.tar.gz: 9d614d0891e3544fefa96c93f213eebabced6496d41914b451d466a9268d3f50
 SHA512:
-  metadata.gz: 2415425f2098278894c025a5451b3d342755ef30cd918ae612f934ad148cc9d622be5d630c532039ab1d9c65414ce44f3ed71159c9120114185f5caf7ce0ebff
-  data.tar.gz: b1ca87670b8bf33b589b895a9cd563cc11dae8d65f18dbbec1900ec50c95f88b25f6146113929667dc47f88ba20485c510559598e80a1b15a97c0a92af30d8d0
+  metadata.gz: bf53e08ca4e325a6c5160bd10be916f687938e0d6b6b836fd3af92cd5891ebf40ea528718fafc738aa8f467fe2f4eb4ffe38ddfe85f38bcb680af27a2c27b745
+  data.tar.gz: 53f829615c9edd13f41b7efca47ac4c6b16407a408eb5b6b841204fd11c09c00ca049b2ef19659517051af7fd5ce0ba66d02121cd6042a4f997f4d8c80314a46

data/lib/bitcoin/grpc/api.rb

@@ -5,6 +5,8 @@ module Bitcoin
       autoload :ListUnspent, 'bitcoin/grpc/api/list_unspent'
       autoload :ListColoredUnspent, 'bitcoin/grpc/api/list_colored_unspent'
       autoload :ListUncoloredUnspent, 'bitcoin/grpc/api/list_uncolored_unspent'
+      autoload :GetBalance, 'bitcoin/grpc/api/get_balance'
+      autoload :GetTokenBalance, 'bitcoin/grpc/api/get_token_balance'
     end
   end
 end

data/lib/bitcoin/grpc/api/get_balance.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Bitcoin
+  module Grpc
+    module Api
+      class GetBalance
+        attr_reader :spv
+
+        def initialize(spv)
+          @spv = spv
+        end
+
+        def execute(request)
+          height = spv.chain.latest_block.height
+          balance = spv.wallet.get_balance(request.account_name)
+          Bitcoin::Grpc::GetBalanceResponse.new(balance: balance)
+        end
+      end
+    end
+  end
+end

data/lib/bitcoin/grpc/api/get_token_balance.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Bitcoin
+  module Grpc
+    module Api
+      class GetTokenBalance
+        attr_reader :spv
+
+        def initialize(spv)
+          @spv = spv
+        end
+
+        def execute(request)
+          height = spv.chain.latest_block.height
+          assets = spv.wallet.list_unspent_assets_in_account(
+            request.asset_type,
+            request.asset_id,
+            account_name: request.account_name,
+            current_block_height: height,
+            min: 0,
+            max: 9999999
+          )
+          balance = assets.sum(&:value)
+          token_balance = assets.sum(&:asset_quantity)
+
+          Bitcoin::Grpc::GetTokenBalanceResponse.new(balance: balance, token_balance: token_balance)
+        end
+      end
+    end
+  end
+end

data/lib/bitcoin/grpc/grpc_pb.rb

@@ -143,6 +143,21 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     add_message "bitcoin.grpc.ListUncoloredUnspentResponse" do
       repeated :utxos, :message, 1, "bitcoin.grpc.Utxo"
     end
+    add_message "bitcoin.grpc.GetBalanceRequest" do
+      optional :account_name, :string, 1
+    end
+    add_message "bitcoin.grpc.GetBalanceResponse" do
+      optional :balance, :uint64, 1
+    end
+    add_message "bitcoin.grpc.GetTokenBalanceRequest" do
+      optional :account_name, :string, 1
+      optional :asset_type, :uint32, 4
+      optional :asset_id, :string, 5
+    end
+    add_message "bitcoin.grpc.GetTokenBalanceResponse" do
+      optional :token_balance, :uint64, 1
+      optional :balance, :uint64, 2
+    end
     add_message "bitcoin.grpc.EventTxConfirmed" do
       optional :request_id, :uint32, 1
       optional :tx_hash, :string, 2
@@ -233,6 +248,10 @@ module Bitcoin
     ListColoredUnspentResponse = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.ListColoredUnspentResponse").msgclass
     ListUncoloredUnspentRequest = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.ListUncoloredUnspentRequest").msgclass
     ListUncoloredUnspentResponse = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.ListUncoloredUnspentResponse").msgclass
+    GetBalanceRequest = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.GetBalanceRequest").msgclass
+    GetBalanceResponse = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.GetBalanceResponse").msgclass
+    GetTokenBalanceRequest = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.GetTokenBalanceRequest").msgclass
+    GetTokenBalanceResponse = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.GetTokenBalanceResponse").msgclass
     EventTxConfirmed = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.EventTxConfirmed").msgclass
     EventUtxoRegistered = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.EventUtxoRegistered").msgclass
     EventUtxoSpent = Google::Protobuf::DescriptorPool.generated_pool.lookup("bitcoin.grpc.EventUtxoSpent").msgclass

data/lib/bitcoin/grpc/grpc_services_pb.rb

@@ -24,6 +24,8 @@ module Bitcoin
         rpc :ListUnspent, ListUnspentRequest, ListUnspentResponse
         rpc :ListColoredUnspent, ListColoredUnspentRequest, ListColoredUnspentResponse
         rpc :ListUncoloredUnspent, ListUncoloredUnspentRequest, ListUncoloredUnspentResponse
+        rpc :GetBalance, GetBalanceRequest, GetBalanceResponse
+        rpc :GetTokenBalance, GetTokenBalanceRequest, GetTokenBalanceResponse
       end
 
       Stub = Service.rpc_stub_class

data/lib/bitcoin/grpc/server.rb

@@ -120,6 +120,14 @@ module Bitcoin
         logger.error("list_uncolored_unspent: #{e.message}")
         logger.error("list_uncolored_unspent: #{e.backtrace}")
       end
+
+      def get_balance(request, _call)
+        logger.info("get_balance: #{request}")
+        Bitcoin::Grpc::Api::GetBalance.new(spv).execute(request)
+      rescue => e
+        logger.error("get_balance: #{e.message}")
+        logger.error("get_balance: #{e.backtrace}")
+      end
     end
 
     class EventsReceiver < Concurrent::Actor::Context

data/lib/bitcoin/grpc/version.rb

@@ -1,5 +1,5 @@
 module Bitcoin
   module Grpc
-    VERSION = "0.1.8"
+    VERSION = "0.1.9"
   end
 end

data/proto/bitcoin/grpc/grpc.proto

@@ -13,6 +13,8 @@ service Blockchain {
   rpc ListUnspent(ListUnspentRequest) returns (ListUnspentResponse);
   rpc ListColoredUnspent(ListColoredUnspentRequest) returns (ListColoredUnspentResponse);
   rpc ListUncoloredUnspent(ListUncoloredUnspentRequest) returns (ListUncoloredUnspentResponse);
+  rpc GetBalance(GetBalanceRequest) returns (GetBalanceResponse);
+  rpc GetTokenBalance(GetTokenBalanceRequest) returns (GetTokenBalanceResponse);
 }
 
 enum Operation {
@@ -186,6 +188,25 @@ message ListUncoloredUnspentResponse {
   repeated Utxo utxos = 1;
 }
 
+message GetBalanceRequest {
+  string account_name = 1;
+}
+
+message GetBalanceResponse {
+  uint64 balance = 1;
+}
+
+message GetTokenBalanceRequest {
+  string account_name = 1;
+  uint32 asset_type = 4;
+  string asset_id = 5;
+}
+
+message GetTokenBalanceResponse {
+  uint64 token_balance = 1;
+  uint64 balance = 2;
+}
+
 message EventTxConfirmed {
   uint32 request_id = 1;
   string tx_hash = 2;
@@ -247,4 +268,4 @@ message AssetOutput {
   uint32 block_height = 6;
   uint64 value = 7;
   string script_pubkey = 8;
-}
+}

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bitcoinrb-grpc
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - Hajime Yamaguchi
@@ -172,6 +172,8 @@ files:
 - config.yml
 - lib/bitcoin/grpc.rb
 - lib/bitcoin/grpc/api.rb
+- lib/bitcoin/grpc/api/get_balance.rb
+- lib/bitcoin/grpc/api/get_token_balance.rb
 - lib/bitcoin/grpc/api/list_colored_unspent.rb
 - lib/bitcoin/grpc/api/list_uncolored_unspent.rb
 - lib/bitcoin/grpc/api/list_unspent.rb