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"]



###########################################################
##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)