from sqlalchemy.orm import Session
from schemas.template_schemas import TemplateCreate, TemplateFieldCreate, TemplateFieldMappingCreate
from repositories.template_repository import TemplateRepository

class TemplateService:
    def __init__(self, db: Session):
        self.repository = TemplateRepository(db)

    def get_template(self, template_id: int):
        return self.repository.get_template(template_id)

    def get_all_templates(self, skip: int = 0, limit: int = 100):
        return self.repository.get_templates(skip, limit)

    def create_template(self, template_data: TemplateCreate):
        return self.repository.create_template(template_data)

    def add_template_field(self, template_id: int, field_data: TemplateFieldCreate):
        return self.repository.add_template_field(template_id, field_data)

    def delete_template(self, template_id: int):
        return self.repository.delete_template(template_id)

    def save_mapping(self, template_id: int, mappings: list[TemplateFieldMappingCreate]):
        return self.repository.save_mapping(template_id, mappings)