#!/usr/bin/env python
# File: record_dialog.py

#    Copyright (C) 2013 Christian Jacobs.

#    This file is part of PyQSO.

#    PyQSO 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.
#
#    PyQSO 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 PyQSO.  If not, see <http://www.gnu.org/licenses/>.

from gi.repository import Gtk, GObject
import logging
import re
import calendar

from callsign_lookup import *

class RecordDialog(Gtk.Dialog):
   
   def __init__(self, root_window, log, index=None):
      logging.debug("New RecordDialog instance created!")
      
      if(index is not None):
         title = "Edit Record %d" % index
      else:
         title = "Add Record"
      Gtk.Dialog.__init__(self, title=title, parent=root_window, flags=Gtk.DialogFlags.DESTROY_WITH_PARENT, buttons=(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OK, Gtk.ResponseType.OK))

      frame = Gtk.Frame()
      label = Gtk.Label("QSO Data")
      frame.set_label_widget(label)
      self.vbox.add(frame)
      vbox_inner = Gtk.VBox(spacing=2)

      # Create label:entry pairs and store them in a dictionary
      self.sources = {}

      # CALL
      hbox_temp = Gtk.HBox(spacing=0)
      label = Gtk.Label(log.SELECTED_FIELD_NAMES_FRIENDLY["CALL"])
      label.set_alignment(0, 0.5)
      hbox_temp.pack_start(label, False, False, 6)
      self.sources["CALL"] = Gtk.Entry()
      hbox_temp.pack_start(self.sources["CALL"], True, True, 6)
      lookup = Gtk.Button("Lookup") # Looks up the callsign on qrz.com for more details.
      lookup.connect("clicked", self.lookup_callback)
      hbox_temp.pack_start(lookup, True, True, 6)
      vbox_inner.pack_start(hbox_temp, False, False, 6)

      # DATE
      hbox_temp = Gtk.HBox(spacing=0)
      label = Gtk.Label(log.SELECTED_FIELD_NAMES_FRIENDLY["DATE"])
      label.set_alignment(0, 0.5)
      hbox_temp.pack_start(label, False, False, 6)
      self.sources["DATE"] = Gtk.Entry()
      hbox_temp.pack_start(self.sources["DATE"], True, True, 6)

      # TIME
      label = Gtk.Label(log.SELECTED_FIELD_NAMES_FRIENDLY["TIME"])
      label.set_alignment(0, 0.5)
      hbox_temp.pack_start(label, False, False, 6)
      self.sources["TIME"] = Gtk.Entry()
      hbox_temp.pack_start(self.sources["TIME"], True, True, 6)
      vbox_inner.pack_start(hbox_temp, False, False, 6)

      # FREQ
      hbox_temp = Gtk.HBox(spacing=0)
      label = Gtk.Label(log.SELECTED_FIELD_NAMES_FRIENDLY["FREQ"])
      label.set_alignment(0, 0.5)
      hbox_temp.pack_start(label, False, False, 6)
      self.sources["FREQ"] = Gtk.Entry()
      hbox_temp.pack_start(self.sources["FREQ"], True, True, 6)

      # BAND
      label = Gtk.Label(log.SELECTED_FIELD_NAMES_FRIENDLY["BAND"])
      label.set_alignment(0, 0.5)
      hbox_temp.pack_start(label, False, False, 6)
      bands = ["", "2190m", "560m", "160m", "80m", "60m", "40m", "30m", "20m", "17m", "15m", "12m", "10m", "6m", "4m", "2m", "1.25m", "70cm", "33cm", "23cm", "13cm", "9cm", "6cm", "3cm", "1.25cm", "6mm", "4mm", "2.5mm", "2mm", "1mm"]
      self.sources["BAND"] = Gtk.ComboBoxText()
      for band in bands:
         self.sources["BAND"].append_text(band)
      self.sources["BAND"].set_active(0) # Set an empty string as the default option.
      hbox_temp.pack_start(self.sources["BAND"], True, True, 6)
      vbox_inner.pack_start(hbox_temp, False, False, 6)

      # MODE
      hbox_temp = Gtk.HBox(spacing=0)
      label = Gtk.Label(log.SELECTED_FIELD_NAMES_FRIENDLY["MODE"])
      label.set_alignment(0, 0.5)
      hbox_temp.pack_start(label, False, False, 6)
      modes = ["", "AM", "AMTORFEC", "ASCI", "ATV", "CHIP64", "CHIP128", "CLO", "CONTESTI", "CW", "DSTAR", "DOMINO", "DOMINOF", "FAX", "FM", "FMHELL", "FSK31", "FSK441", "GTOR", "HELL", "HELL80", "HFSK", "ISCAT", "JT44", "JT4A", "JT4B", "JT4C", "JT4D", "JT4E", "JT4F", "JT4G", "JT65", "JT65A", "JT65B", "JT65C", "JT6M", "MFSK8", "MFSK16", "MT63", "OLIVIA", "PAC", "PAC2", "PAC3", "PAX", "PAX2", "PCW", "PKT", "PSK10", "PSK31", "PSK63", "PSK63F", "PSK125", "PSKAM10", "PSKAM31", "PSKAM50", "PSKFEC31", "PSKHELL", "Q15", "QPSK31", "QPSK63", "QPSK125", "ROS", "RTTY", "RTTYM", "SSB", "SSTV", "THRB", "THOR", "THRBX", "TOR", "V4", "VOI", "WINMOR", "WSPR"]
      self.sources["MODE"] = Gtk.ComboBoxText()
      for mode in modes:
         self.sources["MODE"].append_text(mode)
      self.sources["MODE"].set_active(0) # Set an empty string as the default option.
      hbox_temp.pack_start(self.sources["MODE"], True, True, 6)
      vbox_inner.pack_start(hbox_temp, False, False, 6)

      # RST_SENT
      hbox_temp = Gtk.HBox(spacing=0)
      label = Gtk.Label(log.SELECTED_FIELD_NAMES_FRIENDLY["RST_SENT"])
      label.set_alignment(0, 0.5)
      hbox_temp.pack_start(label, False, False, 6)
      self.sources["RST_SENT"] = Gtk.Entry()
      hbox_temp.pack_start(self.sources["RST_SENT"], True, True, 6)

      # RST_RCVD
      label = Gtk.Label(log.SELECTED_FIELD_NAMES_FRIENDLY["RST_RCVD"])
      label.set_alignment(0, 0.5)
      hbox_temp.pack_start(label, False, False, 6)
      self.sources["RST_RCVD"] = Gtk.Entry()
      hbox_temp.pack_start(self.sources["RST_RCVD"], True, True, 6)
      vbox_inner.pack_start(hbox_temp, False, False, 6)

      if(index is not None):
         # The record already exists, so display its current data in the input boxes.
         record = log.get_record(index)
         field_names = log.SELECTED_FIELD_NAMES_ORDERED
         for i in range(0, len(field_names)):
            if(field_names[i] == "BAND"):
               self.sources[field_names[i]].set_active(bands.index(record.get_data(field_names[i])))
            elif(field_names[i] == "MODE"):
               self.sources[field_names[i]].set_active(modes.index(record.get_data(field_names[i])))
            else:
               self.sources[field_names[i]].set_text(record.get_data(field_names[i]))

      frame.add(vbox_inner)

      self.show_all()

      return

   def get_data(self, field_name):
      if(field_name == "BAND" or field_name == "MODE"):
         return self.sources[field_name].get_active_text()
      else:
         return self.sources[field_name].get_text()

   def is_valid(self, log, field_name, data):
      
      # Allow an empty string, in case the user doesn't want
      # to fill in this field.
      if(data == ""):
         return True

      # Validate the fields with respect to the ADIF specification.
      if(log.SELECTED_FIELD_NAMES_TYPES[field_name] == "N"):
         # Allow a decimal point before and/or after any numbers,
         # but don't allow a decimal point on its own.
         m = re.match("-?(([0-9]+\.?[0-9]*)|([0-9]*\.?[0-9]+))", data)
         if(m is None):
            # Did not match anything.
            return False
         else:
            # Make sure we match the whole string,
            # otherwise there may be an invalid character after the match. 
            return (m.group(0) == data)
      
      elif(log.SELECTED_FIELD_NAMES_TYPES[field_name] == "B"):
         # Boolean
         m = re.match("(Y|N)", data)
         if(m is None):
            return False
         else:
            return (m.group(0) == data)

      elif(log.SELECTED_FIELD_NAMES_TYPES[field_name] == "D"):
         # Date
         pattern = re.compile("([0-9]){4}")
         m_year = pattern.match(data, 0)
         if((m_year is None) or (int(m_year.group(0)) < 1930)):
            # Did not match anything.
            return False
         else:
            pattern = re.compile("([0-9]){2}")
            m_month = pattern.match(data, 4)
            if((m_month is None) or int(m_month.group(0)) > 12 or int(m_month.group(0)) < 1):
               # Did not match anything.
               return False
            else:
               pattern = re.compile("([0-9]){2}")
               m_day = pattern.match(data, 6)
               days_in_month = calendar.monthrange(int(m_year.group(0)), int(m_month.group(0)))
               if((m_day is None) or int(m_day.group(0)) > days_in_month[1] or int(m_day.group(0)) < 1):
                  # Did not match anything.
                  return False
               else:
                  # Make sure we match the whole string,
                  # otherwise there may be an invalid character after the match. 
                  return (len(data) == 8)

      elif(log.SELECTED_FIELD_NAMES_TYPES[field_name] == "T"):
         # Time
         pattern = re.compile("([0-9]){2}")
         m_hour = pattern.match(data, 0)
         if((m_hour is None) or (int(m_hour.group(0)) < 0) or (int(m_hour.group(0)) > 23)):
            # Did not match anything.
            return False
         else:
            pattern = re.compile("([0-9]){2}")
            m_minutes = pattern.match(data, 2)
            if((m_minutes is None) or int(m_minutes.group(0)) < 0 or int(m_minutes.group(0)) > 59):
               # Did not match anything.
               return False
            else:
               if(len(data) == 4):
                  # HHMM format
                  return True
               pattern = re.compile("([0-9]){2}")
               m_seconds = pattern.match(data, 4)
               if((m_seconds is None) or int(m_seconds.group(0)) < 0 or int(m_seconds.group(0)) > 59):
                  # Did not match anything.
                  return False
               else:
                  # Make sure we match the whole string,
                  # otherwise there may be an invalid character after the match. 
                  return (len(data) == 6) # HHMMSS format

      elif(log.SELECTED_FIELD_NAMES_TYPES[field_name] == "E"):
         # Enumeration.
         # We'll assume that this data is valid already,
         # since the user can only select from a pre-defined (valid) list.
         return True

      else:
         return True

   def lookup_callback(self, widget):
      # TODO: If a session doesn't already exist: Show a username and password dialog, and initiate a session.
      # Get the callsign-related data from the qrz.com database.
      return