""" Serve static files. """

import os
import io

from flask import send_file, jsonify, redirect, url_for, abort

from vasl_templates.webapp import app
from vasl_templates.webapp.file_server.vasl_mod import VaslMod
from vasl_templates.webapp.config.constants import DATA_DIR

vasl_mod = None
if app.config.get( "VASL_MOD" ):
    vasl_mod = VaslMod( app.config["VASL_MOD"], DATA_DIR )

# ---------------------------------------------------------------------

class FileServer:
    """Serve static files."""

    def __init__( self, base_dir ):
        self.base_dir = os.path.abspath( base_dir )

    def get_file( self, fname ):
        """Serve a file."""
        if not fname:
            return None
        fname = os.path.join( self.base_dir, fname )
        fname = os.path.abspath( fname )
        if not os.path.isfile( fname ):
            return None
        prefix = os.path.commonpath( [ self.base_dir, fname ] )
        if prefix != self.base_dir:
            return None # nb: files must be sub-ordinate to the configured directory
        return fname

# ---------------------------------------------------------------------

def install_vasl_mod( new_vasl_mod ):
    """Install a new VASL module."""
    global vasl_mod
    vasl_mod = new_vasl_mod

# ---------------------------------------------------------------------

@app.route( "/counter/<gpid>/<side>/<int:index>" )
@app.route( "/counter/<gpid>/<side>", defaults={"index":0} )
def get_counter_image( gpid, side, index ):
    """Get a counter image."""

    # check if a VASL module has been configured
    if not vasl_mod:
        return redirect( url_for( "static", filename="images/missing-image.png" ), code=302 )

    # return the specified counter image
    image_path, image_data = vasl_mod.get_piece_image( int(gpid), side, int(index) )
    if not image_data:
        abort( 404 )
    return send_file(
        io.BytesIO( image_data ),
        attachment_filename = os.path.split( image_path )[1]## nb: so Flask can figure out the MIME type
    )

# ---------------------------------------------------------------------

@app.route( "/vasl-piece-info" )
def get_vasl_piece_info():
    """Get information about the VASL pieces."""

    # check if a VASL module has been configured
    if not vasl_mod:
        return jsonify( {} )

    # return the VASL piece info
    return jsonify( vasl_mod.get_piece_info() )