import math
import os
from PIL import Image, ImageDraw, ImageFont

def add_watermark(image, watermark_text, font_path, font_size, opacity, color, right_margin, bottom_margin):
    # Load the custom font
    font = ImageFont.truetype(font_path, font_size)

    # Create a drawing context
    draw = ImageDraw.Draw(image)

    # Get the image dimensions
    image_width, image_height = image.size

    # Calculate the position for the watermark
    text_bbox = draw.textbbox((0, 0), watermark_text, font=font)
    text_width = text_bbox[2] - text_bbox[0]
    text_height = text_bbox[3] - text_bbox[1] + math.floor(font_size / 2)
    x = image_width - text_width - int(image_width * right_margin)
    y = image_height - text_height - int(image_height * bottom_margin)  # Добавляем отступ снизу

    # Create a new image for the watermark with alpha channel
    watermark = Image.new('RGBA', (text_width, text_height), (0, 0, 0, 0))
    watermark_draw = ImageDraw.Draw(watermark)

    # Convert the color tuple to a hexadecimal color string
    hex_color = "#%02x%02x%02x" % color
    
    print(color, hex_color)

    # Draw the watermark text on the new image
    watermark_draw.text((0, 0), watermark_text, font=font, fill=color + (int(255 * opacity),))

    # Paste the watermark onto the original image
    image.paste(watermark, (x, y), mask=watermark)



def process_image(image_path, output_dir, watermark_text=None, font_path=None, font_size=30, opacity=0.5,
                  color="#000000", right_margin=0.05, bottom_margin=0.02):
    # Convert the color string to a tuple
    color_tuple = tuple(int(color.lstrip('#')[i:i+2], 16) for i in (0, 2, 4))
    
    # Create output folder 
    os.makedirs(output_dir, exist_ok=True)

    # Открываем исходное изображение
    with Image.open(image_path) as img:
        # Конвертируем изображение в режим RGBA, если необходимо
        if img.mode != 'RGBA':
            img = img.convert('RGBA')

        # Добавляем водяной знак на исходное изображение, если указан текст
        if watermark_text and font_path:
            add_watermark(img, watermark_text, font_path, font_size, opacity, color_tuple, right_margin, bottom_margin)

        # Сохраняем исходное изображение с водяным знаком
        watermarked_file = os.path.join(output_dir, 'watermarked.png')
        img.save(watermarked_file, format='PNG')

        # Создаем миниатюру изображения
        thumbnail_size = (200, 200)  # Укажите нужный размер миниатюры
        img.thumbnail(thumbnail_size)

        # Сохраняем миниатюру
        thumbnail_file = os.path.join(output_dir, 'thumbnail.png')
        img.save(thumbnail_file, format='PNG')
        
    # Возвращаем список с путями к полученным файлам
    return [watermarked_file, thumbnail_file]