1
0
mirror of https://github.com/chatopera/cosin.git synced 2025-06-29 07:11:38 +08:00
cosin/public/plugins/chatbot/classes/ChatbotComposer.java
2021-02-10 17:14:00 +08:00

60 lines
1.9 KiB
Java

package com.chatopera.cc.plugins.chatbot;
import com.chatopera.cc.basic.Constants;
import com.chatopera.cc.basic.plugins.PluginRegistry;
import com.chatopera.cc.socketio.message.ChatMessage;
import com.chatopera.compose4j.Composer;
import com.chatopera.compose4j.Middleware;
import com.chatopera.compose4j.exception.Compose4jRuntimeException;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.annotation.Configuration;
import org.springframework.stereotype.Component;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.annotation.PostConstruct;
@Component
public class ChatbotComposer implements ApplicationContextAware {
private final static Logger logger = LoggerFactory.getLogger(
ChatbotComposer.class);
private static ApplicationContext applicationContext;
private Composer<ChatbotContext> composer;
@Autowired
private PluginRegistry pluginRegistry;
@Override
public void setApplicationContext(ApplicationContext ac) throws BeansException {
applicationContext = ac;
}
@PostConstruct
public void postConstruct() {
applicationContext.getBeansWithAnnotation(Configuration.class);
composer = new Composer<>();
pluginRegistry.getPlugin(Constants.CSKEFU_MODULE_MESSENGER).ifPresent(p -> {
composer.use((Middleware) applicationContext.getBean(
p.getPluginId() + PluginRegistry.PLUGIN_CHATBOT_MESSAGER_SUFFIX));
});
}
public void handle(ChatMessage resp) {
ChatbotContext ctx = new ChatbotContext();
ctx.setResp(resp);
try {
composer.handle(ctx);
} catch (Compose4jRuntimeException e) {
logger.info("[chatbot send] error", e);
}
}
}