rapid7/nexpose-client

module Nexpose
  require 'json'
  # Configuration structure for password policies.
  class PasswordPolicy < APIObject
    attr_accessor :policy_name
    attr_accessor :min_length
    attr_accessor :max_length
    attr_accessor :capitals
    attr_accessor :digits
    attr_accessor :special_chars
    attr_accessor :expiration_days

    def initialize(policy_name:, min_length:, max_length:, special_chars:, capitals:, digits:, expiration_days: 0)
      @policy_name     = policy_name.to_s
      @min_length      = min_length.to_i
      @max_length      = max_length.to_i
      @special_chars   = special_chars.to_i
      @capitals        = capitals.to_i
      @digits          = digits.to_i
      @expiration_days = expiration_days.to_i
    end

    def self.from_hash(hash)
      new(policy_name: hash[:policyName],
          min_length: hash[:minLength],
          max_length: hash[:maxLength],
          special_chars: hash[:specialChars],
          capitals: hash[:capitals],
          digits: hash[:digits],
          expiration_days: hash[:expirationDays])
    end

    def to_h
      {
        policyName: @policy_name,
        minLength: @min_length,
        maxLength: @max_length,
        specialChars: @special_chars,
        capitals: @capitals,
        digits: @digits,
        expirationDays: @expiration_days
      }
    end

    def to_json
      JSON.generate(to_h)
    end

    def save(nsc)
      params = to_json
      AJAX.post(nsc, '/api/2.1/password_policy/', params, AJAX::CONTENT_TYPE::JSON)
    end

    def self.load(nsc)
      uri  = '/api/2.1/password_policy/'
      resp = AJAX.get(nsc, uri, AJAX::CONTENT_TYPE::JSON)
      hash = JSON.parse(resp, symbolize_names: true)
      self.from_hash(hash)
    end
  end
end