1 ! Copyright (c) 2008 Slava Pestov
\r
2 ! See http://factorcode.org/license.txt for BSD license.
\r
3 USING: accessors assocs namespaces kernel sequences sets
\r
4 destructors combinators fry logging
\r
5 io.encodings.utf8 io.encodings.string io.binary random
\r
6 checksums checksums.sha2 urls
\r
10 http.server.dispatchers
\r
15 furnace.auth.providers
\r
16 furnace.auth.providers.db ;
\r
19 SYMBOL: logged-in-user
\r
21 : logged-in? ( -- ? )
\r
22 logged-in-user get >boolean ;
\r
24 : username ( -- string/f )
\r
25 logged-in-user get dup [ username>> ] when ;
\r
27 GENERIC: init-user-profile ( responder -- )
\r
29 M: object init-user-profile drop ;
\r
31 M: dispatcher init-user-profile
\r
32 default>> init-user-profile ;
\r
34 M: filter-responder init-user-profile
\r
35 responder>> init-user-profile ;
\r
37 : profile ( -- assoc ) logged-in-user get profile>> ;
\r
39 : user-changed ( -- )
\r
40 logged-in-user get t >>changed? drop ;
\r
42 : uget ( key -- value )
\r
45 : uset ( value key -- )
\r
49 : uchange ( quot key -- )
\r
50 profile swap change-at
\r
51 user-changed ; inline
\r
53 SYMBOL: capabilities
\r
55 V{ } clone capabilities set-global
\r
57 : define-capability ( word -- ) capabilities get adjoin ;
\r
59 TUPLE: realm < dispatcher name users checksum secure ;
\r
61 GENERIC: login-required* ( description capabilities realm -- response )
\r
63 GENERIC: user-registered ( user realm -- response )
\r
65 M: object user-registered 2drop URL" $realm" <redirect> ;
\r
67 GENERIC: init-realm ( realm -- )
\r
69 GENERIC: logged-in-username ( realm -- username )
\r
71 : login-required ( description capabilities -- * )
\r
72 realm get login-required* exit-with ;
\r
74 : new-realm ( responder name class -- realm )
\r
82 : users ( -- provider )
\r
85 TUPLE: user-saver user ;
\r
87 C: <user-saver> user-saver
\r
89 M: user-saver dispose
\r
90 user>> dup changed?>> [ users update-user ] [ drop ] if ;
\r
92 : save-user-after ( user -- )
\r
93 <user-saver> &dispose drop ;
\r
95 : init-user ( user -- )
\r
96 [ [ logged-in-user set ] [ save-user-after ] bi ] when* ;
\r
98 \ init-user DEBUG add-input-logging
\r
100 M: realm call-responder* ( path responder -- response )
\r
104 dup logged-in-username
\r
105 dup [ users get-user ] when
\r
110 : encode-password ( string salt -- bytes )
\r
111 [ utf8 encode ] [ 4 >be ] bi* append
\r
112 realm get checksum>> checksum-bytes ;
\r
114 : >>encoded-password ( user string -- user )
\r
115 32 random-bits [ encode-password ] keep
\r
116 [ >>password ] [ >>salt ] bi* ; inline
\r
118 : valid-login? ( password user -- ? )
\r
119 [ salt>> encode-password ] [ password>> ] bi = ;
\r
121 : check-login ( password username -- user/f )
\r
122 users get-user dup [ [ valid-login? ] keep and ] [ 2drop f ] if ;
\r
124 : if-secure-realm ( quot -- )
\r
125 realm get secure>> [ if-secure ] [ call ] if ; inline
\r
127 TUPLE: secure-realm-only < filter-responder ;
\r
129 C: <secure-realm-only> secure-realm-only
\r
131 M: secure-realm-only call-responder*
\r
132 '[ _ _ call-next-method ] if-secure-realm ;
\r
134 TUPLE: protected < filter-responder description capabilities ;
\r
136 : <protected> ( responder -- protected )
\r
140 : have-capabilities? ( capabilities -- ? )
\r
141 realm get secure>> secure-connection? not and [ drop f ] [
\r
142 logged-in-user get {
\r
143 { [ dup not ] [ 2drop f ] }
\r
144 { [ dup deleted>> 1 = ] [ 2drop f ] }
\r
145 [ capabilities>> subset? ]
\r
149 M: protected call-responder* ( path responder -- response )
\r
151 dup capabilities>> have-capabilities?
\r
152 [ call-next-method ] [
\r
153 [ drop ] [ [ description>> ] [ capabilities>> ] bi ] bi*
\r
154 realm get login-required*
\r
157 : <auth-boilerplate> ( responder -- responder' )
\r
158 <boilerplate> { realm "boilerplate" } >>template ;
\r
160 : password-mismatch ( -- * )
\r
161 "passwords do not match" validation-error
\r
162 validation-failed ;
\r
164 : same-password-twice ( -- )
\r
165 "new-password" value "verify-password" value =
\r
166 [ password-mismatch ] unless ;
\r
168 : user-exists ( -- * )
\r
169 "username taken" validation-error
\r
170 validation-failed ;
\r