####################################
##Test Bot                        ##
##Credit: Char                    ##
####################################

##This is where you put your imports
import ch
import requests
import os
import sys
import urllib
import time
import traceback
import json
import re
import cgi
import random
import datetime
import binascii

if sys.version_info[0] > 2:
    import urllib.request as urlreq
else:
    import urllib2 as urlreq

############
##Variables
API_KEY = "AIzaSyA98i5h1IXWNAKVl9TAAJgxVhFJgBJoLZc"
prefix = "."
username = input("Username: ")
password = input("Password: ")
rooms = ["yt", "ch", "botquery", "animeotakuchatroom"]



###########################################################
##Startup

class bot(ch.RoomManager):
    def onInit(self):
        self.setNameColor("000000")
        self.setFontColor("000000")
        self.setFontFace("Arial")
        self.setFontSize(11)
        self.enableBg()
        self.enableRecording()

    def onConnect(self, room):
        self.setNameColor("000000")
        self.setFontColor("000000")
        self.setFontFace("Arial")
        self.setFontSize(9)
        self.enableBg()
        self.enableRecording()
        print("[INFO] Connected to "+room.name)

    def onReconnect(self, room):
        self.setNameColor("000000")
        self.setFontColor("000000")
        self.setFontFace("Arial")
        self.setFontSize(9)
        self.enableBg()
        self.enableRecording()
        print("[INFO] Reconnected to : "+room.name)
        room.message("Reconnected")

    def onDisconnect(self, room):
        self.setNameColor("000000")
        self.setFontColor("000000")
        self.setFontFace("Arial")
        self.setFontSize(9)
        self.enableBg()
        self.enableRecording()
        print("[INFO] Disconnected from : "+room.name)


    def onFloodWarning(self, room):
        print("[INFO] FloodWarning in : "+room.name)
        time.sleep(5)
        room.reconnect()



    def onBan(self, room, user, target):
        room.message(target.name.title()+" was banned by "+user.name.title())
        print("[INFO] "+target.name.title()+" was banned by "+user.name.title()+" in "+room.name.title())

    def onUnban(self, room, user, target):
        room.message(target.name.title()+" has been unbanned by "+user.name.title())
        print("[INFO] "+target.name.title()+" was unbanned by "+user.name.title()+" in "+room.name.title())





    def onMessage(self, room, user, message):
            msgdata = message.body.split(" ",1)
            if len(msgdata) > 1:
                cmd, args = msgdata[0], msgdata[1]
            else:
                cmd, args = msgdata[0],""
            cmd=cmd.lower()

            if len(cmd) >0:
                if cmd[0]==prefix:
                    used_prefix = True
                    cmd = cmd[1:]
                else:
                    used_prefix = False
            else:
                return

##########################################################
##Detection

            #Detect Message
            if  not used_prefix:
                print(("[TEXT/MESSAGE]"), ("[")+room.name+("]"), user.name, (":"), message.body, (":"), message.ip)

            #Detect Command
            if used_prefix:
                print(("[COMMAND]"), ("[")+room.name+("]"), user.name, (":"), cmd, (":"), args, (":"), message.ip)

#########################
##Commands


            if used_prefix and cmd=="say":
                if len(args)>0:
                    room.message(user.name.title()+" told me to say: "+args)
                else:
                    room.message("You didn't say anything moron!")


            elif used_prefix and cmd=="yt" and len(args)>0:

                try:
                    string=args.lower()
                    def search(string):
                        apikey = API_KEY
                        api = requests.get("https://www.googleapis.com/youtube/v3/search?part=snippet&maxResults=1&order=relevance&q=%s&safeSearch=none&key=%s" % (string, apikey)).json()
                        id = api['items'][0]['id']['videoId']
                        link = "https://www.youtube.com/watch?v=%s" % id
                        title = api['items'][0]['snippet']['title']
                        auth = api['items'][0]['snippet']['channelTitle']
                        try:
                            return title + " by " + auth + " " + link
                        except:
                            return "fucked your shit up"
                    room.message(search(string))
                except:
                    room.message("You dun fucked up kid")
                    print(traceback.format_exc())




###########################
##Bot Start Code

if __name__ == "__main__":
   bot.easy_start(rooms, username, password)