import os
import objetive
from flask import Flask
from flask import jsonify
from flask import request
from getrails import search
os.sys.path.append('modules/')
from generator import main

port = int(os.environ.get("PORT", 5000))
app = Flask(__name__)
msg = "Its OWASP D4N155 project for API, see: https://github.com/OWASP/D4N155, branch: api"

# See all data
@app.route('/')
def index():
    # Get all registers in DB
    helper = {
        "Query URL": "/domain/<max_num_of_response:int>?url=<url:str>",
        "Query domain": "/domain/<domain:str>",
        "Query words": "/make/<words_with_space:str>"
              }
    response = jsonify(result = helper, helpus = "{}".format(msg))
    response.headers['Access-Control-Allow-Origin'] = '*'
    return response

# Read url contents
@app.route('/domain/<int:limit>')
def read_page(limit):
    # Get id in DB
    read_url = main(objetive.text(request.args['url']), int(limit))
    response = jsonify(result = read_url, helpus = "{}".format(msg))
    response.headers['Access-Control-Allow-Origin'] = '*'
    return response

# See all urls of domain
@app.route('/domain/<param>')
def domain(param):
    # Get id in DB
    get_urls = search("site:{}".format(param))
    response = jsonify({ "result": { "data": get_urls, "length": len(get_urls) }, "helpus": msg })
    response.headers['Access-Control-Allow-Origin'] = '*'
    return response

# Make new analyze
@app.route('/make/<param>')
def make(param):
    response = jsonify(result = main(param), helpus = "{}".format(msg))
    response.headers['Access-Control-Allow-Origin'] = '*'
    return response

if __name__ == '__main__':
    app.config['JSON_AS_ASCII'] = False
    app.run(debug=False,host='0.0.0.0', port=port)