# coding=utf-8
setThrowException(True)

class Konqueror:

   REMOTE = 0
   LOCAL = 1

   _tab = None

   _dirs = {}

   def __init__(self):
      self._dirs = {self.LOCAL: [], self.REMOTE: []}
      self._startKonqueror()
      self._initWebdav()
      sleep(2)
      self._initLocal()
      sleep(2)

   def _startKonqueror(self):
      openApp("/home/dotxp/bin/start_konqueror.sh")
      sleep(1)
      wait(Pattern().similar(0.80).firstN(1))

   def _initWebdav(self):
      type("l", KEY_CTRL)
      paste("webdav://webdav")
      type(Key.ENTER)
      self._tab = self.REMOTE
      self._dirs[self.REMOTE].append("/")

   def _initLocal(self):
      type("t", KEY_CTRL)
      sleep(1)
      self._tab = self.LOCAL
      type("l", KEY_CTRL)
      paste("/home/dotxp/Desktop/Temp/put_test/down")
      type(Key.ENTER)
      self._dirs[self.LOCAL].append("/")
      self._dirs[self.LOCAL].append("down")

   def switchRemote(self):
      if (self._tab == self.LOCAL):
         type(",", KEY_CTRL)
         self._tab = self.REMOTE
         sleep(0.4)

   def switchLocal(self):
      if (self._tab == self.REMOTE):
         type(",", KEY_CTRL)
         self._tab = self.LOCAL
         sleep(0.4)

   def openLocal(self, dirImg, dirName):
      self.switchLocal()
      self._open(dirImg, dirName)

   def _open(self, dirImg, dirName):
      doubleClick(dirImg)
      self._dirs[self._tab].append(dirName)

   def openRemote(self, dirImg, dirName):
      self.switchRemote()
      self._open(dirImg, dirName)

   def upRemote(self, dirName=None):
      self.switchRemote()
      sleep(1)
      self.goUp(dirName)

   def upLocal(self, dirName=None):
      self.switchLocal()
      sleep(1)
      self.goUp(dirName)

   def goUp(self, dirName):
      if dirName == None:
         click(Pattern().similar(0.60).firstN(1))
         self._dirs[self._tab].pop()
         return
      while self._dirs[self._tab][-1] != dirName:
         click(Pattern().similar(0.60).firstN(1))
         self._dirs[self._tab].pop()

   def copy(self):
      type("c", KEY_CTRL)

   def paste(self):
      type("v", KEY_CTRL)

   def rename(self, fileImg, newName, newFileImg):
      click(fileImg)
      sleep(0.2)
      type(Key.F2)
      wait(Pattern().similar(0.50).firstN(1))

      type("a", KEY_CTRL)
      sleep(0.2)
      paste(newName)
      sleep(0.1)
      type(Key.ENTER)
      wait(newFileImg)

   def createDir(self, dirName, dirImg):
      type(Key.F10)
      createDirDiag = wait(Pattern().similar(0.80).firstN(1))
      if len(createDirDiag) == 0:
         raise SystemExit("Could not find new dir dialog")
      paste(dirName)
      type(Key.ENTER)
      sleep(1)
      find(dirImg)


class KonquerorWebdavTest:

   _client = None

   def __init__(self, client):
      self._client = client

   def run(self):
      self.testInitial()
      self.testListCollection()
      self.testDownloadSingle()
      self.testListSubdir()
      self.testDownloadMultiple()
      self.testCreateNewdir()
      self.testListNewdir()
      self.testUploadSingle()
      self.testCreateNewsubdir()
      self.testListNewsubdir()
      self.testUploadSingleOverwrite()
      self.testDeleteNewdir()
      self.testUploadNew()
      self.testDownloadUploaded()
      self.testRenameFiles()
      self.testCopyFilesRemote()
      self.testRenameCollection()
      self.testCopyCollectionSame()
      pass

   def testInitial(self):
      self._client.switchRemote()
      find(Pattern().similar(0.80).firstN(1))

   def testListCollection(self):
      self._client.openRemote(Pattern().similar(0.80).firstN(1), "collection")
      find(Pattern().similar(0.80).firstN(1))

   def testDownloadSingle(self):
      self._client.switchRemote()
      click(Pattern().similar(0.80).firstN(1))
      self._client.copy()
      self._client.switchLocal()
      sleep(1)
      self._client.paste()
      sleep(1)
      find(Pattern().similar(0.80).firstN(1))

   def testListSubdir(self):
      self._client.openRemote(Pattern().similar(0.80).firstN(1), "subdir")
      find(Pattern().similar(0.80).firstN(1))

   def testDownloadMultiple(self):
      self._client.switchRemote()
      click(Pattern().similar(0.80).firstN(1))
      type(Key.RIGHT, KEY_SHIFT)
      self._client.copy()
      sleep(1)
      self._client.switchLocal()
      sleep(1)
      self._client.paste()
      sleep(1)
      find(Pattern().similar(0.80).firstN(1))

   def testCreateNewdir(self):
      self._client.switchRemote()
      self._client.createDir("newdir", Pattern().similar(0.79).firstN(1))

   def testListNewdir(self):
      self._client.openRemote(Pattern().similar(0.79).firstN(1), "newdir")
      find(Pattern().similar(0.80).firstN(1))

   def testUploadSingle(self):
      self._client.switchLocal()
      click(Pattern().similar(0.80).firstN(1))
      self._client.copy()
      self._client.switchRemote()
      self._client.paste()
      find(Pattern().similar(0.80).firstN(1))

   def testCreateNewsubdir(self):
      self._client.switchRemote()
      self._client.createDir("newsubdir", Pattern().similar(0.80).firstN(1))

   def testListNewsubdir(self):
      self._client.openRemote(Pattern().similar(0.80).firstN(1), "newsubdir")
      find(Pattern().similar(0.80).firstN(1))

   def testUploadSingleOverwrite(self):
      self._client.switchLocal()
      click(Pattern().similar(0.80).firstN(1))
      self._client.copy()
      self._client.switchRemote()
      self._client.paste()
      find(Pattern().similar(0.80).firstN(1))
      self._client.switchLocal()
      click(Pattern().similar(0.80).firstN(1))
      self._client.copy()
      self._client.switchRemote()
      self._client.paste()
      wait(Pattern().similar(0.30).firstN(1))
      click(Pattern().similar(0.70).firstN(1))
      find(Pattern().similar(0.80).firstN(1))

   def testDeleteNewdir(self):
      self._client.upRemote("collection")
      click(Pattern().similar(0.80).firstN(1))
      type(Key.DELETE)
      wait(Pattern().similar(0.70).firstN(1))
      click(Pattern().similar(0.70).firstN(1))
      find(Pattern().similar(0.80).firstN(1))

   def testUploadNew(self):
      self._client.upLocal("/")
      self._client.openLocal(Pattern().similar(0.90).firstN(1), "up")
      type("a", KEY_CTRL)
      sleep(0.5)
      self._client.copy()
      self._client.switchRemote()
      sleep(1)
      self._client.paste()
      sleep(2)
      find(Pattern().similar(0.80).firstN(1))

   def testDownloadUploaded(self):
      self._client.switchRemote()
      # delete first, to copy everything else
      click(Pattern().similar(0.80).firstN(1))
      type(Key.DELETE)
      sleep(0.5)
      type(Key.ENTER)
      sleep(1)
      type("a", KEY_CTRL)
      sleep(1)
      self._client.copy()
      self._client.upLocal("/")
      self._client.openLocal(Pattern().similar(0.80).firstN(1), "down")
      self._client.paste()
      find(Pattern().similar(0.80).firstN(1))

   def testRenameFiles(self):
      self._client.switchRemote()
      self._client.rename(Pattern().similar(0.80).firstN(1), u"put_test_renamed.xml", Pattern().similar(0.80).firstN(1))
      sleep(0.5)
      self._client.rename(Pattern().similar(0.80).firstN(1), u"put_test_utf8_\u00f6\u00e4\u00fc\u00df.txt", Pattern().similar(0.80).firstN(1))
      sleep(0.5)
      self._client.rename(Pattern().similar(0.70).firstN(1), u"put_non_utf8_test.txt", Pattern().similar(0.80).firstN(1))

   def testCopyFilesRemote(self):
      self._client.switchRemote()
      click(Pattern().similar(0.80).firstN(1))
      type(Key.RIGHT, KEY_SHIFT)
      sleep(1)
      type(Key.RIGHT, KEY_SHIFT)
      sleep(1)
      self._client.copy()
      self._client.openRemote(Pattern().similar(0.80).firstN(1), "collection")
      sleep(1)
      self._client.paste()
      wait(Pattern().similar(0.70).firstN(1))

   def testRenameCollection(self):
      self._client.upRemote()
      self._client.rename(Pattern().similar(0.80).firstN(1), "renamed_collection", Pattern().similar(0.80).firstN(1))

   def testCopyCollectionSame(self):
      click(Pattern().similar(0.80).firstN(1))
      self._client.copy()
      sleep(1)
      self._client.paste()
      wait(Pattern().similar(0.80).firstN(1))
      type(Key.ENTER)

konqueror = Konqueror()
test = KonquerorWebdavTest(konqueror)
test.run()