Create attractive VASL scenarios, with loads of useful information embedded to assist with game play.
https://vasl-templates.org
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.3 KiB
75 lines
2.3 KiB
""" 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() )
|
|
|