feat: add self.parse method to Address

Author: classicalliu

lib/ckb/address.rb

@@ -8,22 +8,24 @@ class Address
     PREFIX_MAINNET = "ckb"
     PREFIX_TESTNET = "ckt"
 
-    def initialize(blake160, mode: MODE::TESTNET)
+    DEFAULT_MODE = MODE::TESTNET
+
+    TYPE = "01"
+    BIN_IDX = "P2PH"
+
+    def initialize(blake160, mode: DEFAULT_MODE)
+      @mode = mode
       @blake160 = blake160
-      @prefix = if mode == MODE::TESTNET
-                  PREFIX_TESTNET
-                elsif mode == MODE::MAINNET
-                  PREFIX_MAINNET
-                end
+      @prefix = self.class.prefix(mode: mode)
     end
 
     # Generates address assuming default lock script is used
     # payload = type(01) | bin-idx("P2PH" => "50/32/50/48") | pubkey blake160
     # see https://github.com/nervosnetwork/ckb/wiki/Common-Address-Format for more info.
     def generate
       blake160_bin = [blake160[2..-1]].pack("H*")
-      type = ["01"].pack("H*")
-      bin_idx = ["P2PH".each_char.map { |c| c.ord.to_s(16) }.join].pack("H*")
+      type = [TYPE].pack("H*")
+      bin_idx = [self.class.bin_idx_ord].pack("H*")
       payload = type + bin_idx + blake160_bin
       ConvertAddress.encode(@prefix, payload)
     end
@@ -32,10 +34,15 @@ def generate
 
     # Parse address into lock assuming default lock script is used
     def parse(address)
+      self.class.parse(address, mode: @mode)
+    end
+
+    def self.parse(address, mode: DEFAULT_MODE)
       decoded_prefix, data = ConvertAddress.decode(address)
-      raise "Invalid prefix" if decoded_prefix != @prefix
 
-      raise "Invalid type/bin-idx" if data.slice(0..4) != ["0150325048"].pack("H*")
+      raise "Invalid prefix" if decoded_prefix != prefix(mode: mode)
+
+      raise "Invalid type/bin-idx" if data.slice(0..4) != [TYPE + bin_idx_ord].pack("H*")
 
       CKB::Utils.bin_to_hex(data.slice(5..-1))
     end
@@ -46,8 +53,21 @@ def self.blake160(pubkey)
       Utils.bin_to_hex(hash_bin[0...20])
     end
 
-    def self.from_pubkey(pubkey, mode: MODE::TESTNET)
+    def self.from_pubkey(pubkey, mode: DEFAULT_MODE)
       new(blake160(pubkey), mode: mode)
     end
+
+    def self.bin_idx_ord
+      BIN_IDX.each_char.map { |c| c.ord.to_s(16) }.join
+    end
+
+    def self.prefix(mode: DEFAULT_MODE)
+      case mode
+      when MODE::TESTNET
+        PREFIX_TESTNET
+      when MODE::MAINNET
+        PREFIX_MAINNET
+      end
+    end
   end
 end

spec/ckb/address_spec.rb

@@ -8,7 +8,7 @@
   let(:prefix) { "ckt" }
   let(:address) { "ckt1q9gry5zgxmpjnmtrp4kww5r39frh2sm89tdt2l6v234ygf" }
 
-  describe "from pubkey" do
+  context "from pubkey" do
     let(:addr) { CKB::Address.from_pubkey(pubkey) }
 
     it "pubkey blake160" do
@@ -32,7 +32,7 @@
     end
   end
 
-  describe "from pubkey hash" do
+  context "from pubkey hash" do
     let(:addr) { CKB::Address.new(pubkey_blake160) }
 
     it "generate_address" do
@@ -47,4 +47,24 @@
       ).to eq pubkey_blake160
     end
   end
+
+  context "self.parse" do
+    it "success" do
+      expect(
+        CKB::Address.parse(address)
+      ).to eq pubkey_blake160
+    end
+
+    it "failed if mainnet mode" do
+      expect {
+        CKB::Address.parse(address, mode: CKB::MODE::MAINNET)
+      }.to raise_error(RuntimeError)
+    end
+
+    it "eq to parse" do
+      expect(
+        CKB::Address.parse(address)
+      ).to eq CKB::Address.new(pubkey_blake160).parse(address)
+    end
+  end
 end