from django.db import models

from . import interfaces
from .api import make_api


class Bot(models.Model):
	"""
	Пользователь заводит ботов указывая токен. Имя бота можно получить из API.
	Из админки нужен интерфейс публикации и отключения бота, а также отображение актуального состояния.
	"""

	token = models.CharField(max_length=128, unique=True)
	username = models.CharField(max_length=64, blank=True, editable=False)
	name = models.CharField(max_length=128, blank=True, editable=False)
	interface = models.CharField(
		max_length=255, blank=True,
		help_text='Класс, определяющий поведение бота. Меняется на лету.',
	)
	is_active = models.BooleanField('опубликован', default=False, editable=False)
	update_offset = models.BigIntegerField(default=0, editable=False)
	heartbeat_at = models.DateTimeField(null=True, blank=True, editable=False)
	last_error = models.TextField(blank=True, editable=False)
	created_at = models.DateTimeField(auto_now_add=True)

	class Meta:
		verbose_name = 'бот'
		verbose_name_plural = 'боты'

	def __str__(self):
		return f'@{self.username}' if self.username else f'bot #{self.pk}'

	def api(self):
		return make_api(self.token)

	def get_interface(self) -> interfaces.Interface | None:
		cls = interfaces.get(self.interface)
		return cls() if cls else None

	def publish(self):
		"""Проверяет токен, подтягивает имя бота и включает его обслуживание раннером."""
		api = self.api()
		me = api.get_me()
		api.delete_webhook()
		self.username = me.username or ''
		self.name = me.first_name or ''
		self.is_active = True
		self.last_error = ''
		self.save()

	def revoke(self):
		self.is_active = False
		self.save(update_fields=['is_active'])


class User(models.Model):
	"""
	Когда клиент из тг пишет боту личное сообщение, то пользователя нужно завести в базу.
	"""

	tg_id = models.BigIntegerField(unique=True)
	username = models.CharField(max_length=64, blank=True)
	first_name = models.CharField(max_length=128, blank=True)
	last_name = models.CharField(max_length=128, blank=True)
	language_code = models.CharField(max_length=16, blank=True)
	created_at = models.DateTimeField(auto_now_add=True)
	updated_at = models.DateTimeField(auto_now=True)

	class Meta:
		verbose_name = 'пользователь telegram'
		verbose_name_plural = 'пользователи telegram'

	def __str__(self):
		return f'@{self.username}' if self.username else str(self.tg_id)


class Chat(models.Model):
	"""
	При добавлении бота в группу, нужно добавить запись об этом в базу.
	При удалении, нужно пометить запись как удаленную, но не удалять.

	Чат принадлежит боту: у одного пользователя с разными ботами разные чаты и разное состояние диалога.
	"""

	bot = models.ForeignKey(Bot, on_delete=models.CASCADE, related_name='chats')
	chat_id = models.BigIntegerField()
	type = models.CharField(max_length=16)
	title = models.CharField(max_length=255, blank=True)
	user = models.ForeignKey(
		User, on_delete=models.SET_NULL, null=True, blank=True, related_name='chats',
		help_text='Собеседник в личном чате',
	)
	state = models.CharField(max_length=128, blank=True, help_text='Состояние диалога, задается интерфейсом бота')
	data = models.JSONField(default=dict, blank=True, help_text='Данные диалога, задаются интерфейсом бота')
	created_at = models.DateTimeField(auto_now_add=True)
	deleted_at = models.DateTimeField(null=True, blank=True)

	class Meta:
		verbose_name = 'чат'
		verbose_name_plural = 'чаты'
		constraints = [models.UniqueConstraint(fields=['bot', 'chat_id'], name='telegram_chat_unique')]

	def __str__(self):
		return self.title or str(self.user or self.chat_id)


class Message(models.Model):
	"""
	Исходящие и входящие сообщения между администратором и пользователем.
	"""

	class Direction(models.TextChoices):
		IN = 'in', 'входящее'
		OUT = 'out', 'исходящее'

	chat = models.ForeignKey(Chat, on_delete=models.CASCADE, related_name='messages')
	user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True, related_name='messages')
	direction = models.CharField(max_length=3, choices=Direction.choices)
	message_id = models.BigIntegerField(null=True, blank=True)
	text = models.TextField(blank=True)
	created_at = models.DateTimeField(auto_now_add=True)

	class Meta:
		verbose_name = 'сообщение'
		verbose_name_plural = 'сообщения'
		ordering = ['-created_at']

	def __str__(self):
		return f'{self.get_direction_display()}: {self.text[:50]}'