import time
import os
import requests


class SeafileClient:

    def __init__(self, server, username, password=None, token=None):
        self.server = server
        self.token = token
        self.session = requests.Session()
        if token:
            self.session.headers.update(
                {'Authorization': "Token %s" % self.token.token})
        self.login = (username, password)

    def api_endpoint(self):
        return "%s/api2" % self.server

    def ping(self, auth=False):
        try:
            return self.session.get("%s%s/ping" % (self.api_endpoint(), "/auth" if auth else "")).text == "\"pong\""
        except:
            return False

    def obtain_token(self):
        user, passw = self.login
        try:
            req = requests.post("%s/auth-token/" % self.api_endpoint(),
                                data={'username': user, 'password': passw})
            json = req.json()
            if "non_field_errors" in json:
                print(json["non_field_errors"])
                return False
            return SeafileToken(user, json["token"])
        except:
            return False

    def authorize(self):
        self.token = self.obtain_token()
        if self.token:
            self.session.headers.update(
                {'Authorization': "Token %s" % self.token.token})
        return self.token != False

    # curl -H 'Authorization: Token 24fd3c026886e3121b2ca630805ed425c272cb96' -H 'Accept: application/json; indent=4' https://cloud.seafile.com/api2/repos/
    def libraries(self):
        resp = self.session.get("%s/repos/" % self.api_endpoint(), headers={
                                'Authorization': "Token %s" % self.token.token, 'Accept': 'application/json; indent=4'})
        if not resp.status_code == 200:
            return
        libraries = []
        for lib in resp.json():
            if not lib['encrypted']:
                libraries.append(SeafileLibrary(
                    self, lib['id'], lib['name'], lib['owner']))
        return libraries


class SeafileLibrary:

    def __init__(self, client, id, name, owner):
        self.client = client
        self.session = client.session
        self.id = id
        self.name = name
        self.owner = owner

    def api_endpoint(self):
        return "%s/repos/%s" % (self.client.api_endpoint(), self.id)

    def upload(self, file, file_name, directory, link=None):
        def obtain_link():
            # curl -H "Authorization: Token f2210dacd9c6ccb8133606d94ff8e61d99b477fd" https://cloud.seafile.com/api2/repos/99b758e6-91ab-4265-b705-925367374cf0/upload-link/
            print("%s/upload-link" % self.api_endpoint())
            quoted = self.session.get("%s/upload-link" % self.api_endpoint(), headers={
                                      'Authorization': "Token %s" % self.client.token.token, }).text
            return quoted[1:-1]

        # curl -H "Authorization: Token f2210dacd9c6ccb8133606d94ff8e61d99b477fd" -F file=@test.txt -F filename=test.txt -F parent_dir=/ http://cloud.seafile.com:8082/upload-api/73c5d117-3bcf-48a0-aa2a-3f48d5274ae3
        resp = self.session.post(link or obtain_link(),
                                 files={'file': (file_name, open(
                                     file, 'rb')), 'parent_dir': directory, 'target_file': "%s/%s" % (directory, file_name)},
                                 headers={
                                     'Authorization': "Token %s" % self.client.token.token, }
                                 )
        if resp.status_code == 200:
            return SeafileFile(self, ("%s/%s" % (directory, file_name)).replace('//', '/').replace('//', '/'), resp.text)
        return False

    # curl -H 'Authorization: Token f2210dacd3606d94ff8e61d99b477fd' -H 'Accept: application/json; charset=utf-8; indent=4' https://cloud.seafile.com/api2/repos/dae8cecc-2359-4d33-aa42-01b7846c4b32/file/detail/?p=/foo.c
    def file_info(self, path):
        resp = self.session.get("%s/file/detail/?p=%s" % (self.api_endpoint(), path), headers={
                                'Authorization': "Token %s" % self.token.token, 'Accept': 'application/json; indent=4'})
        if resp.status_code == 200:
            json = resp.json()
            return SeafileFile(self, path, json['id'], json['size'])
        return False

    def __str__(self):
        return "%s on %s by %s" % (self.name, self.client.server, self.owner)


class SeafileFile:

    def __init__(self, library, path, id, size=0):
        self.id = id
        self.path = path
        self.library = library
        self.session = library.session
        self.size = size

    # curl -v  -X PUT -d "p=/foo.md" -H 'Authorization: Token f2210dacd9c6ccb8133606d94ff8e61d99b477fd' -H 'Accept: application/json; indent=4' https://cloud.seafile.com/api2/repos/afc3b694-7d4c-4b8a-86a4-89c9f3261b12/file/shared-link/
    def share(self,expire_days=None):
        data = {'p': self.path}
        if expire_days and not expire_days == 0:
            data['expire'] = expire_days
        resp = self.session.put("%s/repos/%s/file/shared-link/" % (self.library.client.api_endpoint(), self.library.id),
                                headers={'Authorization': "Token %s" % self.library.client.token.token,
                                         'Accept': 'application/json; indent=4'},
                                data = data
                                )
        return resp.headers.get("location")

    def update(self, file):
        def obtain_link():
            # curl -H "Authorization: Token f2210dacd9c6ccb8133606d94ff8e61d99b477fd" https://cloud.seafile.com/api2/repos/99b758e6-91ab-4265-b705-925367374cf0/upload-link/
            quoted = self.session.get("%s/update-link" % self.library.api_endpoint(), headers={
                                      'Authorization': "Token %s" % self.library.client.token.token, }).text
            return quoted[1:-1]
        directory, name = os.path.split(self.path)
        return self.library.upload(file, name, directory, obtain_link())


class SeafileToken:

    def __init__(self, username, token):
        self.username = username
        self.token = token

    def __str__(self):
        return "%s@%s" % (self.token, self.username)