001/**
002 * Copyright (c) 2025-2026, Michael Yang 杨福海 (fuhai999@gmail.com).
003 * <p>
004 * Licensed under the GNU Lesser General Public License (LGPL) ,Version 3.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 * <p>
008 * http://www.gnu.org/licenses/lgpl-3.0.txt
009 * <p>
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package dev.tinyflow.core.searchengine;
017
018import dev.tinyflow.core.searchengine.impl.BochaaiSearchEngineImpl;
019
020import java.util.ArrayList;
021import java.util.List;
022
023public class SearchEngineManager {
024
025    public List<SearchEngineProvider> providers = new ArrayList<>();
026
027    private static class ManagerHolder {
028        private static final SearchEngineManager INSTANCE = new SearchEngineManager();
029    }
030
031    private SearchEngineManager() {
032        BochaaiSearchEngineImpl bochaaiSearchEngine = new BochaaiSearchEngineImpl();
033        providers.add(id -> {
034            if ("bocha".equals(id) || "bochaai".equals(id)) {
035                return bochaaiSearchEngine;
036            }
037            return null;
038        });
039    }
040
041    public static SearchEngineManager getInstance() {
042        return ManagerHolder.INSTANCE;
043    }
044
045    public void registerProvider(SearchEngineProvider provider) {
046        providers.add(provider);
047    }
048
049    public void removeProvider(SearchEngineProvider provider) {
050        providers.remove(provider);
051    }
052
053    public SearchEngine geSearchEngine(Object searchEngineId) {
054        for (SearchEngineProvider provider : providers) {
055            SearchEngine searchEngine = provider.getSearchEngine(searchEngineId);
056            if (searchEngine != null) {
057                return searchEngine;
058            }
059        }
060        return null;
061    }
062}