@EzySingleton
@AllArgsConstructor
public class AdminEventKnowledgeDataListFetcherEventHandler
    extends AbstractEventHandler<Map<String, Object>, Object> {

    private final AdminEventRepository eventRepository;

    @SuppressWarnings("unchecked")
    @Override
    public Object handleEventData(Map<String, Object> data) {
        if (!"event".equals(data.get("dataType"))) {
            return Collections.emptyList();
        }
        Collection<Long> ids = (Collection<Long>) data.get("dataIds");
        return eventRepository.findListByIds(ids)
            .stream()
            .map(it -> EzyMapBuilder.mapBuilder()
                .put("id", it.getId())
                .put("type", "event")
                .put("title", it.getName())
                .put("slug", it.getSlug())
                .put("content", it.getDescription())
                .put("dataUrl", "/events/" + it.getSlug())
                .toMap()
            )
            .collect(Collectors.toList());
    }

    @Override
    public String getEventName() {
        return "ezyai_knowledge_data_list_fetcher";
    }
}

@EzySingleton
@AllArgsConstructor
public class AdminCourseKnowledgeDataFetcher
    implements EzyAIKnowledgeDataFetcher {

    private static final String DATA_TYPE = "course";

    private final AdminCourseRepository courseRepository;

    @Override
    public KnowledgeData getKnowledgeDataByTypeAndIdOrCode(
        String type,
        Long id,
        String code,
        Map<String, Object> parameters
    ) {
        if (!DATA_TYPE.equals(type)) {
            return null;
        }
        Course course = id != null
            ? courseRepository.findById(id)
            : courseRepository.findByField("code", code);
        return course != null ? toKnowledgeData(course) : null;
    }

    @Override
    public List<KnowledgeData> getKnowledgeDataListByTypeAndIds(
        String type,
        Collection<Long> ids,
        Map<String, Object> parameters
    ) {
        if (!DATA_TYPE.equals(type)) {
            return Collections.emptyList();
        }
        return newArrayList(
            courseRepository.findListByIds(ids),
            this::toKnowledgeData
        );
    }

    private KnowledgeData toKnowledgeData(Course course) {
        return KnowledgeData.builder()
            .id(course.getId())
            .type(DATA_TYPE)
            .title(course.getName())
            .slug(course.getCode())
            .content(course.getDescription())
            .dataUrl("/courses/" + course.getCode())
            .build();
    }
}

@EzySingleton
@AllArgsConstructor
public class AdminVectorKnowledgeSearchStrategy
    implements KnowledgeSearchStrategy {

    private final AdminVectorSearchClient vectorSearchClient;

    @Override
    public List<KnowledgeData> searchKnowledgeDataList(
        String query,
        Map<String, Object> parameters,
        int limit
    ) {
        String collectionName = toStringOrNull(
            parameters.get(META_KEY_RAG_COLLECTION_NAME)
        );
        if (isBlank(collectionName)) {
            return Collections.emptyList();
        }
        return newArrayList(
            vectorSearchClient.search(collectionName, query, limit),
            it -> KnowledgeData.builder()
                .id(it.getId())
                .type(it.getType())
                .title(it.getTitle())
                .content(it.getText())
                .dataUrl(it.getUrl())
                .chunked(true)
                .build()
        );
    }

    @Override
    public int getPriority() {
        return 0;
    }
}

@EzySingleton
@AllArgsConstructor
public class AdminHelpCenterKnowledgeDataSource
    implements KnowledgeDataSource {

    private final AdminHelpCenterClient helpCenterClient;

    @Override
    public List<KnowledgeData> searchKnowledgeDataList(
        String query,
        Map<String, Object> parameters,
        int limit
    ) {
        return newArrayList(
            helpCenterClient.searchArticles(query, limit),
            this::toKnowledgeData
        );
    }

    @Override
    public KnowledgeData getKnowledgeDataByTypeAndIdOrCode(
        String type,
        Long id,
        String code,
        Map<String, Object> parameters
    ) {
        HelpCenterArticle article = helpCenterClient.getArticle(id, code);
        return article != null ? toKnowledgeData(article) : null;
    }

    @Override
    public List<KnowledgeData> getKnowledgeDataListByTypeAndIds(
        String type,
        Collection<Long> ids,
        Map<String, Object> parameters
    ) {
        return newArrayList(
            helpCenterClient.getArticles(ids),
            this::toKnowledgeData
        );
    }

    @Override
    public String getName() {
        return "HELP_CENTER";
    }

    @Override
    public String getDisplayName() {
        return "Help center";
    }

    private KnowledgeData toKnowledgeData(HelpCenterArticle article) {
        return KnowledgeData.builder()
            .id(article.getId())
            .type("help_center_article")
            .title(article.getTitle())
            .content(article.getBody())
            .dataUrl(article.getUrl())
            .build();
    }
}

@EzySingleton
@AllArgsConstructor
public class AdminWorkspaceKnowledgeDataSourceProvider
    implements KnowledgeDataSourceProvider {

    private static final String PREFIX = "WORKSPACE_";

    private final AdminWorkspaceService workspaceService;
    private final AdminWorkspaceSearchClient searchClient;

    @Override
    public KnowledgeDataSource provideKnowledgeDataSource(
        String datasourceName
    ) {
        if (!datasourceName.startsWith(PREFIX)) {
            return null;
        }
        long workspaceId = toLongOrZero(
            datasourceName.substring(PREFIX.length())
        );
        return workspaceId > 0
            ? new WorkspaceKnowledgeDataSource(
                datasourceName,
                workspaceId,
                searchClient
            )
            : null;
    }

    @Override
    public List<KnowledgeDataSourceNameModel> getKnowledgeDataSourceNames() {
        return newArrayList(
            workspaceService.getActiveWorkspaces(),
            it -> KnowledgeDataSourceNameModel.builder()
                .name(PREFIX + it.getId())
                .displayName(it.getName())
                .build()
        );
    }
}