substrate_common.rb 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f72ffe22e9039554feee750f0d5eef00f6cc95f63df3e20780174f1926b46530
-  data.tar.gz: 67ba8ed682462fc23ad1439982dada786d63527310c96a567ec2107c6eec8969
+  metadata.gz: c601856518e431bc9a8a4623a659406c6d7a9d1826fcbb8b6cca5e0a90a06bfa
+  data.tar.gz: 71d880364a9777dffe1dc3f70b0dedfc2ea44c2bb12e4200d66d55ed5350f38e
 SHA512:
-  metadata.gz: 906b043f2735ab30950fbbf1204d28d31b769d550660b1fffafd6eaa42e90460b308c409dc332642dd7aafa01810d225a6ff33b9e057a9a6fefe3f8c0be4b1a3
-  data.tar.gz: 4167d6c20c335b432b9ef6a77e0348bb452ea085c06f96c5badaabf31f7a6ac31c20b9700eb30c308d7453ff4e219dedf8a283debdc892c413ccc8084e5112ad
+  metadata.gz: b3876a5c03ff8215684cf984c73ace0674045d9964c4d55f1628a66ac2c482f93f0b11ab235dabf71a6c611d15876c2d72a07e297da4bbafdf0eee31239810bd
+  data.tar.gz: ecefaeb3df900db65c6e48354814cd6a2792b9a144c695e39757e18557e711ea62f75956695d0694f60e1221a300850367b884d0f87b72f96f0e19ff11ea66f6

data/.ruby-version

@@ -0,0 +1 @@
+2.6.4

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    substrate_common.rb (0.1.0)
+    substrate_common.rb (0.1.1)
       base58
       blake2b
       xxhash
@@ -13,19 +13,20 @@ GEM
     blake2b (0.10.0)
     coderay (1.1.2)
     method_source (0.9.2)
+    minitest (5.12.2)
     pry (0.12.2)
       coderay (~> 1.1.0)
       method_source (~> 0.9.0)
-    rake (10.5.0)
+    rake (12.3.1)
     xxhash (0.4.0)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
-  bundler (~> 1.17)
+  minitest
   pry
-  rake (~> 10.0)
+  rake (~> 12.0)
   substrate_common.rb!
 
 BUNDLED WITH

data/Rakefile

@@ -1,2 +1,13 @@
 require "bundler/gem_tasks"
-task :default => :spec
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test"
+  t.test_files = Dir.glob("#{__dir__}/test/*_test.rb")
+  t.warning = false
+  t.verbose = true
+end
+
+task :default => :test
+
+

data/lib/substrate_common.rb

@@ -3,28 +3,31 @@ require 'xxhash'
 require 'blake2b'
 require 'base58'
 
+
+require 'substrate_common/address'
+
 class Array
-  def to_hex_string
+  def bytes_to_hex
     raise "Not a byte array" unless self.is_byte_array?
     '0x' + self.map { |b| b.to_s(16).rjust(2, '0') }.join
   end
 
-  def to_bin_string
+  def bytes_to_bin
     raise "Not a byte array" unless self.is_byte_array?
     '0b' + self.map { |b| b.to_s(2).rjust(8, '0') }.join
   end
 
-  def to_hex_array
+  def bytes_to_hex
     raise "Not a byte array" unless self.is_byte_array?
     self.map { |b| b.to_s(16).rjust(2, '0') }
   end
 
-  def to_bin_array
+  def bytes_to_bin
     raise "Not a byte array" unless self.is_byte_array?
     self.map { |b| b.to_s(2).rjust(8, '0') }
   end
 
-  def to_utf8
+  def bytes_to_utf8
     raise "Not a byte array" unless self.is_byte_array?
     self.pack('C*').force_encoding('utf-8')
   end
@@ -39,7 +42,7 @@ class String
     Object.const_get(self)
   end
 
-  def to_byte_array
+  def hex_to_bytes
     data = self.start_with?('0x') ? self[2..] : self
     raise "Not valid hex string" if data.length % 2 != 0
     data.scan(/../).map(&:hex)
@@ -54,20 +57,3 @@ def xxhash128(data)
   end
   bytes.to_hex_string
 end
-
-# https://github.com/paritytech/substrate/wiki/External-Address-Format-(SS58)
-# base58encode ( concat ( <address-type>, <address>, <checksum> ) )
-def ss58_encode(address, address_type=42)
-  address_bytes = address.to_byte_array
-  checksum_length = address_bytes.length == 32 ? 2 : 1
-  input = [address_type] + address_bytes
-  checksum = sshash(input)
-  content = input + checksum[0...checksum_length]
-  Base58.encode(content.to_hex_string.to_i(16), :bitcoin)
-end
-
-def sshash(input)
-  checksum_prefix = 'SS58PRE'.unpack('C*')
-  data = checksum_prefix + input
-  Blake2b.bytes(input.to_utf8, Blake2b::Key.none, 64)
-end

data/lib/substrate_common/address.rb

@@ -0,0 +1,84 @@
+require 'blake2b'
+require 'base58'
+
+class Address
+  SS58_PREFIX = 'SS58PRE'
+
+  TYPES = [
+    # Polkadot Live (SS58, AccountId)
+    0, 1,
+    # Polkadot Canary (SS58, AccountId)
+    2, 3,
+    # Kulupu (SS58, Reserved)
+    16, 17,
+    # Dothereum (SS58, AccountId)
+    20, 21, 
+    # Generic Substrate wildcard (SS58, AccountId)
+    42, 43,
+
+    # Schnorr/Ristretto 25519 ("S/R 25519") key
+    48,
+    # Edwards Ed25519 key
+    49,
+    # ECDSA SECP256k1 key
+    50,
+
+    # Reserved for future address format extensions.
+    *64..255
+  ]
+
+  class << self
+
+    def array_to_hex_string(arr)
+      body = arr.map { |i| i.to_s(16).rjust(2, '0') }.join
+      "0x#{body}"
+    end
+
+    def decode(address, addr_type = 42, ignore_checksum = true)
+      decoded = Base58.base58_to_binary(address, :bitcoin)
+      is_pubkey = decoded.size == 35
+
+      size = decoded.size - ( is_pubkey ? 2 : 1 )
+
+      prefix = decoded[0, 1].unpack("C*").first
+
+      raise "Invalid address type" unless TYPES.include?(addr_type)
+      
+      hash_bytes = make_hash(decoded[0, size])
+      if is_pubkey
+        is_valid_checksum = decoded[-2].unpack("C*").first == hash_bytes[0] && decoded[-1].unpack("C*").first == hash_bytes[1]
+      else
+        is_valid_checksum = decoded[-1].unpack("C*").first == hash_bytes[0]
+      end
+
+      raise "Invalid decoded address checksum" unless is_valid_checksum && ignore_checksum
+
+      decoded[1...size].unpack("H*").first
+    end
+
+
+    def encode(pubkey, addr_type = 42)
+      pubkey = pubkey[2..-1] if pubkey =~ /^0x/i
+      key = [pubkey].pack("H*")
+
+      u8_array = key.bytes
+
+      u8_array.unshift(addr_type)
+
+      bytes = make_hash(u8_array.pack("C*"))
+      
+      checksum = bytes[0, key.size == 32 ? 2 : 1]
+
+      u8_array.push(*checksum)
+
+      input = u8_array.pack("C*")
+
+      Base58.binary_to_base58(input, :bitcoin)
+    end
+
+    def make_hash(body)
+      Blake2b.bytes("#{SS58_PREFIX}#{body}", Blake2b::Key.none, 64)
+    end
+
+  end
+end

data/lib/substrate_common/version.rb

@@ -1,3 +1,3 @@
 module SubstrateCommon
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/substrate_common.gemspec

@@ -38,7 +38,8 @@ Gem::Specification.new do |spec|
   spec.add_dependency "xxhash"
   spec.add_dependency "base58"
 
-  spec.add_development_dependency "bundler", "~> 1.17"
-  spec.add_development_dependency "rake", "~> 10.0"
+  # spec.add_development_dependency "bundler", "~> 1.17"
+  spec.add_development_dependency "rake", "~> 12.0"
   spec.add_development_dependency "pry"
+  spec.add_development_dependency "minitest"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: substrate_common.rb
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Wu Minzhe
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-10-22 00:00:00.000000000 Z
+date: 2019-10-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: blake2b
@@ -53,35 +53,35 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: bundler
+  name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.17'
+        version: '12.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.17'
+        version: '12.0'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: pry
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '0'
 - !ruby/object:Gem::Dependency
-  name: pry
+  name: minitest
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -102,6 +102,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".ruby-version"
 - Gemfile
 - Gemfile.lock
 - LICENSE.txt
@@ -110,6 +111,7 @@ files:
 - bin/console
 - bin/setup
 - lib/substrate_common.rb
+- lib/substrate_common/address.rb
 - lib/substrate_common/version.rb
 - substrate_common.gemspec
 homepage: https://github.com/wuminzhe/substrate_common.rb