Change modules to use the new add_feature module API method.

[prosody.git] / plugins / mod_vcard.lua

diff --git a/plugins/mod_vcard.lua b/plugins/mod_vcard.lua
index 67c8309ac9b4950d6d9bf38006ae93ecdb9644c7..4f78867b9fd40f66fc168e3dc3895a932d54daf2 100644 (file)
--- a/plugins/mod_vcard.lua
+++ b/plugins/mod_vcard.lua
@@ -1,15 +1,36 @@
+-- Prosody IM v0.1
+-- Copyright (C) 2008 Matthew Wild
+-- Copyright (C) 2008 Waqas Hussain
+-- 
+-- This program is free software; you can redistribute it and/or
+-- modify it under the terms of the GNU General Public License
+-- as published by the Free Software Foundation; either version 2
+-- of the License, or (at your option) any later version.
+-- 
+-- This program is distributed in the hope that it will be useful,
+-- but WITHOUT ANY WARRANTY; without even the implied warranty of
+-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+-- GNU General Public License for more details.
+-- 
+-- You should have received a copy of the GNU General Public License
+-- along with this program; if not, write to the Free Software
+-- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+--
+
+
 
 require "util.datamanager"
 local datamanager = datamanager;
 
 local st = require "util.stanza"
-local send = require "core.sessionmanager".send_to_session
 local t_concat, t_insert = table.concat, table.insert;
 
 require "util.jid"
 local jid_split = jid.split;
 
-add_iq_handler({"c2s", "s2sin"}, "vcard-temp", 
+module:add_feature("vcard-temp");
+
+module:add_iq_handler({"c2s", "s2sin"}, "vcard-temp", 
                function (session, stanza)
                        if stanza.tags[1].name == "vCard" then
                                local to = stanza.attr.to;
@@ -24,29 +45,30 @@ add_iq_handler({"c2s", "s2sin"}, "vcard-temp",
                                                vCard = st.deserialize(datamanager.load(session.username, session.host, "vCard"));-- load user's own vCard
                                        end
                                        if vCard then
-                                               send(session, st.reply(stanza):add_child(vCard)); -- send vCard!
+                                               session.send(st.reply(stanza):add_child(vCard)); -- send vCard!
                                        else
-                                               send(session, st.error_reply(stanza, "cancel", "item-not-found"));
+                                               session.send(st.error_reply(stanza, "cancel", "item-not-found"));
                                        end
                                elseif stanza.attr.type == "set" then
                                        if not to or to == session.username.."@"..session.host then
                                                if datamanager.store(session.username, session.host, "vCard", st.preserialize(stanza.tags[1])) then
-                                                       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, "auth", "forbidden"));
+                                               session.send(st.error_reply(stanza, "auth", "forbidden"));
                                        end
                                end
                                return true;
                        end
                end);
 
-add_event_hook("stream-features", 
+local feature_vcard_attr = { var='vcard-temp' };
+module:add_event_hook("stream-features", 
                                        function (session, features)                                                                                            
                                                if session.type == "c2s" then
-                                                       t_insert(features, "<feature var='vcard-temp'/>");
+                                                       features:tag("feature", feature_vcard_attr):up();
                                                end
                                        end);