from pygame import *

init()

# ------------------------------------------------------------------------------------------

# set up code


# Set up screen


width = 1200

height = 700

screen = display.set_mode((width, height))

clock = time.Clock()

# ------------------------------------------------------------------------------------------

# load alien and ship pictures and scale it!


shipImage = image.load("ship.xcf")

shipImage = transform.scale(shipImage, (60, 60))

alienImage = image.load("alien.jpeg")

alienImage = transform.scale(alienImage, (60, 60))

# ------------------------------------------------------------------------------------------

# alien properties


x = 350  # position of first alien

y = 20

alienList = []

count = 0

row = 0



# ------------------------------------------------------------------------------------------

# colours


black = (0, 0, 0)

red = (200, 0, 0)

bright_red = (255, 0, 0)

white = (255, 255, 255)

green = (0, 200, 0)

bright_green = (0, 255, 0)


# ------------------------------------------------------------------------------------------

# function to quit the game


def quitgame():
    quit()


# ------------------------------------------------------------------------------------------

# function to create aliens


def createAliens(count, row):
    x = 350

    y = 100

    while count < 11 and row < 5:  # count = number of aliens, row = number of rows

        alien1 = Rect(x, y, 60, 60)

        alienList.append(alien1)

        x = x + 60  # distance of aliens on x-axis

        count = count + 1

        if count == 11:
            count = 0

            y = y + 70  # distance of aliens on y-axis

            row = row + 1

            x = 350

    return alienList


aliens = createAliens(count, row)  # calls function







# ------------------------------------------------------------------------------------------

# start of game and the entire game in one function

def game_loop():
    global pause

    # ship properties
    # spawn point


    sx = 600

    sy = 615

    x_change = 0

    # create ship and its rect

    ship = Rect(sx, sy, 60, 60)

    gameOver = False

    score = 0


    dx = 5  # direction

    dy = 0

    while gameOver == False:


        def player (sx,sy):
            screen.blit(shipImage, (sx, sy))

        # ------------------------------------------------------------------------------------------

        # Button inputs

        for e in event.get():

            if e.type == constants.QUIT:  # to quit game (close tab)

                gameOver = True

            if e.type == KEYDOWN:  # to pause game (p)

                if e.key == K_p:
                    pause = True

                    paused()

                if e.key == K_RIGHT:
                    x_change = 8

                if e.key == K_LEFT:
                    x_change = -8


            if e.type == KEYUP:
                if e.key == K_RIGHT:
                    x_change = 0

                if e.key == K_LEFT:
                    x_change = 0


        screen.fill((0, 0, 0))
        # ------------------------------------------------------------------------------------------
        # Ship movement
        sx = sx + x_change

        player(sx,sy)
        if sx > width - 60:
            sx = width - 60
        if sx < 0:
            sx = 0

        # ------------------------------------------------------------------------------------------

        # draw aliens

        for a in alienList:
            screen.blit(alienImage, a)


        # ------------------------------------------------------------------------------------------

        # moving aliens

        for alien in alienList:

            alien.move_ip(dx, dy)

            if alien.x + alien.w >= 1200 or alien.x <= 0:

                # alien.x is the left side of image and alien.w is the width of image

                dx = dx * -1

                # moves alien down

                for alien in alienList:
                    dy = 5

                    alien.move_ip(dx, dy)

                dy = 0

            alien.move_ip(dx, 0)

            display.flip()
        clock.tick(60)


# ------------------------------------------------------------------------------------------


# Functions .

def text_objects(text, font):
    textSurface = font.render(text, True, black)

    return textSurface, textSurface.get_rect()


def message_display(text):
    largeText = pygame.font.Font('freesansbold.ttf', 115)

    TextSurf, TextRect = text_objects(text, largeText)

    TextRect.center = ((width / 2), (height / 2))

    screen.blit(TextSurface, TextRect)

    screen.update()


def things(thingx, thingy, thingw, thingh, colour):
    draw.rect(screen, colour, [thingx, thingy, thingw, thingh])


def button(msg, x, y, w, h, ic, ac, action=None):
    m = mouse.get_pos()

    click = mouse.get_pressed()

    if x + w > m[0] > x and y + h > m[1] > y:

        draw.rect(screen, ac, (x, y, w, h))

        if click[0] == 1 and action != None:

            action()

            if action == "play":

                game_loop()



            elif action == "quit":

                quit()

    else:

        draw.rect(screen, ic, (x, y, w, h))

    smallText = font.Font("freesansbold.ttf", 20)

    textSurf, textRect = text_objects(msg, smallText)

    textRect.center = ((x + (w / 2)), (y + (h / 2)))

    screen.blit(textSurf, textRect)


pause = False


def unpause():
    global pause

    pause = False


# ------------------------------------------------------------------------------------------


# pause function - credit to sentdex for tutorial


def paused():
    pause = True

    while pause:

        for e in event.get():

            if e.type == QUIT:
                quitgame()

        screen.fill((white))

        largeText = font.Font('freesansbold.ttf', 115)

        TextSurf, TextRect = text_objects("PAUSED", largeText)

        TextRect.center = ((width / 2), (height / 2))

        screen.blit(TextSurf, TextRect)

        button("CONTINUE?", 150, 450, 150, 50, green, bright_green, game_loop)

        button("QUIT", 850, 450, 100, 50, red, bright_red, "quit")

        display.update()

        clock.tick(15)

    # ------------------------------------------------------------------------------------------


# Credit to Harsh for helping me with this , and credit to sentdex for tutorial


def game_intro():
    intro = True

    while intro:

        for e in event.get():

            if e.type == QUIT:
                quitgame()

        screen.fill((white))

        largeText = font.Font('freesansbold.ttf', 115)

        TextSurf, TextRect = text_objects("Skerritt Invaders", largeText)

        TextRect.center = ((width / 2), (height / 2))

        screen.blit(TextSurf, TextRect)

        button("PLAY", 150, 450, 100, 50, green, bright_green, game_loop)

        button("QUIT", 850, 450, 100, 50, red, bright_red, quit)

        display.update()

        clock.tick(15)


game_intro()