from pygame import *

init()

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

# set up code


# Set up screen


width = 1200

height = 700

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

clock = time.Clock()

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

# load 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))

bulletImage = image.load("bullet.xcf")

bulletImage = transform.scale(bulletImage, (40, 50))

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

# 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 < 5 and row < 1:  # 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

main_font = font.SysFont("comicsans", 40)











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

# start of game and the entire game in one function

def game_loop():
    global pause


# ship properties
    sx = 600

    sy = 640

    x_change = 0

    ship = Rect(sx,sy,60,60)
# function to create bullets

    def createbullets(bullets,x,y):
        bullet = Rect(sx,sy+10,15,40)
        bullets.append(bullet)
        return bullets

    bullets = []

    gameOver = False




    dx = 1  # direction

    dy = 0

    score = 0
    lives = 3



    while gameOver == False:

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

        # 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 = 30

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

                if e.key == K_SPACE:
                    createbullets(bullets,x,y)


            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))
        score_label = main_font.render(f"score: {score}", 1, (255, 255, 255))
        lives_label = main_font.render(f"lives: {lives}", 1, (255, 255, 255))


        # score

        screen.blit(score_label, (10, 10))

        # lives
        screen.blit(lives_label, (1000, 10))

# ship movement


        sx = sx + x_change
        if sx > width - 60:
            sx = width - 60

        if sx < 0:
            sx = 0



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

        # draw aliens

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

        # Draw ship

        screen.blit(shipImage, (sx, sy))


# create bullet
        for b in bullets:
            screen.blit(bulletImage, b)
# move bullet
        for b in bullets:
            b.move_ip(0,-10)
            if b.y <= 0:
                bullets.remove(b)



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

        # 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 = 10

                    alien.move_ip(dx, dy)

                dy = 0

            alien.move_ip(dx, 0)


# bullet to alien hit detection

            for alien in alienList:
                for bullet in bullets:
                    if bullet.colliderect(alien):
                        bullets.remove(bullet)
                        alienList.remove(alien)
                        score = score + 1


# gameover when alien hits player

            for alien in alienList:
                if alien.colliderect(ship):
                    game_over()



            display.flip()
            clock.tick(60)
        # -----------------------------------------------------------------------------

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


# Functions .
def game_over():

    screen.fill(bright_red)

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

    TextSurf, TextRect = text_objects("GAME OVER", largeText)

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

    screen.blit(TextSurf, TextRect)

    while True:

        for e in event.get():

            if e.type == QUIT:
                quitgame()

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

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

        display.update()

        clock.tick(15)



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()



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


# 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))

        main_font = font.SysFont("comicsans", 40)

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


        TextSurf, TextRect = text_objects("Skerritt Invaders", largeText)
        TextRect.center = ((width / 2), (height / 2))
        screen.blit(TextSurf, TextRect)


        TextSurf, TextRect = text_objects("Controls:", main_font)
        TextRect.center = (200, 525)
        screen.blit(TextSurf, TextRect)

        TextSurf, TextRect = text_objects("P - Pause", main_font)
        TextRect.center = (200, 575)
        screen.blit(TextSurf, TextRect)

        TextSurf, TextRect = text_objects("Space - Shoot", main_font)
        TextRect.center = (200, 625)
        screen.blit(TextSurf, TextRect)

        TextSurf, TextRect = text_objects("Left/Right arrow keys - move", main_font)
        TextRect.center = (200, 675)
        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()