<?php
/*
 * Laconica - a distributed open-source microblogging tool
 * Copyright (C) 2008, Controlez-Vous, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

if (!defined('LACONICA')) { exit(1); }

require_once('XMPPHP/XMPP.php');

function jabber_valid_base_jid($jid) {
	# Cheap but effective
	return Validate::email($jid);
}

function jabber_normalize_jid($jid) {
	if (preg_match("/(?:([^\@]+)\@)?([^\/]+)(?:\/(.*))?$/", $jid, $matches)) {
		$node = $matches[1];
		$server = $matches[2];
		$resource = $matches[3];
		return strtolower($node.'@'.$server);
	} else {
		return NULL;
	}
}

function jabber_daemon_address() {
	return common_config('xmpp', 'user') . '@' . common_config('xmpp', 'server');
}

function jabber_connect($resource=NULL) {
	static $conn = NULL;
	if (!$conn) {
		$conn = new XMPPHP_XMPP(common_config('xmpp', 'host') ?
								common_config('xmpp', 'host') :
								common_config('xmpp', 'server'),
								common_config('xmpp', 'port'),
								common_config('xmpp', 'user'),
								common_config('xmpp', 'password'),
								($resource) ? $resource :
								common_config('xmpp', 'resource'),
								common_config('xmpp', 'server'),
								common_config('xmpp', 'debug') ?
								true : false,
								common_config('xmpp', 'debug') ?
								XMPPHP_Log::LEVEL_VERBOSE :  NULL
								);
		$conn->autoSubscribe();

		if (!$conn) {
			return false;
		}
		$conn->connect(true); # true = persistent connection
		if ($conn->isDisconnected()) {
			return false;
		}
    	$conn->processUntil('session_start');
	}
	return $conn;
}

function jabber_send_notice($to, $notice) {
	$conn = jabber_connect();
	if (!$conn) {
		return false;
	}
	$profile = Profile::staticGet($notice->profile_id);
	if (!$profile) {
		common_log(LOG_WARNING, 'Refusing to send notice with ' .
		           'unknown profile ' . common_log_objstring($notice),
		           __FILE__);
		return false;
	}
	$msg = jabber_format_notice($profile, $notice);
	$entry = jabber_format_entry($profile, $notice);
	$stanza = jabber_make_stanza($to, $msg, 'chat', NULL, $entry);
	common_log(LOG_DEBUG, 'full stanza = ' . $stanza, __FILE__);
	$conn->send($stanza);
	return true;
}

function jabber_make_stanza($to, $body, $type = 'chat', $subject = null, $payload = null) {
	$to	  = htmlspecialchars($to);
	$body	= htmlspecialchars($body);
	$subject = htmlspecialchars($subject);

	$jid = jabber_daemon_address();
	
	$out = "<message from='$jid' to='$to' type='$type'>";
	if($subject) $out .= "<subject>$subject</subject>";
	$out .= "<body>$body</body>";
	if($payload) $out .= $payload;
	$out .= "</message>";
	
	return $out;
}

# Extra stuff defined by Twitter, needed by twitter clients

function jabber_format_entry($profile, $notice) {
	$noticeurl = common_local_url('shownotice',
								  array('notice' => $notice->id));
	$msg = jabber_format_notice($profile, $notice);
	$entry = "\n<entry xmlns='http://www.w3.org/2005/Atom'>\n";
	$entry .= "<source>\n";
	$entry .= "<title>" . $profile->nickname . " - " . common_config('site', 'name') . "</title>\n";
	$entry .= "<link href='" . $profile->profileurl . "'/>\n";
	$entry .= "<link rel='self' type='application/rss+xml' href='" . common_local_url('userrss', array('nickname' => $profile->nickname)) . "'/>\n";
	$entry .= "<author><name>" . $profile->nickname . "</name></author>\n";
	$entry .= "<icon>" . common_profile_avatar_url($profile, AVATAR_PROFILE_SIZE) . "</icon>\n";
	$entry .= "</source>\n";
	$entry .= "<title>" . htmlspecialchars($msg) . "</title>\n";
	$entry .= "<summary>" . htmlspecialchars($msg) . "</summary>\n";
	$entry .= "<link rel='alternate' href='" . $noticeurl . "' />\n";
	$entry .= "<id>". $notice->uri . "</id>\n";
	$entry .= "<published>".common_date_w3dtf($notice->created)."</published>\n";
	$entry .= "<updated>".common_date_w3dtf($notice->modified)."</updated>\n";
	$entry .= "</entry>\n";
	
	$event = "<event xmlns='http://jabber.org/protocol/pubsub#event'>\n";
    $event .= "<items xmlns='http://jabber.org/protocol/pubsub' ";
	$event .= "node='" . common_local_url('public') . "'>\n";
	$event .= "<item id='" . $notice->uri ."' />\n";
	$event .= "</items>\n";
	$event .= "</event>\n";
	# FIXME: include the pubsub event, too.
	return $entry;
#	return $entry . "\n" . $event;
}

function jabber_send_message($to, $body, $type='chat', $subject=NULL) {
	$conn = jabber_connect();
	if (!$conn) {
		return false;
	}
	$conn->message($to, $body, $type, $subject);
	return true;
}

function jabber_send_presence($status, $show='available', $to=Null) {
	$conn = jabber_connect();
	if (!$conn) {
		return false;
	}
	$conn->presence($status, $show, $to);
	return true;
}

function jabber_confirm_address($code, $nickname, $address) {
	$body = 'User "' . $nickname . '" on ' . common_config('site', 'name') . ' ' .
			'has said that your Jabber ID belongs to them. ' .
    	    'If that\'s true, you can confirm by clicking on this URL: ' .
        	common_local_url('confirmaddress', array('code' => $code)) .
        	' . (If you cannot click it, copy-and-paste it into the ' .
        	'address bar of your browser). If that user isn\'t you, ' .
        	'or if you didn\'t request this confirmation, just ignore this message.';

	return jabber_send_message($address, $body);
}

function jabber_special_presence($type, $to=NULL, $show=NULL, $status=NULL) {
	$conn = jabber_connect();

	$to = htmlspecialchars($to);
	$status = htmlspecialchars($status);
	$out = "<presence";
	if($to) $out .= " to='$to'";
	if($type) $out .= " type='$type'";
	if($show == 'available' and !$status) {
		$out .= "/>";
	} else {
		$out .= ">";
		if($show && ($show != 'available')) $out .= "<show>$show</show>";
		if($status) $out .= "<status>$status</status>";
		$out .= "</presence>";
	}
	$conn->send($out);
}

function jabber_broadcast_notice($notice) {
	# First, get users subscribed to this profile
	# XXX: use a join here rather than looping through results
	$profile = Profile::staticGet($notice->profile_id);
	if (!$profile) {
		common_log(LOG_WARNING, 'Refusing to broadcast notice with ' .
		           'unknown profile ' . common_log_objstring($notice),
		           __FILE__);
		return false;
	}
	$sub = new Subscription();
	$sub->subscribed = $notice->profile_id;
	if ($sub->find()) {
		while ($sub->fetch()) {
			$user = User::staticGet($sub->subscriber);
			if ($user && $user->jabber && $user->jabbernotify) {
				common_log(LOG_INFO,
						   'Sending notice ' . $notice->id . ' to ' . $user->jabber,
						   __FILE__);
				$success = jabber_send_notice($user->jabber, $notice);
				if (!$success) {
					# XXX: Not sure, but I think that's the right thing to do
					return false;
				}
			}
		}
	}
	return true;
}

function jabber_format_notice(&$profile, &$notice) {
	return $profile->nickname . ': ' . $notice->content;
}