#!/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 check_wz(text):
    if not len(text): return True
    for tm in text:
        if ord(tm)<33 or ord(tm)>127: return True
    return None

def get_weather(text):
    wzc = cur_execute_fetchall('select code from wz where code ilike %s or city ilike %s or counry ilike %s',(text,text,text))
    if not wzc: return 'Not Found'
    if len(wzc) != 1: return 'Not Found'
    return load_page('http://weather.noaa.gov/pub/data/observations/metar/decoded/%s.TXT' % wzc[0][0].upper())

def weather(type, jid, nick, text):
    if check_wz(text): msg = L('Error in parameters. Read the help about command.','%s/%s'%(jid,nick))
    else:
        wzz = get_weather(text)
        if 'Not Found' in wzz: msg = L('City not found!','%s/%s'%(jid,nick))
        elif len(wzz.split('\n')) < 9: msg = L('Unexpected error','%s/%s'%(jid,nick))
        else:
            wzz = wzz.split('\n')
            wzr = []
            wzr.append(wzz[0])            # 0
            wzr.append(wzz[1])            # 1
            wzr.append(sfind(wzz,'Temperature'))    # 2
            wzr.append(sfind(wzz,'Wind'))        # 3
            wzr.append(sfind(wzz,'Relative'))    # 4
            wzr.append(sfind(wzz,'Sky'))        # 5
            wzr.append(sfind(wzz,'Weather'))    # 6
            wzr.append(sfind(wzz,'Visibility'))    # 7
            wzr.append(sfind(wzz,'Pressure'))    # 8
            if ')' in wzr[0]: msg = wzr[0][:wzr[0].find(')')+1]
            else: msg = wzr[0]
            msg += '\n'+ wzr[1]
            wzz1 = wzr[2].find(':')+1 # Temperature
            wzz2 = wzr[2].find('(',wzz1)
            wzz3 = wzr[2].find(')',wzz2)
            msg += '\n'+ wzr[2][:wzz1] + ' ' + wzr[2][wzz2+1:wzz3]
            wzz1 = wzr[3].find('(')
            wzz2 = wzr[3].find(')',wzz1)
            wzz3 = wzr[3].find(':',wzz2)
            msg += '\n'+ wzr[3][:wzz1-1] + wzr[3][wzz2+1:wzz3]
            msg += '\n'+ wzr[4]
            if len(wzr[5]): msg += ','+ wzr[5][wzr[5].find(':')+1:]
            if len(wzr[6]): msg += ','+ wzr[6][wzr[6].find(':')+1:]
            if not (len(wzr[5])+len(wzr[6])): msg += ', clear'
            msg += '\n'+ wzr[7][:-2]
            wzz1 = wzr[8].find('(')
            wzz2 = wzr[8].find(':',wzz1)
            wzz3 = wzr[8].find('(',wzz2)
            msg += ', '+ wzr[8][:wzz1-1]+': '+wzr[8][wzz3+1:-1]
    send_msg(type, jid, nick, msg)

def weather_short(type, jid, nick, text):
    if check_wz(text): msg = L('Error in parameters. Read the help about command.','%s/%s'%(jid,nick))
    else:
        wzz = get_weather(text)
        if 'Not Found' in wzz: msg = L('City not found!','%s/%s'%(jid,nick))
        elif len(wzz.split('\n')) < 9: msg = L('Unexpected error','%s/%s'%(jid,nick))
        else:
            wzz = wzz.split('\n')
            wzr = []
            wzr.append(wzz[0])            # 0
            wzr.append(sfind(wzz,'Temperature'))    # 2
            wzr.append(sfind(wzz,'Wind'))        # 3
            wzr.append(sfind(wzz,'Relative'))    # 4
            wzr.append(sfind(wzz,'Sky'))        # 5
            wzr.append(sfind(wzz,'Weather'))    # 6
            if ')' in wzr[0]: msg = wzr[0][:wzr[0].find(')')+1]
            else: msg = wzr[0]
            wzz1 = wzr[1].find(':')+1 # Temperature
            wzz2 = wzr[1].find('(',wzz1)
            wzz3 = wzr[1].find(')',wzz2)
            msg += ' | '+ wzr[1][:wzz1] + ' ' + wzr[1][wzz2+1:wzz3]
            wzz1 = wzr[2].find('(')
            wzz2 = wzr[2].find(')',wzz1)
            wzz3 = wzr[2].find(':',wzz2)
            msg += ' | '+ wzr[2][:wzz1-1] + wzr[2][wzz2+1:wzz3]
            msg += ' | '+ wzr[3]
            if len(wzr[4]): msg += ','+ wzr[4][wzr[4].find(':')+1:]
            if len(wzr[5]): msg += ','+ wzr[5][wzr[5].find(':')+1:]
            if not (len(wzr[4])+len(wzr[5])): msg += ', clear'
    send_msg(type, jid, nick, msg)

def weather_raw(type, jid, nick, text):
    if check_wz(text): msg = L('Error in parameters. Read the help about command.','%s/%s'%(jid,nick))
    else:
        msg = get_weather(text)[:-1]
        if 'Not Found' in msg: msg = L('City not found!','%s/%s'%(jid,nick))
    send_msg(type, jid, nick, msg)

def weather_search(type, jid, nick, text):
    if len(text):
        wzc = cur_execute_fetchall('select code,city,counry from wz where code ilike %s or city ilike %s or counry ilike %s',(text,text,text))
        if not wzc: msg = msg = L('City not found!','%s/%s'%(jid,nick))
        else:
            msg = ''
            for tmp in wzc: msg += '\n%s - %s (%s)' % tmp
            msg = L('Found: %s','%s/%s'%(jid,nick)) % msg
    else: msg = L('What?','%s/%s'%(jid,nick))
    send_msg(type, jid, nick, msg)

global execute

execute = [(3, 'wzz', weather_raw, 2, 'Weather by airport code. Full version.'),
     (3, 'wzs', weather_short, 2, 'Weather by airport code. Short version.'),
     (3, 'wz', weather, 2, 'Weather by airport code. Optimized version.'),
     (3, 'wzsearch', weather_search, 2, 'Search weather by code, city, country.')]