name = 'Omaha Beach'
version = '1.0'
author = 'Kestrel'
description = 'infilfied by izzy'
extensions = {
     'grenadelauncher': False
  }

from pyspades.constants import *
from pyspades.server import ServerConnection
from random import choice

def get_entity_location(team, entity_id):
    if entity_id == BLUE_FLAG:
        z = team.protocol.map.get_z(100, 249)
        return (100, 249, z)
    if entity_id == BLUE_BASE:
        z = team.protocol.map.get_z (72, 216)
        return (72, 216, z)
    if entity_id == GREEN_BASE:
        z = team.protocol.map.get_z(434, 256)
        return (434, 256, z)

spawn_locations_green = [
    	(441, 166),
    	(437, 185),
    	(448, 232),
    	(448, 337),
    	(413, 379)
]

spawn_locations_blue = [
    	(177, 158),
        (210, 161),
        (202, 222),
        (203, 308),
    	(172, 353),
        (201, 369)
]

def get_spawn_location(connection):
    if connection.team is connection.protocol.green_team:
	x, y = choice(spawn_locations_green)
	z = connection.protocol.map.get_z(x, y)
	return x, y, z
    if connection.team is connection.protocol.blue_team:
	x, y = choice(spawn_locations_blue)
	z = connection.protocol.map.get_z(x, y)
	return x, y, z