ail-framework/bin/lib/objects/ail_objects.py

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

import os
import sys
import uuid
import redis

from abc import ABC
from flask import url_for


sys.path.append(os.path.join(os.environ['AIL_BIN'], 'lib/'))
import ConfigLoader

sys.path.append(os.path.join(os.environ['AIL_BIN'], 'lib/objects'))
from CryptoCurrencies import CryptoCurrency
from Decodeds import Decoded
from Domains import Domain
from Items import Item
from Pgps import Pgp
from Screenshots import Screenshot
from Usernames import Username

##################################################################
##################################################################
#sys.path.append(os.path.join(os.environ['AIL_BIN'], 'lib/'))

#sys.path.append(os.path.join(os.environ['AIL_BIN'], 'packages/'))
##################################################################
##################################################################

config_loader = ConfigLoader.ConfigLoader()
r_serv_metadata = config_loader.get_redis_conn("ARDB_Metadata")
config_loader = None

class AILObjects(object):
    initial = 0
    ongoing = 1
    completed = 2


# # TODO: check change paste => item
def get_all_objects():
    return ['domain', 'item', 'pgp', 'cryptocurrency', 'decoded', 'screenshot', 'username']

def get_object(obj_type, subtype, id):
    if obj_type == 'item':
        return Item(id)
    elif obj_type == 'domain':
        return Domain(id)
    elif obj_type == 'decoded':
        return Decoded(id)
    elif obj_type == 'screenshot':
        return Screenshot(id)
    elif obj_type == 'cryptocurrency':
        return CryptoCurrency(id, subtype)
    elif obj_type == 'pgp':
        return Pgp(id, subtype)
    elif obj_type == 'username':
        return Username(id, subtype)

def get_object_svg(obj_type, subtype, id):
    object = get_object(obj_type, subtype, id)
    return object.get_svg_icon()

def get_objects_meta(l_dict_objs, icon=False, url=False, flask_context=False):
    l_meta = []
    for dict_obj in l_dict_objs:
        object = get_object(dict_obj['type'], dict_obj['subtype'], dict_obj['id'])
        dict_meta = object.get_default_meta(tags=True)
        if icon:
            dict_meta['icon'] = object.get_svg_icon()
        if url:
            dict_meta['link'] = object.get_link(flask_context=flask_context)
        l_meta.append(dict_meta)
    return l_meta

# # TODO: CHECK IF object already have an UUID
def get_misp_object(obj_type, subtype, id):
    object = get_object(obj_type, subtype, id)
    return object.get_misp_object()

# get misp relationship
def get_objects_relationship(obj_1, obj2):
    relationship = {}
    obj_types = ( obj_1.get_type(), obj2.get_type() )

    ##############################################################
    # if ['cryptocurrency', 'pgp', 'username', 'decoded', 'screenshot']:
    #     {'relation': '', 'src':, 'dest':}
    #     relationship[relation] =
    ##############################################################
    if 'cryptocurrency' in obj_types:
        relationship['relation'] = 'extracted-from'
        if obj1_type == 'cryptocurrency':
            relationship['src'] = obj1_id
            relationship['dest'] =  obj2_id
        else:
            relationship['src'] = obj2_id
            relationship['dest'] =  obj1_id

    elif 'pgp' in obj_types:
        relationship['relation'] = 'extracted-from'

    elif 'username' in obj_types:
        relationship['relation'] = 'extracted-from'

    elif 'decoded' in obj_types:
        relationship['relation'] = 'included-in'

    elif 'screenshot' in obj_types:
        relationship['relation'] = 'screenshot-of'

    elif 'domain' in obj_types:
        relationship['relation'] = 'extracted-from'

    # default
    else:
        pass






    return relationship




################################################################################
################################################################################
################################################################################

def delete_obj(obj_type, subtype, id):
    object = get_object(obj_type, subtype, id)
    return object.delete()

################################################################################
################################################################################
################################################################################
################################################################################
################################################################################