import olex
import olx
import os
import glob
import sys
from olexFunctions import OV
from ImageTools import IT
import htmlTools
from . import restraints

table_col = OV.GetVar('HtmlTableFirstcolColour')
green_text = OV.GetParam('gui.green_text')
green = OV.GetParam('gui.green')
red = OV.GetParam('gui.red')
orange = OV.GetParam('gui.orange')
white = "#ffffff"
black = "#000000"

def FileOpen(title, filter, location, default='', default_name=''):
  res = olx.FileOpen(title, filter,location, default_name)
  if not res:
    return default
  return res

def FileSave(title, filter, location, default='', default_name=''):
  res = olx.FileSave(title, filter,location, default_name)
  if not res:
    return default
  return res

def SaveModel():
  fn = FileSave("Please choose the file to save to", "Olex2 model files|*.oxm", olx.FilePath())
  if fn:
    olx.Save("model", fn)

olex.registerFunction(FileOpen, False, "gui.dialog")
olex.registerFunction(FileSave, False, "gui.dialog")
olex.registerFunction(SaveModel, False, "gui.dialog")

def About():
  sw = 650+2*10+2
  sh = 400+2*15+60
  x,y = GetBoxPosition(sw, sh)
  olx.Popup("about", olx.BaseDir() + "/etc/gui/help/about.htm",
            x=x, y=y, w=sw, h=sh)

olex.registerFunction(About, False, "gui")

def SwitchSettings(name="solve"):
  name = name.lower()
  auto_close_settings = OV.GetParam('user.auto_close_settings_panel')
  if not auto_close_settings:
    t = "cbtn* 1 cbtn-%s 1 *settings 0 %s-settings 0 1" %(name, name)
  else:
    t = "cbtn* 1 cbtn-%s 1 *settings 0" %name
  olx.html.ItemState(t)

olex.registerFunction(SwitchSettings, False, "gui")


def SwitchPanel(name="home"):
  name = name.lower()
  if name == "home":
    OV.setItemstate("* 0 tab* 2 tab-home 1 logo1 1 index-home* 1 info-title 1")
    #olx.html.ItemState("index* 0 index-home 1 tab* 2")
  elif name == "work":
    if olx.IsFileLoaded() != "false":
      OV.setItemstate("* 0 tab* 2 tab-work 1 logo1 1 index-work* 1 info-title 1")
  elif name == "view":
    OV.setItemstate("* 0 tab* 2 tab-view 1 logo1 1 index-view* 1 info-title 1")
  elif name == "tools":
    OV.setItemstate("* 0 tab* 2 tab-tools 1 logo1 1 index-tools* 1 info-title 1")
  elif name == "info":
    OV.setItemstate("* 0 tab* 2 tab-info 1 logo1 1 index-info* 1 info-title 1")
  else:
    print("Invalid argument for the panel name: " + name)
  return ""

olex.registerFunction(SwitchPanel, False, "gui")


def SwitchTool(name=None):
  #e.g h2-tools-images
  l = name.split("-")
  SwitchPanel(l[1])
  OV.setItemstate("%s 1" %name)
olex.registerFunction(SwitchTool, False, "gui")

def PopTool(name=None):
  #e.g h2-tools-images
  pop_name = name
  wFilePath = os.path.join(OV.BaseDir(), "etc", name)
  title = name
  olx.Popup(pop_name, wFilePath,
    b="tcr", t=title)
olex.registerFunction(PopTool, False, "gui")

def UpdateWeight():
  w = OV.GetParam('snum.refinement.suggested_weight')
  if not w:
    print("No suggested weighting scheme present. Please refine and try again.")
    return ""
  olex.m("UpdateWght %s %s" %(w[0], w[1]))
  print("Weighting scheme has been updated")
olex.registerFunction(UpdateWeight, False, "gui")

def GetPathParam(variable, default=None):
  retVal = OV.GetParam(variable, default)
  if not retVal:
    return retVal
  if "()" in retVal:
    func = retVal.split("()")[0]
    rest = retVal.split("()")[1]
    res = getattr(OlexFunctions, func)
    retVal = res(OV) + rest
  return OV.standardizePath(retVal).replace("\\\\", "\\")
olex.registerFunction(GetPathParam, False, "gui")

def GetItemsFromPhilChoices(param):
  _ = OV.GetChoices(param)
  t = ""
  for choice in _:
    t += choice.lstrip('*') + ";"
  return t
OV.registerFunction(GetItemsFromPhilChoices, False, 'gui')

def GetFileList(root, extensions):
  import ntpath
  l = []
  if type(extensions) == str:
    extensions = extensions.split(";")
  for extension in extensions:
    extension = extension.strip("'")
    g = glob.glob(r"%s%s*.%s" %(root, os.sep, extension))
    for f in g:
      f = OV.standardizePath(f)
      l.append((ntpath.basename(f),f))
  return l
olex.registerFunction(GetFileList, False, "gui")

def GetFileListAsDropdownItems(root, extensions):
  l = GetFileList(root, extensions)
  txt = ""
  for item in l:
    txt += "%s<-%s;" %(item[0], item[1])
  return txt
olex.registerFunction(GetFileListAsDropdownItems, False, "gui")

def GetDropdownItemsFromList(l):
  txt = ""
  for item in l:
    if len(item) == 1:
      txt += "%s;" %(item)
    elif len(item) == 2:
      txt += "%s<-%s;" %(item[0], item[1])
  return txt
olex.registerFunction(GetDropdownItemsFromList, False, "gui")

def GetFolderList(root):
  import os
  t = ""
  assert root
  root_c = olex.f(root)
  t = []
  for root, dirs, files in os.walk(root_c, topdown=True):
    pre = root[len(root_c):].replace('\\', '/').lstrip('/')
    for dir in dirs:
      if pre: dir = "%s/%s" %(pre, dir)
      t.append(dir)
  t.sort()
  t = ";".join(t)
  return t
olex.registerFunction(GetFolderList, False, "gui")

def GetBoxPosition(w, h):
  sz = [int(x) for x in olx.GetWindowSize().split(',')]
  y = sz[1] + (sz[3]-h)//2
  x = sz[0] + (sz[2]-w)//2
  return x,y

#'static' class
class ImageListener_:
  listeners = []

  def Register(self, listener):
    ImageListener.listeners.append(listener)

  def Unregister(self, listener):
    ImageListener.listeners.remove(listener)

  def OnChange(self):
    for i in ImageListener.listeners:
      i()

ImageListener = ImageListener_()


def IsMouseLocked(which=None,state=None):
  if not which:
    return

  l = ["translation", "rotation", "zooming"]
  l.remove(which)

  all_locked = True
  for what in l:
    _ = olx.mouse.IsEnabled(what)
    if _ == "true":
      all_locked = False
      break
  if all_locked:
    all_locked = (state=='true')

  if state == "false":
    olx.mouse.Enable(which)
  else:
    olx.mouse.Disable(which)

  return all_locked
olex.registerFunction(IsMouseLocked, False, "gui")


def do_sort():
  args = []
  args.append('+%s%s%s%s' %(OV.GetParam("user.sorting.cat1", ''),
    OV.GetParam("user.sorting.cat2", ''),
    OV.GetParam("user.sorting.cat3", ''),
    OV.GetParam("user.sorting.cat4", '')))
  try:
    if olx.html.GetState('atom_sequence_inplace') == 'true':
      args[0] += 'w'
    elif olx.html.GetState('atom_sequence_first') == 'true':
      args[0] += 'f'
  except:
    pass
  if OV.GetParam("user.sorting.h", False):
    args[0] += 'h'
  args += olx.GetVar("sorting.atom_order", "").split()
  arg3 = OV.GetParam("user.sorting.moiety")
  if arg3 is not None:
    args.append("moiety")
    if arg3 != 'default':
      args.append('+' + arg3)
  args += olx.GetVar("sorting.moiety_order", "").split()
  olx.Sort(*args)
olex.registerFunction(do_sort, False, "gui")


def copytree(src, dst, symlinks=False, ignore=None):
  ## From https://groups.google.com/forum/embed/#!topic/comp.lang.python/8MpGFEhFCm0
  import os
  from shutil import copy2, copystat, Error

  names = os.listdir(src)
  if ignore is not None:
      ignored_names = ignore(src, names)
  else:
      ignored_names = set()

  try:
      os.makedirs(dst)
  except OSError as exc:
      # XXX - this is pretty ugly
      if "file already exists" in exc[1]:  # Windows
          pass
      elif "File exists" in exc[1]:        # Linux
          pass
      else:
          raise

  errors = []
  for name in names:
      if name in ignored_names:
          continue
      srcname = os.path.join(src, name)
      dstname = os.path.join(dst, name)
      try:
          if symlinks and os.path.islink(srcname):
              linkto = os.readlink(srcname)
              os.symlink(linkto, dstname)
          elif os.path.isdir(srcname):
              copytree(srcname, dstname, symlinks, ignore)
          else:
              copy2(srcname, dstname)
          # XXX What about devices, sockets etc.?
      except (IOError, os.error) as why:
          errors.append((srcname, dstname, str(why)))
      # catch the Error from the recursive copytree so that we can
      # continue with other files
      except Error as err:
          errors.extend(err.args[0])
  try:
      copystat(src, dst)
  except WindowsError:
      # can't copy file access times on Windows
      pass
  except OSError as why:
      errors.extend((src, dst, str(why)))
  if errors:
      raise Error(errors)

def copy_datadir_items(force=False):
  '''
  This will copy the directories containg the shipped samples as well as a directory called 'customisation' to the DataDir(). It will only copy those sub-directories that are NOT present already. When force=True, all directories will be *merged*. Existing files will not be overwritten.
  This function can be called from the command line with spy.gui.copy_datadir_items(TRUE/FALSE) and is called on every startup of Olex2 from InitPy (with FALSE).
  '''
  import shutil

  ignore_patterns = shutil.ignore_patterns('*.svn')

  svn_samples_directory = '%s%ssample_data' %(OV.BaseDir(),os.sep)
  user_samples_directory = OV.GetParam('user.sample_dir')
  if not user_samples_directory:
    user_samples_directory = '%s%ssamples' %(OV.DataDir(),os.sep)
    OV.SetParam('user.sample_dir', user_samples_directory)

  svn_customisation_directory = '%s%setc%scustomisation' %(OV.BaseDir(),os.sep,os.sep)
  user_customisation_directory = OV.GetParam('user.customisation_dir')
  if not user_customisation_directory:
    user_customisation_directory = '%s%scustomisation' %(OV.DataDir(),os.sep)

  dirs = ((svn_samples_directory, user_samples_directory), )
  for src, dest in dirs:
    if not os.path.exists(dest):
      os.makedirs(dest)
      if "sample_data" in src:
        OV.SetVar('sample_dir', dest)
      elif "customisation" in src:
        OV.SetParam('user.customisation_dir',dest)
    else:
      if "sample_data" in src:
        OV.SetVar('sample_dir', dest)
      elif "customisation" in src:
        OV.SetParam('user.customisation_dir',dest)
    if not os.path.exists(src):
      print("Sorry - installation does not have samples")
      return
    things = os.listdir(src)
    for thing in things:
      if thing == '.svn': continue
      try:
        from_f = '%s%s%s' %(src,os.sep,thing)
        to_f = '%s%s%s' %(dest,os.sep,thing)
        if not force and os.path.exists(to_f):
          continue
        copytree(from_f, to_f, ignore=ignore_patterns)
      except Exception as err:
        print(err)

olex.registerFunction(copy_datadir_items, False, "gui")

def focus_on_control():
  highlight = OV.GetVar('HtmlHighlightColour')
  control = OV.GetVar('set_focus_on_control_please',None)
  if control == "None": control = None
  if control:
    print("focus on %s" %control)
    if OV.IsControl(control):
      olx.html.SetBG(control,highlight)
      olx.html.SetFocus(control)
      OV.SetVar('set_focus_on_control_please','None')
      return
  olx.Focus()
olex.registerFunction(focus_on_control, False, "gui")

def escape_for_html(s):
  s = s.replace("(", "&#40;")
  s = s.replace(")", "&#41;")
  return s
olex.registerFunction(escape_for_html, False, "gui")

def helpImagesToOlexVFS(p, base='images'):
  import glob
  import OlexVFS
  file_l =  glob.glob(f"{p}/*.[jJpP][pPnN][gG]")
  for item in file_l:
    with open(item, 'rb') as fc:
      fc = fc.read()
      fn =  f"{base}/{os.path.basename(item)}"
      OlexVFS.write_to_olex(fn, fc)

def zipToOlexVFS(zip_path, base=None):
  import OlexVFS
  import zipfile
  zf = zipfile.ZipFile(zip_path)
  files = []
  for item in zf.filelist:
    filename = item.filename
    ext = filename.split('.')[-1:][0]
    fc = zf.open(filename,'r').read()
    base_full = filename.split("/")[-1:][0]
    base = base_full.split('.')[0]
    #templates[base] = fc
    OlexVFS.write_to_olex(filename, fc)
olex.registerFunction(zipToOlexVFS, False, "tools")

def get_default_notification(txt="", txt_col='green_text'):
  txt_col = OV.GetVar("gui.%s" %txt_col, '#ff0000')
  poly = ""
  _ = olx.xf.latt.IsPolymeric()
  if _ == 'true':
    poly = " | Polymeric structure"
  set_notification("<font color='%s'>%s</font>%s;%s;%s" %(txt_col, txt, poly, table_col,'#888888'))

#https://stackoverflow.com/questions/753052/strip-html-from-strings-in-python
def strip_html_tags(html):
  from io import StringIO
  from html.parser import HTMLParser

  class MLStripper(HTMLParser):
    def __init__(self):
      super().__init__()
      self.reset()
      self.strict = False
      self.convert_charrefs= True
      self.text = StringIO()
    def handle_data(self, d):
      self.text.write(d)
    def get_data(self):
      return self.text.getvalue()
  s = MLStripper()
  s.feed(html)
  return s.get_data()

def set_notification(string):
  if not OV.HasGUI():
    olx.Echo(strip_html_tags(string.split(';')[0]), m="error")
    with open(os.path.join(OV.StrDir(), "refinement.check"), "w") as out:
      out.write(string)
    return
  if string is None:
    get_default_notification()
  OV.SetVar('GuiNotification', string)
  olx.Freeze(True)
  OV.htmlUpdate()
  olx.Freeze(False)

def get_notification():
  _ = OV.GetVar('GuiNotification',None)
  d = {}
  if not _:
    return
  col_bg = OV.GetVar('gui.dark_yellow')
  col_fg = "#000000"
  txt = _
  if ";" in _:
    _ = _.split(";")
    txt = _[0]
    col_bg = _[1]
    if len(_) > 2:
      col_fg = _[2]

  d = {'col_bg':col_bg,
       'col_fg':col_fg,
       'txt':txt}
  import gui
  notification = gui.tools.TemplateProvider.get_template('gui_notification')%d
  return notification

olex.registerFunction(get_default_notification, False, "gui")
olex.registerFunction(set_notification, False, "gui")
olex.registerFunction(get_notification, False, "gui")

def file_open(path, base="", mode='r', readlines=False, binary=False, encoding=None):
  ''' Open a file, either from a real file, or, if that is not found, from the OlexVFS.
      -- path: the FULL path to the file
      -- base: the everything up to where the directory lives
  '''
  import OlexVFS
  retVal = None
  if os.path.exists(path):
    if readlines:
      retVal = open(path, mode, encoding=encoding).readlines()
    else:
      retVal = open(path, mode, encoding=encoding).read()
  else:
    paths = []
    if base:
      paths = [path, base]
    common_prefix = os.path.commonprefix(paths)
    if common_prefix in paths:
      path = [os.path.relpath(path, common_prefix) for path in paths]
      path = path[0].replace("\\","/")
    try:
      path = path.replace("\\","/")
      retVal = OlexVFS.read_from_olex(path)
      if not binary:
        try:
          retVal = retVal.decode()
        except:
          pass
    except:
      print("gui.file_open malfunctioned with getting %s" %path)

    if readlines:
      if retVal is None:
        print("Failed to fetch %s" %path)
        return ""
      retVal = retVal.splitlines()
  return retVal

olex.registerFunction(file_open, False, "tools")

def NamingMode():
  if not OV.IsControl("naming*start"):
    print("Available from the GUI only!")
    return
  args = []
  start = olx.html.GetValue("naming*start", "")
  if start:
    args.append(start)
  suffix = olx.html.GetValue("naming*suffix", "")
  elm = olx.html.GetValue("naming*type", "")
  auto = 0
  if olx.html.GetState("naming*auto*on", "false") == "true":
    auto = 1
    if olx.html.GetState("naming*auto*stop_type", "false") == "true":
      auto += 2
    if olx.html.GetState("naming*auto*stop_part", "false") == "true":
      auto += 4
    if olx.html.GetState("naming*auto*stop_branch", "true") == "false":
      auto += 8
  olx.Mode("Name", *args, a=auto, t=elm, s=suffix)

olex.registerFunction(NamingMode, False, "gui")

def FixFree(target):
  from gui.tools import GetNParams
  target = target.upper()
  if target == "FIX ALL":
    olx.Run("sel -u>>sel $*>>fix xyz,adp,occu>>fix HUiso")
  elif target == "FREE XYZ":
    olx.Run("sel -u>>sel $*,H>>free xyz")
  elif target == "FREE ADP":
    olx.Run("sel -u>>sel $*,H>>free adp")
  elif target == "FREE H XYZ":
    olx.Run("sel -u>>sel $H>>free xyz -cs>>afix 0")
  elif target == "FREE H UISO":
    olx.Run("sel -u>>sel $H>>free Uiso")
  elif target == "AFIX":
    olx.Run("kill $H>>HAdd")
  olx.Labels(f=True, h=True, a=True, r=True)
  GetNParams()

olex.registerFunction(FixFree, False, "gui")

def get_refinement_lists():
  if OV.IsEDRefinement():
    return "None<-n/a;Other<-other;4: Fc_sq, Fo_sq, sig_Fo_sq<-4"
  return "None<-n/a;Other<-other;3: Fo, sig_Fo, A_re and B_im<-3;4: Fc_sq, Fo_sq, sig_Fo_sq<-4;6: Fo_sq, sig_Fo_sq, Fc, phase<-6"

def set_refinement_list(val):
  if "other" == val:
    return
  if "n/a" == val:
    olx.DelIns("list")
  else:
    olx.DelIns("list")
    olx.AddIns("list", val)

olex.registerFunction(get_refinement_lists, False, "gui")
olex.registerFunction(set_refinement_list, False, "gui")

def set_client_mode(v):
  v = str(v).lower()
  if v == "true":
    v = True
  elif v == "false":
    v = False
  else:
    return
  OV.SetParam("user.refinement.client_mode", v)
  if OV.IsControl("cpus_label@refine"):
    olx.html.SetLabel("cpus_label@refine", get_thread_n_label())
    olx.html.SetItems("cpus@refine", get_thread_n_selection())
    cpu_n = "-1%" if v else "-1"
    olx.html.SetValue("cpus@refine", cpu_n)
    OV.SetParam("user.refinement.thread_n", cpu_n)

olex.registerFunction(set_client_mode, False, "gui")

def get_thread_n_selection():
  if OV.IsClientMode():
    return "Def<--1%;100<-100%;75<-75%;50<-50%;25<-25%;1"
  else:
    rv_l = []
    cpu_n = os.cpu_count()
    while cpu_n > 1:
      if cpu_n % 2 == 0:
        rv_l.append(int(cpu_n))
      cpu_n -= 2
    rv_l.sort()
    rv_l = ["Def<--1;1"] + [str(item) for item in rv_l]
    return ";".join(rv_l)

olex.registerFunction(get_thread_n_selection, False, "gui")

def get_thread_n_label():
  return "Threads " + ('%' if OV.IsClientMode() else '#')

olex.registerFunction(get_thread_n_label, False, "gui")

def get_openblas_thread_n_selection():
  try:
    max_ob_th = int(olx.app.OptValue("openblas.thread_n_max", 24))
    th_n = int(olx.app.OptValue('openblas.thread_n', "-1"))
    max_ob_th_actual = 0
    try:
      from fast_linalg import env
      if env.initialised:
        config = str(env.build_config)
        max_ob_th_actual = int(config.split("MAX_THREADS=")[1].split()[0])
    except:
      pass
    # update max if needed
    if max_ob_th_actual > 0 and max_ob_th_actual != max_ob_th:
      olx.app.SetOption("openblas.thread_n_max", max_ob_th_actual)
      olx.app.SaveOptions()
      max_ob_th = max_ob_th_actual
    max_ob_th = min(os.cpu_count(), max_ob_th)

    rv_l_ = [max_ob_th]
    for fract in (3./4, 2./3, 1./2, 1./3, 1./4, 1./max_ob_th):
      v  = int(max_ob_th * fract)
      if v in rv_l_: break
      rv_l_.append(v)
    rv_l = ["Def<--1"]
    if th_n > 1 and th_n not in rv_l_:
      rv_l.append(str(th_n))
    rv_l += [str(item) for item in rv_l_]
    return ";".join(rv_l)
  except Exception as e:
    print(e)
    return "1"

olex.registerFunction(get_openblas_thread_n_selection, False, "gui")

def set_openblas_thread_n(val):
  if not val:
    return
  olx.app.SetOption('openblas.thread_n', val)
  olx.app.SaveOptions()
  try:
    import fast_linalg
    val = int(val)
    if val < 1:
      val = int(min(os.cpu_count(), int(olx.app.OptValue("openblas.thread_n_max", 24))) * 2 /3)
    if fast_linalg.env.initialised:
      fast_linalg.env.threads = int(val)
  except:
    olx.Echo("Failed to set OpenBlas thread number", m="error")

olex.registerFunction(set_openblas_thread_n, False, "gui")

def create_history_branch(switch):
  switch = OV.get_bool_from_any(switch)
  name = OV.describe_refinement()
  branch_name = OV.GetUserInput(1, "Branch name", name)
  if not branch_name:
    return
  from History import hist, tree
  current_node = tree.active_node
  hist.create_history(True, branch_name)
  if not switch:
    tree.active_node = current_node
  OV.htmlUpdate()

olex.registerFunction(create_history_branch, False, "gui")

def create_archive(node_id=None):
  from datetime import datetime
  from History import hist
  now = datetime.now()
  description = hist.describe_node(node_id) if node_id else OV.describe_refinement()
  name = now.strftime("%Y-%m-%d_%H_%M-") + description + ".zip"
  file_name = olx.FileSave("Archive name", "*.zip", olx.FilePath(), name)
  #file_name = OV.GetUserInput(1, "File name", name)
  if not file_name:
    return
  hist.create_archive(name, node_id)
  print("%s data archive has been created" %name)

olex.registerFunction(create_archive, False, "gui")