from flask import Flask, Blueprint, jsonify, send_from_directory
from flask_sqlalchemy import SQLAlchemy
from app.models.sql_factory import connection_url, load_sql_datasets

from app.server.api import api

db = SQLAlchemy()

def create_app(script_info=None):
  app = Flask(__name__, static_folder='static', static_url_path='')
  app.config['SQLALCHEMY_DATABASE_URI'] = connection_url
  app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

  db.init_app(app)
  datasets = load_sql_datasets(replace=False, base_model=db.Model)

  app.register_blueprint(api, url_prefix='/api')
  app.add_url_rule('/<path:file_relative_path_to_root>', 'serve_page', serve_page, methods=['GET'])

  @app.route('/', methods=['GET'])
  def index():
    return app.send_static_file('index.html')

  @app.shell_context_processor
  def shell_context():
    return { 'app': app, 'db': db }

  @app.route("/site-map")
  def site_map():
    links = []
    for rule in app.url_map.iter_rules():
      # url = url_for(rule.endpoint, **(rule.defaults or {}))
      # print(url)
      links.append((rule.endpoint))
    return(jsonify(links))

  return app

def serve_page(file_relative_path_to_root):
  if file_relative_path_to_root[-1] == '/':
    file_relative_path_to_root += 'index.html'
  return send_from_directory("static", file_relative_path_to_root)