X-Git-Url: https://git.enpas.org/?a=blobdiff_plain;f=plugins%2Fmod_vcard.lua;h=2e8aea02b41f50f0ef58472140543c0ae971aeac;hb=2f107f1ef33466161ada9f796a1a56e2cce973a4;hp=67c8309ac9b4950d6d9bf38006ae93ecdb9644c7;hpb=8a7849df4d545c002b897ca54a4be7315cd91049;p=prosody.git

diff --git a/plugins/mod_vcard.lua b/plugins/mod_vcard.lua
index 67c8309a..2e8aea02 100644
--- a/plugins/mod_vcard.lua
+++ b/plugins/mod_vcard.lua
@@ -1,15 +1,25 @@
+-- Prosody IM v0.4
+-- Copyright (C) 2008-2009 Matthew Wild
+-- Copyright (C) 2008-2009 Waqas Hussain
+-- 
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
+--
 
-require "util.datamanager"
-local datamanager = datamanager;
+
+
+local hosts = _G.hosts;
+local datamanager = require "util.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 = 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;
@@ -18,35 +28,36 @@ add_iq_handler({"c2s", "s2sin"}, "vcard-temp",
 					if to then
 						local node, host = jid_split(to);
 						if hosts[host] and hosts[host].type == "local" then
-							vCard = st.deserialize(datamanager.load(node, host, "vCard")); -- load vCard for user or server
+							vCard = st.deserialize(datamanager.load(node, host, "vcard")); -- load vCard for user or server
 						end
 					else
-						vCard = st.deserialize(datamanager.load(session.username, session.host, "vCard"));-- load user's own vCard
+						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));
+						if datamanager.store(session.username, session.host, "vcard", st.preserialize(stanza.tags[1])) then
+							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", 
-					function (session, 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);