import pygame
import sys

pygame.init()
display = pygame.display.set_mode((400, 300))

class TextInput(pygame.sprite.Sprite):
    def __init__(self, x, y, width=100, height=50, color=(0, 0, 0), bgcolor=(0,255,0), selectedColor=(0,0,255,)):
        super().__init__()
        self.text_value = ""
        self.isSelected = False
        self.color = color
        self.bgcolor = bgcolor
        self.selectedColor = selectedColor

        self.font = pygame.font.SysFont("Verdana", 20)
        self.text = self.font.render(self.text_value, True, self.color)
        self.bg = pygame.Rect(x, y, width, height)

    def render(self, display):
        self.pos = self.text.get_rect(center = (self.bg.x +self.bg.width/2, self.bg.y + self.bg.height/2))
        if self.isSelected:
            pygame.draw.rect(display, self.selectedColor, self.bg)
        else:
            pygame.draw.rect(display, self.bgcolor, self.bg)
        display.blit(self.text, self.pos)
    
    def clicked(self, mousePos):
        if self.bg.collidepoint(mousePos):
            self.ifSelected = not(self.isSelected)
            return True
        return False
    
    def update_text(self, new_text):
        temp = self.font.render(new_text, True, self.color)
        if temp.get_rect().width >= (self.bg.width - 20):
            return
        self.text_value = new_text
        self.text = temp


class CustomGroup(pygame.sprite.Group):
    def __init__(self):
        super().__init__()
        self.current = None

TextInputGroup = CustomGroup()
TextInputGroup.add(TextInput(x=100, y=100, width = 200))
TextInputGroup.add(TextInput(x=100, y=200, width = 200))

while True:
    mouse_pos = pygame.mouse.get_pos()
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        if event.type == pygame.MOUSEBUTTONDOWN:
            for textinput in TextInputGroup:
                if textinput.clicked(mouse_pos):
                    if TextInputGroup.current:
                        TextInputGroup.current.isSelected = False
                    textinput.isSelected = True
                    TextInputGroup.current = textinput
                    break
       
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_BACKSPACE:
                TextInputGroup.current.update_text(TextInputGroup.current.text_value[:-1])

            if event.key == pygame.K_RETURN:
                if TextInputGroup.current:
                    print(TextInputGroup.current.text_value)

        if event.type == pygame.TEXTINPUT:
            new_text = TextInputGroup.current.text_value + event.text
            TextInputGroup.current.update_text(new_text)
        

    for textinput in TextInputGroup:
        textinput.render(display)

    pygame.display.update()