1  #
   2  # httpauth/htpasswd -- Apache compatible htpasswd file
   3  #
   4  # Author: IPR -- Internet Programming with Ruby -- writers
   5  # Copyright (c) 2003 Internet Programming with Ruby writers. All rights
   6  # reserved.
   7  #
   8  # $IPR: htpasswd.rb,v 1.4 2003/07/22 19:20:45 gotoyuzo Exp $
   9 
  10  require 'webrick/httpauth/userdb'
  11  require 'webrick/httpauth/basicauth'
  12  require 'tempfile'
  13 
  14  module WEBrick
  15    module HTTPAuth
  16      class Htpasswd
  17        include UserDB
  18 
  19        def initialize(path)
  20          @path = path
  21          @mtime = Time.at(0)
  22          @passwd = Hash.new
  23          @auth_type = BasicAuth
  24          open(@path,"a").close unless File::exist?(@path)
  25          reload
  26        end
  27 
  28        def reload
  29          mtime = File::mtime(@path)
  30          if mtime > @mtime
  31            @passwd.clear
  32            open(@path){|io|
  33              while line = io.gets
  34                line.chomp!
  35                case line
  36                when %r!\A[^:]+:[a-zA-Z0-9./]{13}\z!
  37                  user, pass = line.split(":")
  38                when /:\$/, /:\{SHA\}/
  39                  raise NotImplementedError,
  40                        'MD5, SHA1 .htpasswd file not supported'
  41                else
  42                  raise StandardError, 'bad .htpasswd file'
  43                end
  44                @passwd[user] = pass
  45              end
  46            }
  47            @mtime = mtime
  48          end
  49        end
  50 
  51        def flush(output=nil)
  52          output ||= @path
  53          tmp = Tempfile.new("htpasswd", File::dirname(output))
  54          begin
  55            each{|item| tmp.puts(item.join(":")) }
  56            tmp.close
  57            File::rename(tmp.path, output)
  58          rescue
  59            tmp.close(true)
  60          end
  61        end
  62 
  63        def get_passwd(realm, user, reload_db)
  64          reload() if reload_db
  65          @passwd[user]
  66        end
  67 
  68        def set_passwd(realm, user, pass)
  69          @passwd[user] = make_passwd(realm, user, pass)
  70        end
  71 
  72        def delete_passwd(realm, user)
  73          @passwd.delete(user)
  74        end
  75 
  76        def each
  77          @passwd.keys.sort.each{|user|
  78            yield([user, @passwd[user]])
  79          }
  80        end
  81      end
  82    end
  83  end