/tools/Ruby/lib/ruby/1.8/webrick/httpauth/htpasswd.rb

 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
10require 'webrick/httpauth/userdb'
11require 'webrick/httpauth/basicauth'
12require 'tempfile'
13
14module 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
83end