Geoffrey-Django/api/commands.py

from django.db.models import Q, F
from GeoffreyApp.models import *
from GeoffreyApp.errors import *
from GeoffreyApp.minecraft_api import *
import inspect
import datetime
import re

from GeoffreyApp.util import objects_list_to_json

command_dict = {"GET": {}, "POST": {}, "DELETE": {}}


def match_tunnel(tunnel_direction):
    for direction in Tunnel.TUNNEL_NAMES:
        if re.search("{}.*".format(tunnel_direction), direction[1], re.IGNORECASE):
            return direction[0]

    raise InvalidTunnelError


def get_required_args(func):
    args = inspect.getfullargspec(func)

    return args.args + args.kwonlyargs


def command(type):
    def command_dec(func):
        def add_command():
            command_dict[type][func.__name__] = {"func": func,
                                                 "params": get_required_args(func),
                                                 "help": parse_help(func)}
            return func
        
        return add_command()

    return command_dec


def parse_help(func):
    try:
        match = re.search(".*:help:.*", func.__doc__)

        return match.group(0).partition(":help: ")[-1]
    except:
        return ' '


def get_player(discord_uuid=None, mc_uuid=None):
    try:
        if discord_uuid is not None:
            player = Player.objects.get(discord_uuid__iexact=discord_uuid)
        elif mc_uuid is not None:
            player = Player.objects.get(mc_uuid__iexact=mc_uuid)
        else:
            raise AttributeError
    except Player.DoesNotExist:
        raise PlayerNotFound

    return player


def get_location(owner, name=None, loc_type=Location):
    if name is None:
        loc_list = loc_type.objects.filter(owner=owner).all()

        if len(loc_list) == 1:
            loc = loc_list[0]
        elif len(loc_list) == 0:
            raise NoLocationsInDatabase
        else:
            raise EntryNameNotUniqueError
    else:
        loc_list = loc_type.objects.filter(owner=owner, name__iexact=name).all()

        if len(loc_list) == 1:
            loc = loc_list[0]
        else:
            raise LocationLookUpError

    return loc


def add_location(x_pos, z_pos, name=None, discord_uuid=None, mc_uuid=None, loc_type=Location):
    player = get_player(discord_uuid, mc_uuid)

    name_was_none = False
    if name is None:
        name = "{}'s {}".format(player.name, loc_type.__name__)
        name_was_none = True

    if Location.objects.filter(name__iexact=name).all().count() > 0:
        if name_was_none:
            raise LocationLookUpError
        else:
            raise EntryNameNotUniqueError
    else:
        location = loc_type.objects.create(name=name, x_coord=x_pos, z_coord=z_pos)
        location.owner.add(player)
        location.save()

        return location.json


@command("POST")
def register(player_name, discord_uuid):
    """
    :request: POST
    :param player_name: Minecraft in-game name
    :param discord_uuid: Discord UUID if registering from Discord
    :return: JSON representation of the new Player
    :raise: PlayerInDBError
    :help: Registers your Discord and Minecraft account with the the database
    """

    mc_uuid = grab_UUID(player_name)

    try:
        get_player(mc_uuid=mc_uuid)
        raise PlayerInDBError
    except PlayerNotFound:
        player = Player.objects.create(name=player_name, mc_uuid=mc_uuid, discord_uuid=discord_uuid)
        player.save()
        return player.json


@command("POST")
def add_base(x_pos, z_pos, name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param x_pos: MC X Coordinate
    :param z_pos: MC Z Coordinate
    :param name:  Base Name (If None, Defaults to Player's Base)
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: JSON representation of the new base
    :raises: EntryNameNotUniqueError, PlayerNotFound, LocationLookupError
    :help: Adds your base to the database.
    """

    return add_location(x_pos, z_pos, name=name, discord_uuid=discord_uuid, mc_uuid=mc_uuid, loc_type=Base)


@command("POST")
def add_shop(x_pos, z_pos, name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param x_pos: MC X Coordinate
    :param z_pos: MC Z Coordinate
    :param name:  Shop Name (If None, Defaults to Player's Shop)
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: JSON representation of the new shop
    :raises: EntryNameNotUniqueError, PlayerNotFound, LocationLookupError
    :help: Adds your shop to the database.
    """

    return add_location(x_pos, z_pos, name=name, discord_uuid=discord_uuid, mc_uuid=mc_uuid, loc_type=Shop)


@command("POST")
def add_town(x_pos, z_pos, name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param x_pos: MC X Coordinate
    :param z_pos: MC Z Coordinate
    :param name:  Shop Name (If None, Defaults to Player's Shop)
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: JSON representation of the new town
    :raises: EntryNameNotUniqueError, PlayerNotFound, LocationLookupError
    :help: Adds your town to the database.
    """

    return add_location(x_pos, z_pos, name=name, discord_uuid=discord_uuid, mc_uuid=mc_uuid, loc_type=Town)


@command("POST")
def add_farm(x_pos, z_pos, name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param x_pos: MC X Coordinate
    :param z_pos: MC Z Coordinate
    :param name:  Shop Name (If None, Defaults to Player's Shop)
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: JSON representation of the new town
    :raises: EntryNameNotUniqueError, PlayerNotFound, LocationLookupError
    :help: Adds your public farm to the database.
    """

    return add_location(x_pos, z_pos, name=name, discord_uuid=discord_uuid, mc_uuid=mc_uuid, loc_type=PublicFarm)



@command("POST")
def add_tunnel(tunnel_direction, tunnel_number, location_name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param tunnel_direction: Tunnel Direction
    :param tunnel_number:  Tunnel Coordinate
    :param location_name: Name of the Location to attach the tunnel to
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: JSON Representation of the Tunnel
    :raises: PlayerNotFound, LocationHasTunnelError, EntryNameNotUniqueError, NoLocationsInDatabase, InvalidTunnelError
    :help: Adds your tunnel to the database.
    """
    player = get_player(discord_uuid, mc_uuid)
    if location_name is None:
        loc = get_location(player)
        location_name = loc.name
    else:
        loc = get_location(player, name=location_name)

    tunnel_direction = match_tunnel(tunnel_direction)

    if Tunnel.objects.filter(location=loc).first():
        raise LocationHasTunnelError

    tunnel = Tunnel.objects.create(tunnel_direction=tunnel_direction, tunnel_number=tunnel_number,
                                   location=get_location(player, location_name))

    return tunnel.json


@command("GET")
def find_location(search, limit=25):
    """
    :request: GET
    :param search: Location name or owner to search for
    :param limit: How man locations to return
    :return: List of the matching locations
    :raises: LocationLookupError
    :help: Finds all the locations matching the search term
    """

    locations = Location.objects.filter(Q(name__icontains=search) | Q(owner__name__icontains=search)).all()[:limit]

    if len(locations) == 0:
        raise LocationLookUpError

    return objects_list_to_json(locations)


@command("POST")
def delete(name, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param name: Name of location to delete
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: Location Name
    :raises: LocationLookUpError
    :help: Deletes a location from the database
    """

    owner = get_player(discord_uuid, mc_uuid)
    try:
        Location.objects.get(name__iexact=name, owner=owner).delete()
    except Location.DoesNotExist:
        raise LocationLookUpError

    return name


@command("GET")
def find_around(x_pos, z_pos, radius=200):
    """
    :request: GET
    :param x_pos: MC X Coordinate
    :param z_pos: MC Z Coordinate
    :param radius: Radius to each around (default 200)
    :return: List of all locations in the radius
    :raises: LocationLookupError
    :help: Finds all the locations around a certain point
    """

    x_pos = int(x_pos)
    z_pos = int(z_pos)
    radius = int(radius)

    locations = Location.objects.filter(x_coord__range=(x_pos - radius, x_pos + radius),
                                        z_coord__range=(z_pos - radius, z_pos + radius)).all()

    if len(locations) == 0:
        raise LocationLookUpError

    return objects_list_to_json(locations)


@command("POST")
def add_item(item_name, quantity, diamond_price, shop_name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param item_name: name of the item
    :param quantity: number of items being sold
    :param diamond_price: price in diamonds
    :param shop_name: Name of the shop, can be blank if the user only has one shop
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: Item Listing
    :raises: PlayerNotFound, LocationLookupError, EntryNameNotUniqueError, NoLocationsInDatabase
    :help: Adds an item to a shop's inventory.
    """

    player = get_player(discord_uuid, mc_uuid)

    shop = get_location(player, shop_name, Shop).shop

    item_listing = ItemListing.objects.create(shop=shop, amount=int(quantity), price=int(diamond_price),
                                              item_name=item_name)

    return item_listing.json


@command("POST")
def add_resource(resource_name, farm_name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param resource_name: name of the resource
    :param farm_name: name of the farm to add the resource to. Can be none.
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: Item Listing
    :raises: PlayerNotFound, LocationLookupError, EntryNameNotUniqueError, NoLocationsInDatabase
    :help: Adds a resource to a farm.
    """

    player = get_player(discord_uuid, mc_uuid)

    farm = get_location(player, farm_name, PublicFarm).publicfarm

    resource = Resource.objects.create(farm=farm, resource_name=resource_name)

    return resource.json


@command("GET")
def find_farm(resource_name):
    if len(resource_name) == 0:
        raise EmptryString

    return objects_list_to_json(PublicFarm.objects.filter(resource__resource_name__icontains=resource_name).all())


@command("GET")
def selling(item_name):
    """
    :request: GET
    :param item_name: Item name to search for
    :return: List of top matching shops, sorted by the
    :raises: ItemNotFound
    :help: Lists shops selling an item. Sorted by when they were last restocked.
    """

    return get_selling(item_name, sort="-date_restocked")


@command("GET")
def selling_price(item_name):
    """
    :request: GET
    :param item_name: Item name to search for
    :return: List of top matching shops, sorted by the
    :raises: ItemNotFound
    :help: Lists shops selling an item. Sorted lowest price to highest price.
    """

    return get_selling(item_name, sort="normalized_price")


def get_selling(item_name, sort):
    items = []
    if len(item_name) == 0:
        raise EmptryString

    all_shop_ids = ItemListing.objects.annotate(normalized_price=F('price') / F('amount')) \
        .filter(item_name__icontains=item_name).order_by(sort).values('shop_id').all()

    if len(all_shop_ids) == 0:
        raise ItemNotFound

    # Removes duplicates
    shop_ids = []

    for shop_id in all_shop_ids:
        if shop_id not in shop_ids:
            shop_ids.append(shop_id)

    for shop_id in shop_ids:
        shop = Shop.objects.get(pk=shop_id['shop_id']).json

        item_query = ItemListing.objects.annotate(normalized_price=F('price') / F('amount')) \
            .filter(item_name__icontains=item_name, shop_id=shop_id['shop_id']) \
            .order_by(sort) \
            .values("item_name", "price", "amount")

        item_list = []
        for item in item_query:
            item_list.append(item)

        shop["items"] = item_list
        items.append(shop)

    return items


@command("GET")
def info(location_name):
    """
    :request: GET
    :param location_name: Name of the location to get info on
    :return: JSON representation of location
    :raises: LocationLookupError
    :help: Finds all the locations matching the search term
    """

    location = Location.objects.filter(name__iexact=location_name).first()

    if location is None:
        location = Location.objects.filter(name__iregex=".*{}.*".format(location_name)).first()

    if location is not None:
        loc = location.loc_child_obj
        loc_json = loc.json

        if type(loc) == Shop:
            loc_json["items"] = []
            for item in loc.shop_selling.all():
                loc_json["items"].append(item.json)

        return loc_json
    else:
        raise LocationLookUpError


@command("GET")
def tunnel(player_name):
    """
    :request: GET
    :param player_name: MC player name
    :return: List of all the tunnels a user owns
    :raises: LocationLookUpError
    :raises: LocationLookupError
    :help: Finds all the tunnels a player owns
    """

    tunnels = Tunnel.objects.filter(location__owner__name__icontains=player_name).all()

    if len(tunnels) == 0:
        raise LocationLookUpError

    return objects_list_to_json(tunnels)


@command("POST")
def edit_pos(x, z, loc_name, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param x: New MC X coordinate
    :param z: New MC Z Coordinate
    :param loc_name: Location Name to edit
    :param discord_uuid: Discord UUID
    :param mc_uuid: MC UUID
    :return: Edited Locatio
    :raises:
    :raises: PlayerNotFound, LocationLookupError, EntryNameNotUniqueError, NoLocationsInDatabase
    :help: Edits the position of a location
    """

    player = get_player(discord_uuid=discord_uuid, mc_uuid=mc_uuid)
    location = get_location(player, loc_name)

    location.x_coord = x
    location.z_coord = z
    location.save()

    return location.json


@command("POST")
def edit_tunnel(tunnel_direction, tunnel_number, loc_name, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param tunnel_direction: New Tunnel Direction
    :param tunnel_number: New Tunnel Address
    :param loc_name: Location Name to edit
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: Edited Location
    :raises: PlayerNotFound, LocationLookupError, EntryNameNotUniqueError, NoLocationsInDatabase
    :help: Edits the tunnel of a location
    """

    player = get_player(discord_uuid=discord_uuid, mc_uuid=mc_uuid)
    location = get_location(player, loc_name)
    tunnel_direction = match_tunnel(tunnel_direction)

    try:
        tunnel = Tunnel.objects.get(location=location)
        tunnel.tunnel_direction = tunnel_direction
        tunnel.tunnel_number = tunnel_number
        tunnel.save()
    except Tunnel.DoesNotExist:
        Tunnel.objects.create(tunnel_direction=tunnel_direction, tunnel_number=tunnel_number, location=location)

    return location.json


@command("POST")
def edit_name(new_name, loc_name, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param new_name: New Location Name
    :param loc_name: Old Location name
    :param discord_uuid: Discord UUID
    :param mc_uuid: MC UUID
    :return: Edited Location
    :raises: PlayerNotFound, LocationLookupError, EntryNameNotUniqueError, NoLocationsInDatabase
    :help: Edits the name of a location
    """

    player = get_player(discord_uuid=discord_uuid, mc_uuid=mc_uuid)

    if Location.objects.filter(name__iexact=new_name).count():
        raise EntryNameNotUniqueError

    location = get_location(player, loc_name)

    location.name = new_name
    location.save()
    return location.json


@command("POST")
def delete_item(item, shop_name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param item: Item name to delete
    :param shop_name: Shop selling item, can be None if the user only has one shop
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: Shop where the item was deleted from
    :raises: PlayerNotFound, LocationLookupError, EntryNameNotUniqueError, NoLocationsInDatabase, ItemNotFound
    :help: Deletes an item from a shop
    """

    player = get_player(discord_uuid=discord_uuid, mc_uuid=mc_uuid)

    shop = get_location(player, shop_name, Shop)

    delete_list = ItemListing.objects.filter(item_name__iexact=item, shop=shop).all()

    if len(delete_list) == 0:
        raise ItemNotFound

    delete_list.delete()

    return shop.json


@command("POST")
def delete_resource(resource_name, farm_name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param resource: resource to delete
    :param farm_name: Farm with resource, can be None if the user only has one farm
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: PublicFarm where the resource was deleted from
    :raises: PlayerNotFound, LocationLookupError, EntryNameNotUniqueError, NoLocationsInDatabase, ItemNotFound
    :help: Deletes a resource from a farm
    """

    player = get_player(discord_uuid=discord_uuid, mc_uuid=mc_uuid)

    farm = get_location(player, farm_name, PublicFarm)

    delete_list = Resource.objects.filter(resource_name=resource_name, farm=farm).all()

    if len(delete_list) == 0:
        raise ResourceNotFoundError

    delete_list.delete()

    return farm.json


@command("GET")
def me(discord_uuid=None, mc_uuid=None):
    """
    :request: GET
    :param discord_uuid: Discord UUID
    :param mc_uuid: MC UUID
    :return: Returns a list of all the locations owned by a user
    :raises: NoLocationsInDatabase, PlayerNotFound
    :help: Find all the locations in the database
    """

    try:
        player = get_player(discord_uuid=discord_uuid, mc_uuid=mc_uuid)
    except Player.DoesNotExist:
        raise PlayerNotFound

    locations = Location.objects.filter(owner=player).all()

    if len(locations) == 0:
        raise NoLocationsInDatabase

    return objects_list_to_json(locations)


@command("POST")
def restock(item_name, shop_name=None, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param item_name: Item to restock
    :param shop_name: Shop the item is in, can be none if the only one location is owned by the user
    :param discord_uuid: Discord UUID
    :param mc_uuid: Minecraft UUID
    :return: List of items updated
    :raises: PlayerNotFound, LocationLookupError, EntryNameNotUniqueError, NoLocationsInDatabase, ItemNotFound
    :help: Restocks items matching the item name in your shop
    """
    owner = get_player(discord_uuid, mc_uuid)
    shop = get_location(owner, shop_name, Shop)

    items = ItemListing.objects.filter(item_name__iexact=item_name, shop=shop).all()

    if len(items) == 0:
        raise ItemNotFound
    else:
        for item in items:
            item.date_restocked = datetime.datetime.now()
            item.save()

    return objects_list_to_json(items)


@command("POST")
def add_owner(new_owner_name, location_name, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param new_owner_name: The MC username of the new owner
    :param location_name: The name of the location to add them to
    :param discord_uuid: Discord UUID of the current owner
    :param mc_uuid: MC UUID of the current owner
    :return: Update Location
    :raises: PlayerNotFound, LocationLookupError, IsOwnerError, OwnerNotFound
    :help: Adds a co-owner to a location
    """
    owner = get_player(discord_uuid, mc_uuid)

    try:
        new_owner = Player.objects.get(name__iexact=new_owner_name)
    except Player.DoesNotExist:
        raise OwnerNotFoundError

    location = get_location(owner, location_name, Location)

    if location.owner.filter(location__owner__name__iexact=new_owner_name):
        raise IsOwnerError

    location.owner.add(new_owner)

    location.save()

    return location.json


@command("POST")
def add_resident(new_resident_name, town_name, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param new_resident_name: The MC username of the new resident
    :param town_name: The name of the town to add the resident to, can be blank if the owner has one town
    :param discord_uuid: Discord UUID of the town owner
    :param mc_uuid: MC UUID of the town owner
    :return: Updated Location
    :raises: PlayerNotFound, LocationLookupError, IsResidentError, ResidentNotFoundError
    :help: Adds a resident to a town
    """
    owner = get_player(discord_uuid, mc_uuid)

    try:
        new_resident = Player.objects.get(name__iexact=new_resident_name)
    except Player.DoesNotExist:
        raise ResidentNotFoundError

    town = get_location(owner, town_name, Town)

    if town.residents.filter(Q(town__owner__name__iexact=new_resident_name) |
                             Q(name__iexact=new_resident_name)).all().count():
        raise IsResidentError

    town.residents.add(new_resident)

    town.save()

    return town.json


@command("POST")
def remove_resident(resident_name, town_name, discord_uuid=None, mc_uuid=None):
    """
    :request: POST
    :param resident_name: Name of the resident to remove
    :param town_name: Name of the town, can be blank if the owner has one town
    :param discord_uuid: Owner discord uuid
    :param mc_uuid:  Owner mc uuid
    :raises: PlayerNotFound, LocationLookupError, ResidentNotFoundError
    :return: Updated town
    :help: Removes a resident from a town
    """

    owner = get_player(discord_uuid, mc_uuid)
    town = get_location(owner, town_name, Town)
    try:
        resident = town.residents.get(name__iexact=resident_name)
    except Player.DoesNotExist:
        raise ResidentNotFoundError

    town.residents.remove(resident)

    town.save()
    return town.json