from django.contrib import admin
from .models import (
    Institution, Author, Affiliation, Domain, Field, Subfield, Topic,
    AuthorTopic, Work, AuthorYearlyStats, Concept, AuthorConcept
)


admin.site.site_header = "BridgeMentor Admin"
admin.site.site_title = "BridgeMentor Data Management"
admin.site.index_title = "BridgeMentor Dashboard"


@admin.register(Institution)
class InstitutionAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'ror_id', 'country_code', 'institution_type')
    search_fields = ('name', 'ror_id')
    list_filter = ('country_code', 'institution_type')


@admin.register(Author)
class AuthorAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'orcid', 'h_index',
                    'i10_index', 'cited_by_count', 'works_count', 'updated_at', 'created_at')
    search_fields = ('name', 'orcid')
    list_filter = ('h_index',)


@admin.register(Affiliation)
class AffiliationAdmin(admin.ModelAdmin):
    list_display = ('author', 'institution', 'year', 'is_last_known')
    list_filter = ('year', 'is_last_known', 'institution')
    search_fields = ('author__name', 'institution__name')


@admin.register(Domain)
class DomainAdmin(admin.ModelAdmin):
    list_display = ('id', 'name')
    search_fields = ('name',)


@admin.register(Field)
class FieldAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'domain')
    list_filter = ('domain',)
    search_fields = ('name',)


@admin.register(Subfield)
class SubfieldAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'field')
    list_filter = ('field',)
    search_fields = ('name',)


@admin.register(Topic)
class TopicAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'subfield')
    list_filter = ('subfield',)
    search_fields = ('name',)


@admin.register(AuthorTopic)
class AuthorTopicAdmin(admin.ModelAdmin):
    list_display = ('author', 'topic', 'count', 'share_value')
    list_filter = ('topic',)
    search_fields = ('author__name', 'topic__name')


@admin.register(Work)
class WorkAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'author', 'year', 'cited_by_count')
    list_filter = ('year',)
    search_fields = ('title', 'author__name')


@admin.register(AuthorYearlyStats)
class AuthorYearlyStatsAdmin(admin.ModelAdmin):
    list_display = ('author', 'year', 'works_count', 'cited_by_count')
    list_filter = ('year',)
    search_fields = ('author__name',)


@admin.register(Concept)
class ConceptAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'level', 'score')
    search_fields = ('name',)


@admin.register(AuthorConcept)
class AuthorConceptAdmin(admin.ModelAdmin):
    list_display = ('author', 'concept', 'level', 'score')
    search_fields = ('author__name', 'concept__name')