#!/usr/bin/python
# -*- coding: utf-8 -*-

# --------------------------------------------------------------------------- #
#                                                                             #
#    Plugin for iSida Jabber Bot                                              #
#    Copyright (C) diSabler <dsy@dsy.name>                                    #
#                                                                             #
#    This program is free software: you can redistribute it and/or modify     #
#    it under the terms of the GNU 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 General Public License for more details.                             #
#                                                                             #
#    You should have received a copy of the GNU General Public License        #
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.    #
#                                                                             #
# --------------------------------------------------------------------------- #

def vendor_by_mac(type, jid, nick, text):
	text = text.strip().replace('-','').replace(':','').replace(' ','')[:6].upper()
	if text and len(text) == 6:
		result = html_encode(load_page('http://standards.ieee.org/cgi-bin/ouisearch?%s' % text))
		r = re.findall('\(hex\)(.*?)\n',result,re.S|re.I)
		if 'Sorry!' in result and not r: msg = L('Not found!','%s/%s'%(jid,nick))
		else: msg = r[0].strip().upper()
	else: msg = L('What?','%s/%s'%(jid,nick))
	send_msg(type, jid, nick, msg)

def net_ping(type, jid, nick, text):
	text = text.strip().lower().encode('idna')
	if '.' in text and len(text) > 4 and re.match(r'[-0-9a-z.]+\Z', text, re.U+re.I): msg = deidna(shell_execute('ping -c4 %s' % text,'%s/%s'%(jid,nick)))
	else: msg = L('Smoke help about command!','%s/%s'%(jid,nick))
	send_msg(type, jid, nick, msg)

def get_tld(type, jid, nick, text):
	if len(text) >= 2:
		tld = readfile(tld_list).decode('utf-8')
		tld = tld.split('\n')
		msg = L('Not found!','%s/%s'%(jid,nick))
		for tl in tld:
			if tl.split('\t')[0].lower()==text.lower():
				msg = '.'+tl.replace('\t',' - ',1).replace('\t','\n')
				break
	else: msg = L('What do you want to find?','%s/%s'%(jid,nick))
	send_msg(type, jid, nick, msg)

def is_valid_ip(ipstr):
        try: socket.inet_pton(socket.AF_INET, ipstr)
        except:
                try: socket.inet_pton(socket.AF_INET6, ipstr)
                except: return False
        return True

def get_dns(type, jid, nick, text):
	if is_valid_ip(text):
		try: msg = socket.gethostbyaddr(text)[0]
		except: msg = L('I can\'t resolve it','%s/%s'%(jid,nick))
	else:
		try:
			ans = socket.getaddrinfo(text.encode('idna'), None, 0, 0, socket.IPPROTO_TCP)
			msg = text+' - '
			for an in ans:
                                msg += an[4][0] + ' | '
			msg = msg[:-2]
		except: msg = L('I can\'t resolve it','%s/%s'%(jid,nick))
	send_msg(type, jid, nick, msg)

def srv_nslookup(type, jid, nick, text):
	srv_raw_check(type, jid, nick, 'nslookup '+text)

def srv_dig(type, jid, nick, text):
	srv_raw_check(type, jid, nick, 'dig '+text)

def srv_host(type, jid, nick, text):
	srv_raw_check(type, jid, nick, 'host '+text)

def srv_raw_check(type, jid, nick, text):
	text = enidna_raw(text)
	text = ''.join(re.findall(u'[-a-z0-9\.\_\?\#\=\@\%\ \+]+',text,re.S|re.I)[0])
	send_msg(type, jid, nick, deidna(shell_execute(text,'%s/%s'%(jid,nick))))

def chkserver(type, jid, nick, text):
	for a in ':;&/|\\\n\t\r': text = text.replace(a,' ')
	t = re.findall(u'[-a-zа-я0-9._?#=@%]+',text,re.S|re.I|re.U)
	if len(t) >= 2:
		port = []
		for a in t:
			if a.isdigit(): port.append(a)
		for a in port: t.remove(a)
		if len(t)==1 and len(port)>=1:
			t = t[0]
			port.sort()
			msg = shell_execute('nmap %s -p%s -P0 -T5' % (t.encode('idna'),','.join(port)),'%s/%s'%(jid,nick))
			try: msg = '%s\n%s' % (t.encode('idna'),reduce_spaces_all(re.findall('SERVICE(.*)Nmap',msg,re.S|re.U)[0][1:-2]))
			except:
				try:
					msg = ''
					for a in port:
						sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
						try:
							sock.connect((t.encode('idna'),int(a)))
							s = L('on','%s/%s'%(jid,nick))
						except: s = L('off','%s/%s'%(jid,nick))
						msg += '\n%s %s' % (a,s)
						sock.close()
					msg = '%s%s' % (t,msg)
				except: msg = '%s - %s' % (t,L('unknown','%s/%s'%(jid,nick)))
			msg = L('Port status at %s','%s/%s'%(jid,nick)) % msg
		else: msg = L('What?','%s/%s'%(jid,nick))
	else: msg = L('What?','%s/%s'%(jid,nick))
	send_msg(type, jid, nick, msg)

global execute

execute = [(6, 'nslookup', srv_nslookup, 2, 'Command nslookup'),
		   (6, 'host', srv_host, 2, 'Command host'),
		   (6, 'dig', srv_dig, 2, 'Command dig'),
		   (4, 'port', chkserver, 2, 'Check port activity\nport server port1 [port2 ...]'),
		   (4, 'net_ping', net_ping, 2, 'Net Ping.\nnet_ping ip|domain'),
		   (3, 'dns', get_dns, 2, 'DNS resolver.'),
		   (3, 'tld', get_tld, 2, 'Search domain zones TLD.'),
		   (3, 'mac', vendor_by_mac, 2, 'Show vendor of device by mac')]