pinetwork 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 19ca8427c38f5d1a3cfef85fe1a3e026fb0b0f354b056c106d3d980e5182229e
-  data.tar.gz: a76c9c5b067472f207d10d0d9bd52b92149b0d2192af5b13198bef6bf469e23e
+  metadata.gz: 9cab0d9673b76e3b49f2ccb041f917b9574ca85dedb1c5d4529fd19ada191b2b
+  data.tar.gz: ee2bf09fd869205f4377f7ad72c79897282b419d3551dc72f88facbd8d33ca0c
 SHA512:
-  metadata.gz: 3de9acdecb5e03c5256be0a6a65d4341f3037fb64eb73262577a5dea5c61beae7a42ba7dc02596985683144242da6e7273b1ba6f37cf02019a09a294a2a8fae2
-  data.tar.gz: 9828e6a0cd9fe3a8de440bef58589ac275a51119c94f2adf67874282134db21042f0081a6e1ab23519445be11ea257635bf9a91494fe44448c3fb10b93b2a1ea
+  metadata.gz: 9f3279cb6518e666ef5ee59de68af1f98d147f1e159c2af1d9c775915f0d22624a640b2f36654c48f8f9a729e4ace10db4f9bedff721769eb220f82a67ca420a
+  data.tar.gz: f6d002eae9fd88cfb6e6d86958558c6f01dbb69b365cd7799ebdbe601d9d6242fb30ded18353f51f7d7f1bd43992e68d763f7687d6453f094f9ccbfe384d58e5

data/Rakefile

@@ -0,0 +1,9 @@
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs << "test"
+  t.test_files = FileList["test/*.rb"]
+end
+
+desc "Run tests"
+task default: :test

data/lib/errors.rb

@@ -0,0 +1,45 @@
+class ::PiNetwork
+  module Errors
+    class APIRequestError < StandardError
+      attr_reader :response_body
+      attr_reader :response_status
+
+      def initialize(message, response_status, response_body)
+        super(message)
+        @response_status = response_status
+        @response_body = response_body
+      end
+    end
+
+    class PaymentNotFoundError < StandardError
+      attr_reader :payment_id
+
+      def initialize(message, payment_id)
+        super(message)
+        @payment_id = payment_id
+      end
+    end
+
+    class TxidAlreadyLinkedError < StandardError
+      attr_reader :payment_id
+      attr_reader :txid
+
+      def initialize(message, payment_id, txid)
+        super(message)
+        @payment_id = payment_id
+        @txid = txid
+      end
+    end
+
+    class TxSubmissionError < StandardError
+      attr_reader :tx_error_code
+      attr_reader :op_error_codes
+
+      def initialize(tx_error_code, op_error_codes)
+        super(message)
+        @tx_error_code = tx_error_code
+        @op_error_codes = op_error_codes
+      end
+    end
+  end
+end

data/lib/{pi_network.rb → pinetwork.rb}

@@ -1,5 +1,4 @@
-gem_dir = Gem::Specification.find_by_name("pi_network").gem_dir
-require "#{gem_dir}/lib/errors"
+require_relative 'errors'
 
 class PiNetwork
   require 'faraday'
@@ -12,13 +11,21 @@ class PiNetwork
   attr_reader :base_url
   attr_reader :from_address
 
-  def initialize(api_key, wallet_private_key, options = {})
+  BASE_URL = "https://api.minepi.com".freeze
+  MAINNET_HOST = "api.mainnet.minepi.com".freeze
+  TESTNET_HOST = "api.testnet.minepi.com".freeze
+
+  def initialize(api_key:, wallet_private_key:, faraday: Faraday.new, options: {})
     validate_private_seed_format!(wallet_private_key)
     @api_key = api_key
     @account = load_account(wallet_private_key)
-    @base_url = options[:base_url] || "https://api.minepi.com"
+    @base_url = options[:base_url] || BASE_URL
+    @mainnet_host = options[:mainnet_host] || MAINNET_HOST
+    @testnet_host = options[:testnet_host] || TESTNET_HOST
+    @faraday = faraday
 
     @open_payments = {}
+    @open_payments_mutex = Mutex.new
   end
 
   def get_payment(payment_id)
@@ -42,7 +49,7 @@ class PiNetwork
       payment: payment_data,
     }
 
-    response = Faraday.post(
+    response = @faraday.post(
       base_url + "/v2/payments",
       request_body.to_json,
       http_headers,
@@ -51,58 +58,68 @@ class PiNetwork
     parsed_response = handle_http_response(response, "An unknown error occurred while creating a payment")
     
     identifier = parsed_response["identifier"]
-    @open_payments[identifier] = parsed_response
+    @open_payments_mutex.synchronize do
+      @open_payments[identifier] = parsed_response
+    end
 
     return identifier
   end
 
   def submit_payment(payment_id)
-    payment = @open_payments[payment_id]
+    @open_payments_mutex.synchronize do
+      payment = @open_payments[payment_id]
 
-    if payment.nil? || payment["identifier"] != payment_id
-      payment = get_payment(payment_id)
-      txid = payment["transaction"]["txid"]
-      raise Errors::TxidAlreadyLinkedError.new("This payment already has a linked txid", payment_id, txid) if txid.present?
-    end
+      if payment.nil? || payment["identifier"] != payment_id
+        payment = get_payment(payment_id)
+        txid = payment["transaction"]&.dig("txid")
+        raise Errors::TxidAlreadyLinkedError.new("This payment already has a linked txid", payment_id, txid) if txid.present?
+      end
 
-    set_horizon_client(payment["network"])
-    @from_address = payment["from_address"]
+      set_horizon_client(payment["network"])
+      @from_address = payment["from_address"]
 
-    transaction_data = {
-      amount: payment["amount"],
-      identifier: payment["identifier"],
-      recipient: payment["to_address"]
-    }
+      transaction_data = {
+        amount: payment["amount"],
+        identifier: payment["identifier"],
+        recipient: payment["to_address"]
+      }
 
-    transaction = build_a2u_transaction(transaction_data)
-    txid = submit_transaction(transaction)
+      transaction = build_a2u_transaction(transaction_data)
+      txid = submit_transaction(transaction)
 
-    @open_payments.delete(payment_id)
+      @open_payments.delete(payment_id)
 
-    return txid
+      return txid
+    end
   end
 
-  def complete_payment(identifier, txid)
+  def complete_payment(payment_id, txid)
     body = {"txid": txid}
 
     response = Faraday.post(
-      base_url + "/v2/payments/#{identifier}/complete",
+      base_url + "/v2/payments/#{payment_id}/complete",
       body.to_json,
       http_headers
     )
 
-    @open_payments.delete(identifier)
+    @open_payments_mutex.synchronize do
+      @open_payments.delete(payment_id)
+    end
+
     handle_http_response(response, "An unknown error occurred while completing the payment")
   end
 
-  def cancel_payment(identifier)
+  def cancel_payment(payment_id)
     response = Faraday.post(
-      base_url + "/v2/payments/#{identifier}/cancel",
+      base_url + "/v2/payments/#{payment_id}/cancel",
       {}.to_json,
       http_headers,
     )
 
-    @open_payments.delete(identifier)
+    @open_payments_mutex.synchronize do
+      @open_payments.delete(payment_id)
+    end
+
     handle_http_response(response, "An unknown error occurred while cancelling the payment")
   end
 
@@ -130,7 +147,7 @@ class PiNetwork
 
   def handle_http_response(response, unknown_error_message = "An unknown error occurred while making an API request")
     unless response.status == 200
-      error_message = JSON.parse(response.body).dig("error_message") rescue unknown_err_message
+      error_message = extract_error_message(response.body, unknown_error_message)
       raise Errors::APIRequestError.new(error_message, response.status, response.body)
     end
 
@@ -144,8 +161,9 @@ class PiNetwork
   end
 
   def set_horizon_client(network)
-    host = network == "Pi Network" ? "api.mainnet.minepi.com" : "api.testnet.minepi.com"
-    horizon = network == "Pi Network" ? "https://api.mainnet.minepi.com" : "https://api.testnet.minepi.com"
+    host = (network.starts_with? "Pi Network") ? @mainnet_host : @testnet_host
+    horizon = "https://#{host}"
+
     client = Stellar::Client.new(host: host, horizon: horizon)
     Stellar::default_network = network
 
@@ -186,8 +204,13 @@ class PiNetwork
 
   def submit_transaction(transaction)
     envelope = transaction.to_envelope(self.account.keypair)
-    response = self.client.submit_transaction(tx_envelope: envelope)
-    txid = response._response.body["id"]
+    begin
+      response = self.client.submit_transaction(tx_envelope: envelope)
+      txid = response._response.body["id"]
+    rescue => error
+      result_codes = error.response&.dig(:body, "extras", "result_codes")
+      raise Errors::TxSubmissionError.new(result_codes&.dig("transaction"), result_codes&.dig("operations"))
+    end
   end
 
   def validate_payment_data!(data, options = {})
@@ -200,7 +223,11 @@ class PiNetwork
   end
 
   def validate_private_seed_format!(seed)
-    raise StandardError.new("Private Seed should start with \"S\"") unless seed.upcase.starts_with?("S")
+    raise StandardError.new("Private Seed should start with \"S\"") unless seed.upcase.start_with?("S")
     raise StandardError.new("Private Seed should be 56 characters") unless seed.length == 56
   end
+
+  def extract_error_message(response_body, default_message)
+    JSON.parse(response_body).dig("error_message") rescue default_message
+  end
 end

data/test/a2u_concurrency_test.rb

@@ -0,0 +1,35 @@
+require 'minitest/autorun'
+require_relative '../lib/pinetwork'
+
+class A2UConcurrencyTest < Minitest::Test
+  def test_concurrent_create_payment
+    total_threads = 10000
+    api_key = "api-key"
+    wallet_private_key = "SC2L62EYF7LYF43L4OOSKUKDESRAFJZW3UW6RFZ57UY25VAMHTL2BFER"
+
+    threads = []
+    
+    faraday_stub = Minitest::Mock.new
+    pi = PiNetwork.new(api_key: api_key, wallet_private_key: wallet_private_key, faraday: faraday_stub)
+
+    total_threads.times do
+      threads << Thread.new do
+        faraday_response = Faraday::Response.new(
+          status: 200,
+          body: {identifier: SecureRandom.alphanumeric(12)}.to_json,
+          response_headers: {}
+        )
+        faraday_stub.expect(:post, faraday_response) do |url|
+          url == "https://api.minepi.com/v2/payments"
+        end
+        
+        payment_data = { amount: 1, memo: "test", metadata: {"info": "test"}, uid: "test-uid" }
+        payment_id = pi.create_payment(payment_data)
+      end
+    end
+    threads.each(&:join)
+
+    open_payments_after = pi.instance_variable_get(:@open_payments)
+    assert_equal(total_threads, open_payments_after.values.uniq.count, "open_payments got corrupted!")
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pinetwork
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Pi Core Team
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-01-25 00:00:00.000000000 Z
+date: 2024-02-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: stellar-sdk
@@ -44,13 +44,16 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/pi_network.rb
+- Rakefile
+- lib/errors.rb
+- lib/pinetwork.rb
+- test/a2u_concurrency_test.rb
 homepage: https://github.com/pi-apps/pi-ruby
 licenses:
 - PiOS
 metadata:
   documentation_uri: https://github.com/pi-apps/pi-ruby
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -65,8 +68,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3.1
-signing_key: 
+rubygems_version: 3.4.10
+signing_key:
 specification_version: 4
 summary: Pi Network Ruby
 test_files: []