Registration requests if started with -R.
parent
b1b502ff66
commit
9ae887071e
|
@ -184,6 +184,13 @@ class AuthD::Request
|
||||||
initialize :shared_key, :uid
|
initialize :shared_key, :uid
|
||||||
end
|
end
|
||||||
|
|
||||||
|
class Request::Register < Request
|
||||||
|
property login : String
|
||||||
|
property password : String
|
||||||
|
|
||||||
|
initialize :shared_key, :login, :password
|
||||||
|
end
|
||||||
|
|
||||||
# This creates a Request::Type enumeration. One entry for each request type.
|
# This creates a Request::Type enumeration. One entry for each request type.
|
||||||
{% begin %}
|
{% begin %}
|
||||||
enum Type
|
enum Type
|
||||||
|
|
27
src/main.cr
27
src/main.cr
|
@ -11,6 +11,8 @@ require "./authd.cr"
|
||||||
extend AuthD
|
extend AuthD
|
||||||
|
|
||||||
class AuthD::Service
|
class AuthD::Service
|
||||||
|
property registrations_allowed = false
|
||||||
|
|
||||||
def initialize(@passwd : Passwd, @jwt_key : String)
|
def initialize(@passwd : Passwd, @jwt_key : String)
|
||||||
end
|
end
|
||||||
|
|
||||||
|
@ -66,6 +68,18 @@ class AuthD::Service
|
||||||
@passwd.mod_user request.uid, password_hash: password_hash
|
@passwd.mod_user request.uid, password_hash: password_hash
|
||||||
|
|
||||||
Response::UserEdited.new request.uid
|
Response::UserEdited.new request.uid
|
||||||
|
when Request::Register
|
||||||
|
if ! @registrations_allowed
|
||||||
|
return Response::Error.new "registrations not allowed"
|
||||||
|
end
|
||||||
|
|
||||||
|
if @passwd.user_exists? request.login
|
||||||
|
return Response::Error.new "login already used"
|
||||||
|
end
|
||||||
|
|
||||||
|
user = @passwd.add_user request.login, request.password
|
||||||
|
|
||||||
|
Response::UserAdded.new user
|
||||||
else
|
else
|
||||||
Response::Error.new "unhandled request type"
|
Response::Error.new "unhandled request type"
|
||||||
end
|
end
|
||||||
|
@ -83,11 +97,15 @@ class AuthD::Service
|
||||||
|
|
||||||
case event
|
case event
|
||||||
when IPC::Event::Message
|
when IPC::Event::Message
|
||||||
|
begin
|
||||||
request = Request.from_ipc event.message
|
request = Request.from_ipc event.message
|
||||||
|
|
||||||
response = handle_request request, event.connection
|
response = handle_request request, event.connection
|
||||||
|
|
||||||
event.connection.send response
|
event.connection.send response
|
||||||
|
rescue e
|
||||||
|
STDERR.puts "error: #{e.message}"
|
||||||
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
@ -96,6 +114,7 @@ end
|
||||||
authd_passwd_file = "passwd"
|
authd_passwd_file = "passwd"
|
||||||
authd_group_file = "group"
|
authd_group_file = "group"
|
||||||
authd_jwt_key = "nico-nico-nii"
|
authd_jwt_key = "nico-nico-nii"
|
||||||
|
authd_registrations = false
|
||||||
|
|
||||||
OptionParser.parse do |parser|
|
OptionParser.parse do |parser|
|
||||||
parser.on "-u file", "--passwd-file file", "passwd file." do |name|
|
parser.on "-u file", "--passwd-file file", "passwd file." do |name|
|
||||||
|
@ -110,6 +129,10 @@ OptionParser.parse do |parser|
|
||||||
authd_jwt_key = File.read(file_name).chomp
|
authd_jwt_key = File.read(file_name).chomp
|
||||||
end
|
end
|
||||||
|
|
||||||
|
parser.on "-R", "--allow-registrations" do
|
||||||
|
authd_registrations = true
|
||||||
|
end
|
||||||
|
|
||||||
parser.on "-h", "--help", "Show this help" do
|
parser.on "-h", "--help", "Show this help" do
|
||||||
puts parser
|
puts parser
|
||||||
|
|
||||||
|
@ -119,5 +142,7 @@ end
|
||||||
|
|
||||||
passwd = Passwd.new authd_passwd_file, authd_group_file
|
passwd = Passwd.new authd_passwd_file, authd_group_file
|
||||||
|
|
||||||
AuthD::Service.new(passwd, authd_jwt_key).run
|
AuthD::Service.new(passwd, authd_jwt_key).tap do |authd|
|
||||||
|
authd.registrations_allowed = authd_registrations
|
||||||
|
end.run
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue