#!/usr/bin/ruby -w
##
## SMB Search: SMB classes extender
## Copyright (c) 2007 OSS Technology Co., Japan <http://www.osstech.co.jp/>
##
## Date: 2007-07-06, since 2007-06-18
## Author: SATOH Fumiyasu
##

require 'smb'

## Override SMB::Dir::Entry#uri to hide "domain;user:password" in URI
class SMB::Dir::Entry
  alias original_uri uri
  private :original_uri
  def uri
    return original_uri.sub(%r#^([^/:]+://)[^/]+@#, '\1')
  end
end

class SMBAccount
  attr_reader :sid, :domain, :name, :type

  def initialize(sid, domain, name, type)
    @sid = sid
    @domain = domain
    @name = name
    @type = type
  end

  def ==(b)
    return (
      @sid == b.sid &&
      @domain.downcase == b.domain.downcase &&
      @name.downcase == b.name.downcase &&
      @type == b.type)
  end
end

