# Copyright (c) 2013 Mirantis, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

from flask import Flask, make_response, send_from_directory, send_file, \
                jsonify, request, abort
from werkzeug import secure_filename
import os
from parser import ManifestParser
from archiver import Archiver

app = Flask(__name__)
app.config.from_pyfile('consts.py')

@app.route('/client/ui')
def get_ui_data():
    parser = ManifestParser(app.config["ROOT_DIRECTORY"])
    manifests = parser.parse()
    archive_name = Archiver().create(manifests, "ui_forms")

    return send_from_directory(os.path.dirname(__file__), archive_name)


@app.route('/client/conductor')
def get_conductor_data():
    parser = ManifestParser(app.config["ROOT_DIRECTORY"])
    manifests = parser.parse()
    archive_name = Archiver().create(manifests,
                                "heat_templates",
                                "agent_templates",
                                "scripts")

    return send_from_directory(os.path.dirname(__file__), archive_name)


@app.route('/admin/<data_type>', methods=['GET', 'POST'])
def get_data_type_locations(data_type):
    ####### validation ########
    if data_type not in app.config['DATA_TYPES']:
        abort(404)
    result_path = os.path.join(app.config["ROOT_DIRECTORY"],
                               app.config["DIRECTORIES_BY_TYPE"][
                                   data_type])
    ####### end validation ########
    if request.method == 'GET':
        locations = []

        for path, subdirs, files in os.walk(result_path):
            for name in files:
                locations.append(os.path.join(path, name))
        result = {data_type: locations}
        return jsonify(result)

    if request.method == 'POST':
        file_to_upload = request.files.get('file')
        if file_to_upload:
            filename = secure_filename(file_to_upload.filename)
            file_to_upload.save(os.path.join(result_path, filename))
            return jsonify(result="success")
        else:
            abort(503)


@app.route('/admin/<data_type>/<path:path>', methods=['GET', 'POST'])
def get_data_type_locations_by_path_or_get_file(data_type, path):
    ####### validation ########
    if data_type not in app.config['DATA_TYPES']:
        abort(404)
    result_path = os.path.join(app.config["ROOT_DIRECTORY"],
                               app.config["DIRECTORIES_BY_TYPE"][
                                   data_type],
                               path)
    if not os.path.exists(result_path):
        abort(404)
    ####### end validation ########
    if request.method == 'GET':
        locations = []
        #return file content or directory content
        if os.path.isfile(result_path):
            return send_file(result_path)
        else:
            for file in os.listdir(result_path):
                locations.append(os.path.join(path, file))
            result = {data_type: locations}
            return jsonify(result)

    if request.method == 'POST':
        file_to_upload = request.files.get('file')
        if file_to_upload:
            filename = secure_filename(file_to_upload.filename)
            file_to_upload.save(os.path.join(result_path, filename))
            return jsonify(result="success")
        else:
            abort(403)


@app.route('/admin/<data_type>/<path:path>', methods=['PUT', 'DELETE'])
def create_dirs(data_type, path):
    if data_type not in app.config['DATA_TYPES']:
        abort(404)
    result_path = os.path.join(app.config["ROOT_DIRECTORY"],
                               app.config["DIRECTORIES_BY_TYPE"][
                                   data_type],
                               path)
    if request.method == 'PUT':
        resp = make_response()
        if os.path.exists(result_path):
            return resp
        if data_type == app.config['MANIFESTS']:
            abort(403)
        try:
            os.makedirs(result_path)
        except Exception as e:
            abort(403)
        return resp

    if request.method == 'DELETE':
        if not os.path.exists(result_path):
            abort(404)
        if os.path.isfile(result_path):
            try:
                os.remove(result_path)
            except Exception as e:
                abort(404)
        else:
            try:
                os.rmdir(result_path)
            except Exception as e:
                abort(403)
        resp = make_response()
        return resp


if __name__ == '__main__':
    app.run(debug=True)
    #by default server running on localhost:5000