import os

import mimetypes

from serpens import Serpens, SerpensResponse, error

BASE_DIR = os.path.dirname(__file__)

app = Serpens()

@app.route('/')

def index(request):

    if os.path.isfile(BASE_DIR + '/index.gmi'):

        with open(BASE_DIR + '/index.gmi', 'rb') as f:

            return SerpensResponse(f.read(), 20, mediatype='text/gemini')

    else:

        output = ['# Index']

        for item in os.listdir(BASE_DIR):

            output.append('=> /{} {}'.format(item, item))

        return SerpensResponse('\n'.join(output).encode('utf-8'), 20, 'text/gemini')

@app.route('/{any:path}')

def file_access(request, path):

    full_path = os.path.join(BASE_DIR, path)

    if os.path.isdir(full_path):

        if os.path.isfile(full_path + '/index.gmi'):

            with open(full_path + '/index.gmi', 'rb') as f:

                return SerpensResponse(f.read(), 20, mediatype='text/gemini')

        else:

            output = ['# ' + path]

            for item in os.listdir(full_path):

                output.append('=> /{} {}'.format(os.path.join(path, item), item))

            return SerpensResponse('\n'.join(output).encode('utf-8'), 20, 'text/gemini')

    elif os.path.isfile(full_path):

        mediatype = mimetypes.guess_type(full_path)[0] or 'application/octet-stream'

        with open(full_path + '/index.gmi', 'rb') as f:

            return SerpensResponse(f.read(), 20, mediatype=mediatype)

    else:

        return error('Not found', 51)

application = app.wsgi