package io.github.jwdeveloper.tiktok; import io.github.jwdeveloper.tiktok.events.TikTokEvent; import io.github.jwdeveloper.tiktok.events.TikTokEventBuilder; import io.github.jwdeveloper.tiktok.events.TikTokEventConsumer; import io.github.jwdeveloper.tiktok.events.messages.*; import io.github.jwdeveloper.tiktok.exceptions.TikTokLiveException; import io.github.jwdeveloper.tiktok.handlers.TikTokEventObserver; import io.github.jwdeveloper.tiktok.handlers.TikTokMessageHandlerRegistration; import io.github.jwdeveloper.tiktok.http.TikTokApiService; import io.github.jwdeveloper.tiktok.http.TikTokCookieJar; import io.github.jwdeveloper.tiktok.http.TikTokHttpClient; import io.github.jwdeveloper.tiktok.http.TikTokHttpRequestFactory; import io.github.jwdeveloper.tiktok.listener.TikTokEventListener; import io.github.jwdeveloper.tiktok.listener.TikTokListenersManager; import io.github.jwdeveloper.tiktok.live.LiveClient; import io.github.jwdeveloper.tiktok.websocket.TikTokWebSocketClient; import java.time.Duration; import java.util.ArrayList; import java.util.List; import java.util.function.Consumer; import java.util.logging.Level; import java.util.logging.Logger; public class TikTokLiveClientBuilder implements TikTokEventBuilder { private final ClientSettings clientSettings; private final Logger logger; private final TikTokEventObserver tikTokEventHandler; private final List listeners; public TikTokLiveClientBuilder(String userName) { this.tikTokEventHandler = new TikTokEventObserver(); this.clientSettings = Constants.DefaultClientSettings(); this.clientSettings.setHostName(userName); this.logger = Logger.getLogger(TikTokLive.class.getName()); this.listeners = new ArrayList<>(); } public TikTokLiveClientBuilder configure(Consumer consumer) { consumer.accept(clientSettings); return this; } public TikTokLiveClientBuilder addListener(TikTokEventListener listener) { listeners.add(listener); return this; } private void validate() { if (clientSettings.getTimeout() == null) { clientSettings.setTimeout(Duration.ofSeconds(Constants.DEFAULT_TIMEOUT)); } if (clientSettings.getClientLanguage() == null || clientSettings.getClientLanguage().equals("")) { clientSettings.setClientLanguage(Constants.DefaultClientSettings().getClientLanguage()); } if (clientSettings.getHostName() == null || clientSettings.getHostName().equals("")) { throw new TikTokLiveException("HostName can not be null"); } var params = clientSettings.getClientParameters(); params.put("app_language", clientSettings.getClientLanguage()); params.put("webcast_language", clientSettings.getClientLanguage()); logger.setLevel(clientSettings.getLogLevel()); if(clientSettings.isPrintToConsole() && clientSettings.getLogLevel() == Level.OFF) { logger.setLevel(Level.ALL); } } public LiveClient build() { validate(); var tiktokRoomInfo = new TikTokRoomInfo(); tiktokRoomInfo.setUserName(clientSettings.getHostName()); var listenerManager = new TikTokListenersManager(listeners, tikTokEventHandler); var cookieJar = new TikTokCookieJar(); var requestFactory = new TikTokHttpRequestFactory(cookieJar); var apiClient = new TikTokHttpClient(cookieJar, requestFactory); var apiService = new TikTokApiService(apiClient, logger, clientSettings); var giftManager = new TikTokGiftManager(); var webResponseHandler = new TikTokMessageHandlerRegistration(tikTokEventHandler, clientSettings, logger, giftManager, tiktokRoomInfo); var webSocketClient = new TikTokWebSocketClient(logger, cookieJar, clientSettings, webResponseHandler, tikTokEventHandler); return new TikTokLiveClient(tiktokRoomInfo, apiService, webSocketClient, giftManager, tikTokEventHandler, clientSettings, listenerManager, logger); } public LiveClient buildAndRun() { var client = build(); client.connect(); return client; } public TikTokLiveClientBuilder onUnhandledSocial( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokUnhandledSocialEvent.class, event); return this; } public TikTokLiveClientBuilder onLinkMicFanTicket( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokLinkMicFanTicketEvent.class, event); return this; } public TikTokLiveClientBuilder onEnvelope(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokEnvelopeEvent.class, event); return this; } public TikTokLiveClientBuilder onShopMessage(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokShopMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onDetectMessage( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokDetectMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onLinkLayerMessage( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokLinkLayerMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onConnected(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokConnectedEvent.class, event); return this; } public TikTokLiveClientBuilder onCaption(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokCaptionEvent.class, event); return this; } public TikTokLiveClientBuilder onQuestion(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokQuestionEvent.class, event); return this; } public TikTokLiveClientBuilder onRoomPinMessage( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokRoomPinMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onRoomMessage(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokRoomMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onLivePaused(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokLivePausedEvent.class, event); return this; } public TikTokLiveClientBuilder onLike(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokLikeEvent.class, event); return this; } public TikTokLiveClientBuilder onLinkMessage(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokLinkMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onBarrageMessage( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokBarrageMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onGiftMessage(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokGiftMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onLinkMicArmies( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokLinkMicArmiesEvent.class, event); return this; } public TikTokLiveClientBuilder onEmote(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokEmoteEvent.class, event); return this; } public TikTokLiveClientBuilder onUnauthorizedMember( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokUnauthorizedMemberEvent.class, event); return this; } public TikTokLiveClientBuilder onInRoomBanner( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokInRoomBannerEvent.class, event); return this; } public TikTokLiveClientBuilder onLinkMicMethod( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokLinkMicMethodEvent.class, event); return this; } public TikTokLiveClientBuilder onSubscribe(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokSubscribeEvent.class, event); return this; } public TikTokLiveClientBuilder onPollMessage(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokPollMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onFollow(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokFollowEvent.class, event); return this; } public TikTokLiveClientBuilder onRoomViewerData( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokRoomViewerDataEvent.class, event); return this; } public TikTokLiveClientBuilder onGoalUpdate(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokGoalUpdateEvent.class, event); return this; } public TikTokLiveClientBuilder onComment(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokCommentEvent.class, event); return this; } public TikTokLiveClientBuilder onRankUpdate(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokRankUpdateEvent.class, event); return this; } public TikTokLiveClientBuilder onIMDelete(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokIMDeleteEvent.class, event); return this; } public TikTokLiveClientBuilder onLiveEnded(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokLiveEndedEvent.class, event); return this; } public TikTokLiveClientBuilder onError(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokErrorEvent.class, event); return this; } public TikTokLiveClientBuilder onUnhandled(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokUnhandledWebsocketMessageEvent.class, event); return this; } public TikTokLiveClientBuilder onJoin(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokJoinEvent.class, event); return this; } public TikTokLiveClientBuilder onRankText(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokRankTextEvent.class, event); return this; } public TikTokLiveClientBuilder onShare(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokShareEvent.class, event); return this; } public TikTokLiveClientBuilder onUnhandledMember( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokUnhandledMemberEvent.class, event); return this; } public TikTokLiveClientBuilder onSubNotify(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokSubNotifyEvent.class, event); return this; } public TikTokLiveClientBuilder onLinkMicBattle( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokLinkMicBattleEvent.class, event); return this; } public TikTokLiveClientBuilder onDisconnected( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokDisconnectedEvent.class, event); return this; } public TikTokLiveClientBuilder onGiftBroadcast( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokGiftBroadcastEvent.class, event); return this; } public TikTokLiveClientBuilder onUnhandledControl( TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokUnhandledControlEvent.class, event); return this; } public TikTokLiveClientBuilder onEvent(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokEvent.class, event); return this; } @Override public TikTokLiveClientBuilder onWebsocketMessage(TikTokEventConsumer event) { tikTokEventHandler.subscribe(TikTokWebsocketMessageEvent.class, event); return this; } }