from django.contrib import admin
from .models import Company, CollectiveAgreement, WorkingTimePolicy, VacationPolicy, OnboardingProgress

@admin.register(Company)
class CompanyAdmin(admin.ModelAdmin):
    list_display = ['name', 'industry', 'company_size', 'primary_location', 'created_at']
    list_filter = ['industry', 'company_size', 'created_at']
    search_fields = ['name', 'primary_location']
    readonly_fields = ['id', 'created_at', 'updated_at']

@admin.register(CollectiveAgreement)
class CollectiveAgreementAdmin(admin.ModelAdmin):
    list_display = ['company', 'agreement_type', 'organization', 'overtime_2h_rate', 'night_work_rate']
    list_filter = ['agreement_type', 'created_at']
    search_fields = ['company__name', 'organization']
    readonly_fields = ['id', 'created_at', 'updated_at']

@admin.register(WorkingTimePolicy)
class WorkingTimePolicyAdmin(admin.ModelAdmin):
    list_display = ['company', 'use_flextime', 'use_time_banking', 'lunch_break_paid', 'require_manager_approval']
    list_filter = ['use_flextime', 'use_time_banking', 'lunch_break_paid', 'require_manager_approval']
    search_fields = ['company__name']
    readonly_fields = ['id', 'created_at', 'updated_at']

@admin.register(VacationPolicy)
class VacationPolicyAdmin(admin.ModelAdmin):
    list_display = ['company', 'vacation_accrual_type', 'track_vacation_days', 'include_finnish_holidays']
    list_filter = ['vacation_accrual_type', 'track_vacation_days', 'include_finnish_holidays']
    search_fields = ['company__name']
    readonly_fields = ['id', 'created_at', 'updated_at']

@admin.register(OnboardingProgress)
class OnboardingProgressAdmin(admin.ModelAdmin):
    list_display = ['company', 'current_step', 'is_completed', 'completed_at']
    list_filter = ['current_step', 'is_completed', 'created_at']
    search_fields = ['company__name']
    readonly_fields = ['id', 'created_at', 'updated_at']