# t0pic - pic.t0.vc
# MIT License

import random
import string
import requests

from flask import abort, Flask, request, redirect
from pathlib import Path
from PIL import Image

app = Flask(__name__)
DATA_FOLDER = Path('data')
MAX_SIZE = 1920
PORT = 5003
URL = 'https://pic.t0.vc'
POST_PARAM = 'pic'
MAX_FILE_SIZE = 16 * 1024 * 1024  # 16 MB

def controller_message(message):
    print('Sending controller message: ' + message)
    payload = dict(pic=message)
    r = requests.post('https://tbot.tannercollin.com/message', data=payload, timeout=5)
    if r.status_code == 200:
        return True
    else:
        print('Unable to communicate with controller!')
        return False


def help():
    form = (
        f'<form action="{URL}" method="POST" accept-charset="UTF-8" enctype="multipart/form-data">'
        '<input name="web" type="hidden" value="true">'
        f'<input name="{POST_PARAM}" type="file" accept="image/*" />'
        '<br><br><button type="submit">Submit</button></form>'
    )
    return f"""
<meta name="color-scheme" content="light dark" />
<meta name="viewport" content="width=1%" />
<pre>
                        pic.t0.vc
NAME
    t0pic: command line image host.

USAGE
    &lt;image output&gt; | curl -F '{POST_PARAM}=@/dev/stdin' {URL}
    or upload from the web:

{form}

    or paste the image into this page.

DESCRIPTION
    I got sick of imgur not working on mobile, so I built this
    Images are resized to a max dimension of {MAX_SIZE} px
    Don't use this for anything serious

EXAMPLES
    ~$ cat kitten.jpg | curl -F '{POST_PARAM}=@/dev/stdin' {URL}
       {URL}/YXKV.jpg
    ~$ firefox {URL}/YXKV.jpg

    Add this to your .bashrc:

    alias {POST_PARAM}="curl -F '{POST_PARAM}=@/dev/stdin' {URL}"

    Now you can pipe directly into {POST_PARAM}!

SOURCE CODE
    https://txt.t0.vc/CQQE
    nginx config: https://txt.t0.vc/ZKEH

SEE ALSO
    https://txt.t0.vc
    https://url.t0.vc
</pre>"""

def paste():
    return f"""
    window.addEventListener('paste', e => {{
      const file = e.clipboardData.items[0].getAsFile();
      const url = URL.createObjectURL(file);
      const image = new Image();
      image.src = url;

      const form = new FormData();
      form.append('{POST_PARAM}', file);
      fetch('/', {{
          method: 'POST',
          body: form
        }})
        .then(r => r.text())
        .then(u => window.location = u)
        .catch(e => alert(e.message));
    }});
"""

def new_id():
    return ''.join(random.choice(string.ascii_uppercase) for _ in range(4))

@app.route('/', methods=['GET'])
def index():
    return f'<html><script>{paste()}</script><body>{help()}</body></html>'

@app.route('/', methods=['POST'])
def new():
    try:
        if 'pic' not in request.files or request.files['pic'].filename == '':
            print('No file part')
            abort(400)

        pic = request.files['pic']

        if pic.mimetype not in ['image/png', 'image/jpeg', 'image/gif', 'application/octet-stream']:
            print(f'Unsupported image format: {pic.mimetype}')
            abort(400)

        if pic.seek(0, 2) > MAX_FILE_SIZE:  # Move to end to get file size
            print(f'File size exceeds maximum allowed limit: {pic.tell()} bytes')
            abort(413)
        pic.seek(0)  # Reset file pointer

        nid = new_id()
        while nid in [p.stem for p in DATA_FOLDER.iterdir()]:
            nid = new_id()

        pic = Image.open(pic.stream)

        if pic.format == 'PNG':
            ext = '.png'
        elif pic.format == 'JPEG':
            ext = '.jpg'
        elif pic.format == 'GIF':
            ext = '.gif'
        else:
            print(f'Unsupported image format: {pic.format}')
            abort(400)

        filename = nid + ext
        pic.thumbnail([MAX_SIZE, MAX_SIZE], Image.Resampling.LANCZOS)
        pic.save(DATA_FOLDER / filename)

        print(f'Adding pic {nid}{ext}')

        url = f'{URL}/{nid}{ext}'

        if pic.size != (300, 150):
            msg = 'New t0pic:\n\n' + url
            controller_message(msg)

        if 'web' in request.form:
            return redirect(url)
        else:
            return f'{url}\n'
    except Exception as e:
        print(f'Error processing image upload: {e}')
        abort(400)

if __name__ == '__main__':
    if not DATA_FOLDER.exists():
        DATA_FOLDER.mkdir(parents=True)
    app.run(port=PORT, debug=False)