pymetaverse/metaverse/bot.py

import asyncio
import struct
import math
import random
import datetime
from .eventtarget import EventTarget
from . import login
from . import viewer as Viewer
from .const import *

import logging
logger = logging.getLogger(__name__)

class SimpleBot(EventTarget):
    def __init__(self):
        super().__init__()
        self.agent = Viewer.Agent()
        self.agent.on("Message", self.handleMessage)
        self.agent.on("Event", self.handleEvent)
    
    async def handleSystemMessages(self, simulator, message):
        # We only really care about the parent simulator here
        if simulator == self.simulator:
            pass
        
        if message.name == "RegionHandshake":
            # Send some stuff to make the simulator happy about our presence
            msg = self.messageTemplate.getMessage("AgentThrottle")
            msg.AgentData.AgentID = self.agent.agentId
            msg.AgentData.SessionID = self.agent.sessionId
            msg.AgentData.CircuitCode = self.agent.circuitCode
            msg.Throttle.GenCounter = 0
            msg.Throttle.Throttles = struct.pack("<7f",
                #http://wiki.secondlife.com/wiki/AgentThrottle
                150000,  #Resend
                170000,  #Land
                34000,   #Wind
                34000,   #Cloud
                446000,  #Task
                446000,  #Texture
                220000   #Asset
            )
            self.send(msg)
            
            msg = self.messageTemplate.getMessage("AgentFOV")
            msg.AgentData.AgentID = self.agent.agentId
            msg.AgentData.SessionID = self.agent.sessionId
            msg.AgentData.CircuitCode = self.agent.circuitCode
            msg.FOVBlock.GenCounter = 0
            msg.FOVBlock.VerticalAngle = 6.233185307179586
            self.send(msg)
            
            msg = self.messageTemplate.getMessage("AgentHeightWidth")
            msg.AgentData.AgentID = self.agent.agentId
            msg.AgentData.SessionID = self.agent.sessionId
            msg.AgentData.CircuitCode = self.agent.circuitCode
            msg.HeightWidthBlock.GenCounter = 0
            msg.HeightWidthBlock.Height = 0xffff
            msg.HeightWidthBlock.Width = 0xffff
            self.send(msg)

            self.agentUpdate()

        elif message.name == "ImprovedInstantMessage":
            await self.fire("InstantMessage",
                message.MessageBlock.ID,
                message.AgentData.AgentID,
                message.MessageBlock.FromAgentName.decode(),
                message.MessageBlock.Message.decode(),
                message.MessageBlock.Offline == 1,
                message.MessageBlock.BinaryBucket,
                message.MessageBlock.Dialog,
                datetime.datetime.fromtimestamp(message.MessageBlock.Timestamp)
            )
    
    async def handleSystemEvent(self, sim, name, body):
        if name == "ChatterBoxInvitation":
            if "instantmessage" in body:
                im = body["instantmessage"]["message_params"]
                await sim.capabilities["ChatSessionRequest"].acceptInvitation(im["id"])
                await self.fire("InstantMessage",
                    im["id"],
                    im["from_id"],
                    im["from_name"],
                    im["message"],
                    im["offline"] == 1,
                    im["data"]["binary_bucket"],
                    IM_SESSION_INVITE,
                    datetime.datetime.fromtimestamp(im["timestamp"])
                )
            
    @property
    def simulator(self):
        return self.agent.simulator
    
    @property
    def messageTemplate(self):
        return self.agent.messageTemplate
    
    async def handleEvent(self, sim, name, body):
        await self.handleSystemEvent(sim, name, body)
        await self.fire("Event", sim, name, body, name=name)

    async def handleMessage(self, simulator, message):
        await self.handleSystemMessages(simulator, message)
        await self.fire("Message", simulator, message, name=message.name)
    
    def send(self, message, reliable = False):
        self.agent.send(message, reliable)
    
    async def login(self, *args, **kwargs):
        loginHandle = await login.Login(*args, **kwargs, isBot = True)
        if loginHandle["login"] == "false":
            logger.critical("Login failure: {}".format(loginHandle["message"]))
            raise ValueError("Incorrect username or password")
        
        logger.info("Login success: {}".format(loginHandle["message"]))
        
        await self.agent.login(loginHandle)
    
    async def run(self):
        await self.agent.run()
    
    def logout(self):
        self.agent.logout()
    
    def say(self, channel, message, ctype = CHAT_TYPE_NORMAL):
        if channel >= 0:
            msg = self.messageTemplate.getMessage("ChatFromViewer")
            msg.AgentData.AgentID = self.agent.agentId
            msg.AgentData.SessionID = self.agent.sessionId
            msg.ChatData.Message = message.encode() + b"\0"
            msg.ChatData.Type = ctype
            msg.ChatData.Channel = channel
            self.send(msg)
        
        else:
            # Because ChatFromViewer.Message.Channel is a signed int, we have
            # to use dialogs here
            msg = self.messageTemplate.getMessage("ScriptDialogReply")
            msg.AgentData.AgentID = self.agent.agentId
            msg.AgentData.SessionID = self.agent.sessionId
            msg.Data.ObjectID = self.agent.agentId
            msg.Data.ChatChannel = channel
            msg.Data.ButtonIndex = 0
            msg.Data.ButtonLabel = message.encode() + b"\0"
            self.send(msg)
            
    
    def agentUpdate(self, controls = 0, forward = 0, state = 0, flags = 0):
        angle_rad = math.radians(forward)
        half_angle = angle_rad / 2

        sin_half = math.sin(half_angle)
        cos_half = math.cos(half_angle)
        
        msg = self.messageTemplate.getMessage("AgentUpdate")
        msg.AgentData.AgentID = self.agent.agentId
        msg.AgentData.SessionID = self.agent.sessionId
        msg.AgentData.BodyRotation = (0.0, 0.0, sin_half, cos_half)
        msg.AgentData.HeadRotation = (0.0, 0.0, sin_half, cos_half)
        msg.AgentData.State = state
        msg.AgentData.CameraCenter = (0, 0, 0)
        msg.AgentData.CameraAtAxis = (0, 0.999999, 0)
        msg.AgentData.CameraLeftAxis = (0.999999, 0, 0)
        msg.AgentData.CameraUpAxis = (0, 0, 0.999999)
        msg.AgentData.Far = math.inf
        msg.AgentData.ControlFlags = controls
        msg.AgentData.Flags = flags
        self.send(msg)