ejabberd db dump importer for Prosody
[prosody.git] / plugins / mod_register.lua
index ad0ba47819ac90a44d64eeb75d743a81bf57ce2a..5769bd804b879864ab10c1027f19c0b55c032e07 100644 (file)
@@ -1,10 +1,12 @@
 
 local st = require "util.stanza";
-local send = require "core.sessionmanager".send_to_session;
 local usermanager_user_exists = require "core.usermanager".user_exists;
 local usermanager_create_user = require "core.usermanager".create_user;
+local datamanager_store = require "util.datamanager".store;
 
-add_iq_handler("c2s", "jabber:iq:register", function (session, stanza)
+require "core.discomanager".set("register", "jabber:iq:register");
+
+module:add_iq_handler("c2s", "jabber:iq:register", function (session, stanza)
        if stanza.tags[1].name == "query" then
                local query = stanza.tags[1];
                if stanza.attr.type == "get" then
@@ -13,11 +15,37 @@ add_iq_handler("c2s", "jabber:iq:register", function (session, stanza)
                                :tag("registered"):up()
                                :tag("username"):text(session.username):up()
                                :tag("password"):up();
-                       send(session, reply);
+                       session.send(reply);
                elseif stanza.attr.type == "set" then
                        if query.tags[1] and query.tags[1].name == "remove" then
                                -- TODO delete user auth data, send iq response, kick all user resources with a <not-authorized/>, delete all user data
-                               send(session, st.error_reply(stanza, "cancel", "not-allowed"));
+                               --session.send(st.error_reply(stanza, "cancel", "not-allowed"));
+                               --return;
+                               usermanager_create_user(session.username, nil, session.host); -- Disable account
+                               -- FIXME the disabling currently allows a different user to recreate the account
+                               -- we should add an in-memory account block mode when we have threading
+                               session.send(st.reply(stanza));
+                               local roster = session.roster;
+                               for _, session in pairs(hosts[session.host].sessions[session.username].sessions) do -- disconnect all resources
+                                       session:disconnect({condition = "not-authorized", text = "Account deleted"});
+                               end
+                               -- TODO datamanager should be able to delete all user data itself
+                               datamanager.store(session.username, session.host, "roster", nil);
+                               datamanager.store(session.username, session.host, "vCard", nil);
+                               datamanager.store(session.username, session.host, "private", nil);
+                               datamanager.store(session.username, session.host, "offline", nil);
+                               local bare = session.username.."@"..session.host;
+                               for jid, item in pairs(roster) do
+                                       if jid ~= "pending" then
+                                               if item.subscription == "both" or item.subscription == "to" then
+                                                       -- TODO unsubscribe
+                                               end
+                                               if item.subscription == "both" or item.subscription == "from" then
+                                                       -- TODO unsubscribe
+                                               end
+                                       end
+                               end
+                               datamanager.store(session.username, session.host, "accounts", nil); -- delete accounts datastore at the end
                        else
                                local username = query:child_with_name("username");
                                local password = query:child_with_name("password");
@@ -27,25 +55,25 @@ add_iq_handler("c2s", "jabber:iq:register", function (session, stanza)
                                        password = table.concat(password);
                                        if username == session.username then
                                                if usermanager_create_user(username, password, session.host) then -- password change -- TODO is this the right way?
-                                                       send(session, st.reply(stanza));
+                                                       session.send(st.reply(stanza));
                                                else
                                                        -- TODO unable to write file, file may be locked, etc, what's the correct error?
-                                                       send(session, st.error_reply(stanza, "wait", "internal-server-error"));
+                                                       session.send(st.error_reply(stanza, "wait", "internal-server-error"));
                                                end
                                        else
-                                               send(session, st.error_reply(stanza, "modify", "bad-request"));
+                                               session.send(st.error_reply(stanza, "modify", "bad-request"));
                                        end
                                else
-                                       send(session, st.error_reply(stanza, "modify", "bad-request"));
+                                       session.send(st.error_reply(stanza, "modify", "bad-request"));
                                end
                        end
                end
        else
-               send(session, st.error_reply(stanza, "cancel", "service-unavailable"));
+               session.send(st.error_reply(stanza, "cancel", "service-unavailable"));
        end;
 end);
 
-add_iq_handler("c2s_unauthed", "jabber:iq:register", function (session, stanza)
+module:add_iq_handler("c2s_unauthed", "jabber:iq:register", function (session, stanza)
        if stanza.tags[1].name == "query" then
                local query = stanza.tags[1];
                if stanza.attr.type == "get" then
@@ -54,10 +82,10 @@ add_iq_handler("c2s_unauthed", "jabber:iq:register", function (session, stanza)
                                :tag("instructions"):text("Choose a username and password for use with this service."):up()
                                :tag("username"):up()
                                :tag("password"):up();
-                       send(session, reply);
+                       session.send(reply);
                elseif stanza.attr.type == "set" then
                        if query.tags[1] and query.tags[1].name == "remove" then
-                               send(session, st.error_reply(stanza, "auth", "registration-required"));
+                               session.send(st.error_reply(stanza, "auth", "registration-required"));
                        else
                                local username = query:child_with_name("username");
                                local password = query:child_with_name("password");
@@ -66,21 +94,21 @@ add_iq_handler("c2s_unauthed", "jabber:iq:register", function (session, stanza)
                                        username = table.concat(username);
                                        password = table.concat(password);
                                        if usermanager_user_exists(username, session.host) then
-                                               send(session, st.error_reply(stanza, "cancel", "conflict"));
+                                               session.send(st.error_reply(stanza, "cancel", "conflict"));
                                        else
                                                if usermanager_create_user(username, password, session.host) then
-                                                       send(session, st.reply(stanza)); -- user created!
+                                                       session.send(st.reply(stanza)); -- user created!
                                                else
                                                        -- TODO unable to write file, file may be locked, etc, what's the correct error?
-                                                       send(session, st.error_reply(stanza, "wait", "internal-server-error"));
+                                                       session.send(st.error_reply(stanza, "wait", "internal-server-error"));
                                                end
                                        end
                                else
-                                       send(session, st.error_reply(stanza, "modify", "not-acceptable"));
+                                       session.send(st.error_reply(stanza, "modify", "not-acceptable"));
                                end
                        end
                end
        else
-               send(session, st.error_reply(stanza, "cancel", "service-unavailable"));
+               session.send(st.error_reply(stanza, "cancel", "service-unavailable"));
        end;
 end);