ail-framework/var/www/blueprints/api_rest.py

#!/usr/bin/env python3
# -*-coding:UTF-8 -*

"""
    Blueprint Flask: crawler splash endpoints: dashboard, onion crawler ...
"""

import os
import sys
import json

from functools import wraps
from flask import request, Blueprint, Response

sys.path.append(os.environ['AIL_BIN'])
##################################
# Import Project packages
##################################
from lib import ail_api
from lib import ail_core
from lib import ail_updates
from lib import crawlers

from lib import Tag

from lib.objects import ail_objects
from importer.FeederImporter import api_add_json_feeder_to_queue

from lib.objects import Domains
from lib.objects import Titles



# ============ BLUEPRINT ============
api_rest = Blueprint('api_rest', __name__, template_folder=os.path.join(os.environ['AIL_FLASK'], 'templates'))


# ============ AUTH FUNCTIONS ============

def get_auth_from_header():
    token = request.headers.get('Authorization').replace(' ', '')  # remove space
    return token


def token_required(user_role):
    def actual_decorator(funct):
        @wraps(funct)
        def api_token(*args, **kwargs):
            # Check AUTH Header
            if not request.headers.get('Authorization'):
                return create_json_response({'status': 'error', 'reason': 'Authentication needed'}, 401)

            # Check Role
            if not user_role:
                return create_json_response({'status': 'error', 'reason': 'Invalid Role'}, 401)

            token = get_auth_from_header()
            ip_source = request.remote_addr
            data, status_code = ail_api.authenticate_user(token, ip_address=ip_source)
            if status_code != 200:
                return create_json_response(data, status_code)
            elif data:
                # check user role
                if not ail_api.is_user_in_role(user_role, token):
                    return create_json_response({'status': 'error', 'reason': 'Access Forbidden'}, 403)
                else:
                    # User Authenticated + In Role
                    return funct(*args, **kwargs)
            else:
                return create_json_response({'status': 'error', 'reason': 'Internal'}, 400)

        return api_token
    return actual_decorator


# ============ FUNCTIONS ============

def create_json_response(data, status_code):  # TODO REMOVE INDENT ????????????????????
    return Response(json.dumps(data, indent=2, sort_keys=True), mimetype='application/json'), status_code

# ============= ROUTES ==============

# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # # #        CORE       # # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #

@api_rest.route("api/v1/ping", methods=['GET'])
@token_required('read_only')
def v1_ping():
    return create_json_response({'status': 'pong'}, 200)

@api_rest.route("api/v1/uuid", methods=['GET'])
@token_required('read_only')
def v1_uuid():
    ail_uid = ail_core.get_ail_uuid()
    return create_json_response({'uuid': ail_uid}, 200)

@api_rest.route("api/v1/version", methods=['GET'])
@token_required('read_only')
def v1_version():
    version = ail_updates.get_ail_version()
    return create_json_response({'version': version}, 200)

@api_rest.route("api/v1/pyail/version", methods=['GET'])
@token_required('read_only')
def v1_pyail_version():
    ail_version = 'v1.0.0'
    return create_json_response({'version': ail_version}, 200)


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # # #      CRAWLERS       # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# # TODO: ADD RESULT JSON Response
@api_rest.route("api/v1/add/crawler/task", methods=['POST'])  # TODO V2 Migration
@token_required('analyst')
def add_crawler_task():
    data = request.get_json()
    user_token = get_auth_from_header()
    user_id = ail_api.get_user_from_token(user_token)
    res = crawlers.api_add_crawler_task(data, user_id=user_id)
    if res:
        return create_json_response(res[0], res[1])

    dict_res = {'url': data['url']}
    return create_json_response(dict_res, 200)


@api_rest.route("api/v1/add/crawler/capture", methods=['POST'])  # TODO V2 Migration
@token_required('analyst')
def add_crawler_capture():
    data = request.get_json()
    user_token = get_auth_from_header()
    user_id = ail_api.get_user_from_token(user_token)
    res = crawlers.api_add_crawler_capture(data, user_id)
    if res:
        return create_json_response(res[0], res[1])

    dict_res = {'url': data['url']}
    return create_json_response(dict_res, 200)

# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # #       IMPORTERS       # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
@api_rest.route("api/v1/import/json/item", methods=['POST'])  # TODO V2 Migration
@token_required('user')
def import_json_item():
    data_json = request.get_json()
    res = api_add_json_feeder_to_queue(data_json)
    return Response(json.dumps(res[0]), mimetype='application/json'), res[1]

# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # # #      OBJECTS      # # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # # #      TITLES       # # # # # # # # # # # # # # # # # #
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #

@api_rest.route("api/v1/titles/download", methods=['GET'])
@token_required('analyst')
def objects_titles_download():
    return create_json_response(Titles.Titles().get_contents_ids(), 200)


# TODO
@api_rest.route("api/v1/titles/download/unsafe", methods=['GET'])  # TODO REFACTOR ME
@token_required('analyst')
def objects_titles_download_unsafe():
    all_titles = {}
    unsafe_tags = Tag.unsafe_tags
    for tag in unsafe_tags:
        domains = Tag.get_tag_objects(tag, 'domain')
        for domain_id in domains:
            domain = Domains.Domain(domain_id)
            domain_titles = domain.get_correlation('title').get('title', [])
            for dt in domain_titles:
                title = Titles.Title(dt[1:])
                title_content = title.get_content()
                if title_content and title_content != 'None':
                    if title_content not in all_titles:
                        all_titles[title_content] = []
                    all_titles[title_content].append(domain.get_id())
    return Response(json.dumps(all_titles), mimetype='application/json'), 200