libremetaverse/Programs/Simian/Extensions/FriendManager.cs

using System;
using System.Collections.Generic;
using ExtensionLoader;
using OpenMetaverse;
using OpenMetaverse.Packets;

namespace Simian.Extensions
{
    public class FriendManager : IExtension<Simian>
    {
        Simian server;

        public FriendManager()
        {
            
        }

        public void Start(Simian server)
        {
            this.server = server;

            server.UDP.RegisterPacketCallback(PacketType.ImprovedInstantMessage, new PacketCallback(ImprovedInstantMessageHandler));
        }

        public void Stop()
        {
        }

        void ImprovedInstantMessageHandler(Packet packet, Agent agent)
        {
            ImprovedInstantMessagePacket im = (ImprovedInstantMessagePacket)packet;
            InstantMessageDialog dialog = (InstantMessageDialog)im.MessageBlock.Dialog;

            if (dialog == InstantMessageDialog.FriendshipOffered || dialog == InstantMessageDialog.FriendshipAccepted || dialog == InstantMessageDialog.FriendshipDeclined)
            {
                lock (server.Agents)
                {
                    // HACK: Only works for agents currently online
                    Agent recipient;
                    if (server.Agents.TryGetValue(im.MessageBlock.ToAgentID, out recipient))
                    {
                        ImprovedInstantMessagePacket sendIM = new ImprovedInstantMessagePacket();
                        sendIM.MessageBlock.RegionID = UUID.Random(); //FIXME
                        sendIM.MessageBlock.ParentEstateID = 1;
                        sendIM.MessageBlock.FromGroup = false;
                        sendIM.MessageBlock.FromAgentName = Utils.StringToBytes(agent.Avatar.Name);
                        sendIM.MessageBlock.ToAgentID = im.MessageBlock.ToAgentID;
                        sendIM.MessageBlock.Dialog = im.MessageBlock.Dialog;
                        sendIM.MessageBlock.Offline = (byte)InstantMessageOnline.Online;
                        sendIM.MessageBlock.ID = agent.AgentID;
                        sendIM.MessageBlock.Message = im.MessageBlock.Message;
                        sendIM.MessageBlock.BinaryBucket = new byte[0];
                        sendIM.MessageBlock.Timestamp = 0;
                        sendIM.MessageBlock.Position = agent.Avatar.Position;

                        sendIM.AgentData.AgentID = agent.AgentID;

                        server.UDP.SendPacket(recipient.AgentID, sendIM, PacketCategory.Transaction);

                        if (dialog == InstantMessageDialog.FriendshipAccepted)
                        {
                            bool receiverOnline = server.Agents.ContainsKey(agent.AgentID);
                            bool senderOnline = server.Agents.ContainsKey(recipient.AgentID);

                            if (receiverOnline)
                            {
                                if (senderOnline)
                                {
                                    OnlineNotificationPacket notify = new OnlineNotificationPacket();
                                    notify.AgentBlock = new OnlineNotificationPacket.AgentBlockBlock[0];
                                    notify.AgentBlock[0] = new OnlineNotificationPacket.AgentBlockBlock();
                                    notify.AgentBlock[0].AgentID = agent.AgentID;
                                    server.UDP.SendPacket(recipient.AgentID, notify, PacketCategory.State);
                                }
                                else
                                {
                                    OfflineNotificationPacket notify = new OfflineNotificationPacket();
                                    notify.AgentBlock = new OfflineNotificationPacket.AgentBlockBlock[0];
                                    notify.AgentBlock[0] = new OfflineNotificationPacket.AgentBlockBlock();
                                    notify.AgentBlock[0].AgentID = agent.AgentID;
                                    server.UDP.SendPacket(recipient.AgentID, notify, PacketCategory.State);
                                }
                            }

                            if (senderOnline)
                            {
                                if (receiverOnline)
                                {
                                    OnlineNotificationPacket notify = new OnlineNotificationPacket();
                                    notify.AgentBlock = new OnlineNotificationPacket.AgentBlockBlock[0];
                                    notify.AgentBlock[0] = new OnlineNotificationPacket.AgentBlockBlock();
                                    notify.AgentBlock[0].AgentID = recipient.AgentID;
                                    server.UDP.SendPacket(agent.AgentID, notify, PacketCategory.State);
                                }
                                else
                                {
                                    OfflineNotificationPacket notify = new OfflineNotificationPacket();
                                    notify.AgentBlock = new OfflineNotificationPacket.AgentBlockBlock[0];
                                    notify.AgentBlock[0] = new OfflineNotificationPacket.AgentBlockBlock();
                                    notify.AgentBlock[0].AgentID = recipient.AgentID;
                                    server.UDP.SendPacket(agent.AgentID, notify, PacketCategory.State);
                                }
                            }
                        }   
                    }
                }
            }
        }

    }
}