grempe/tss-rb

module Contracts

  # Custom Contracts
  # See : https://egonschiele.github.io/contracts.ruby/

  class ArrayOfShares
    def self.valid? val
      val.is_a?(Array) &&
      val.length.between?(1,255) &&
      Contracts::ArrayOf[String].valid?(val)
    end

    def self.to_s
      'An Array of split secret shares'
    end
  end

  class SecretArg
    def self.valid? val
      val.is_a?(String) &&
      val.length.between?(1,TSS::MAX_UNPADDED_SECRET_SIZE) &&
      ['UTF-8', 'US-ASCII'].include?(val.encoding.name)
    end

    def self.to_s
      "must be a UTF-8 or US-ASCII String between 1 and #{TSS::MAX_UNPADDED_SECRET_SIZE} characters in length"
    end
  end

  class ThresholdArg
    def self.valid? val
      val.is_a?(Integer) &&
      val.between?(1,255)
    end

    def self.to_s
      'must be an Integer between 1 and 255'
    end
  end

  class NumSharesArg
    def self.valid? val
      val.is_a?(Integer) &&
      val.between?(1,255)
    end

    def self.to_s
      'must be an Integer between 1 and 255'
    end
  end

  class IdentifierArg
    def self.valid? val
      val.is_a?(String) &&
      val.length.between?(1,16) &&
      val =~ /^[a-zA-Z0-9\-\_\.]*$/i
    end

    def self.to_s
      'must be a String between 1 and 16 characters in length limited to [a-z, A-Z, -, _, .]'
    end
  end

  class HashAlgArg
    def self.valid? val
      Contracts::Enum['NONE', 'SHA1', 'SHA256'].valid?(val)
    end

    def self.to_s
      'must be a uppercase String specifying the hash algorithm to use [NONE, SHA1, SHA256].'
    end
  end

  class FormatArg
    def self.valid? val
      Contracts::Enum['BINARY', 'HUMAN'].valid?(val)
    end

    def self.to_s
      'must be a uppercase String specifying the desired String share format [BINARY, HUMAN].'
    end
  end

  class SelectByArg
    def self.valid? val
      Contracts::Enum['FIRST', 'SAMPLE', 'COMBINATIONS'].valid?(val)
    end

    def self.to_s
      'must be a uppercase String specifying the desired way to sample shares provided [FIRST, SAMPLE, COMBINATIONS].'
    end
  end

  class PadBlocksizeArg
    def self.valid? val
      val.is_a?(Integer) &&
      val.between?(0,255)
    end

    def self.to_s
      'must be an Integer between 0 and 255'
    end
  end
end