import random
import pygame
import sys
from pygame.locals import *

#Tao khung hien thi game

width = 1000
height = 500

display_surf = pygame.display.set_mode((width, height))
pygame.display.set_caption("Shooter")

WHITE = (255, 255, 255)

BLACK = (0, 0, 0)

fps_clock = pygame.time.Clock()

#Tau

class Ship:

    def __init__(self, w, h, x, y, speed):
        self.width = w
        self.height = h
        self.x = x
        self.y = y
        self.dir_x = 1
        self.dir_y = 0
        self.speed = speed

    def draw(self):
        pygame.draw.rect(display_surf, WHITE, (self.x, self.y, self.width, self.height))

    def move(self):
        self.x = self.x + self.x * self.dir_x * self.speed
        self.y = self.y + self.y * self.dir_y * self.speed



class Game:

    def __init__(self, ship, speed):
        self.ship = ship
        self.speed = speed

    def draw_arena(self):
        display_surf.fill((0, 0, 0))
        pygame.draw.rect(display_surf, WHITE, (0, 0, width, height), 10)
        self.ship.draw()

    def update(self):
        self.ship.move()


def main():
    pygame.init()
    ship = Ship(20, 50, 10, height/2, 10)
    fps = 15
    die = False
    game = Game(ship, 1)

    while True:
        for event in pygame.event.get():
            if event.type == KEYDOWN:
                if event.key == K_ESCAPE:
                    die = True
                if event.key == K_w and game.ship.dir_y != 1:
                    game.ship.dir_y = -1
                    game.ship.dir_x = 0
                elif event.key == K_s and game.ship.dir_y != -1:
                    game.ship.dir_y = 1
                    game.ship.dir_x = 0
                elif event.key == K_a and game.ship.dir_x != 1:
                    game.ship.dir_y = 0
                    game.ship.dir_x = -1
                elif event.key == K_d and game.ship.dir_x != -1:
                    game.ship.dir_y = 0
                    game.ship.dir_x = 1
            if event.type == QUIT:
                die = True


        if die:
            pygame.quit()
            sys.exit()
        game.draw_arena()
        game.update()
        pygame.display.update()
        fps_clock.tick(fps)


if __name__ == "__main__":
    main()