You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

ChannelEventHandler.java 14KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368
  1. /*
  2. * Copyright (c) 2006-2017 DMDirc Developers
  3. *
  4. * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
  5. * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
  6. * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
  7. * permit persons to whom the Software is furnished to do so, subject to the following conditions:
  8. *
  9. * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
  10. * Software.
  11. *
  12. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
  13. * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS
  14. * OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  15. * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  16. */
  17. package com.dmdirc;
  18. import com.dmdirc.events.ChannelActionEvent;
  19. import com.dmdirc.events.ChannelCtcpEvent;
  20. import com.dmdirc.events.ChannelGotNamesEvent;
  21. import com.dmdirc.events.ChannelGotTopicEvent;
  22. import com.dmdirc.events.ChannelJoinEvent;
  23. import com.dmdirc.events.ChannelKickEvent;
  24. import com.dmdirc.events.ChannelListModesRetrievedEvent;
  25. import com.dmdirc.events.ChannelMessageEvent;
  26. import com.dmdirc.events.ChannelModeChangeEvent;
  27. import com.dmdirc.events.ChannelModeNoticeEvent;
  28. import com.dmdirc.events.ChannelModesDiscoveredEvent;
  29. import com.dmdirc.events.ChannelNickChangeEvent;
  30. import com.dmdirc.events.ChannelNoModesDiscoveredEvent;
  31. import com.dmdirc.events.ChannelNoTopicEvent;
  32. import com.dmdirc.events.ChannelNoticeEvent;
  33. import com.dmdirc.events.ChannelPartEvent;
  34. import com.dmdirc.events.ChannelQuitEvent;
  35. import com.dmdirc.events.ChannelSelfModeChangeEvent;
  36. import com.dmdirc.events.ChannelSelfNickChangeEvent;
  37. import com.dmdirc.events.ChannelSelfPartEvent;
  38. import com.dmdirc.events.ChannelTopicChangeEvent;
  39. import com.dmdirc.events.ChannelTopicUnsetEvent;
  40. import com.dmdirc.events.ChannelUserAwayEvent;
  41. import com.dmdirc.events.ChannelUserBackEvent;
  42. import com.dmdirc.interfaces.Connection;
  43. import com.dmdirc.events.eventbus.EventBus;
  44. import com.dmdirc.parser.common.AwayState;
  45. import com.dmdirc.parser.events.ChannelCTCPEvent;
  46. import com.dmdirc.parser.events.ChannelListModeEvent;
  47. import com.dmdirc.parser.events.ChannelNamesEvent;
  48. import com.dmdirc.parser.events.ChannelTopicEvent;
  49. import com.dmdirc.parser.events.OtherAwayStateEvent;
  50. import com.dmdirc.parser.interfaces.ChannelClientInfo;
  51. import com.dmdirc.parser.interfaces.ChannelInfo;
  52. import com.dmdirc.parser.interfaces.Parser;
  53. import com.google.common.base.Strings;
  54. import java.time.LocalDateTime;
  55. import java.time.ZoneOffset;
  56. import java.util.Optional;
  57. import java.util.stream.Collectors;
  58. import javax.annotation.Nonnull;
  59. import net.engio.mbassy.listener.Handler;
  60. /**
  61. * Handles events for channel objects.
  62. */
  63. public class ChannelEventHandler extends EventHandler {
  64. /** The channel that owns this event handler. */
  65. private final Channel owner;
  66. /** Event bus to send events on. */
  67. private final EventBus eventBus;
  68. private final GroupChatUserManager groupChatUserManager;
  69. public ChannelEventHandler(final Channel owner, final EventBus eventBus,
  70. final GroupChatUserManager groupChatUserManager) {
  71. this.owner = owner;
  72. this.eventBus = eventBus;
  73. this.groupChatUserManager = groupChatUserManager;
  74. }
  75. @Nonnull
  76. @Override
  77. protected Connection getConnection() {
  78. return owner.getConnection().get();
  79. }
  80. /**
  81. * Determines if the specified client represents us.
  82. *
  83. * @param client The client to be tested
  84. *
  85. * @return True if the client is ourself, false otherwise.
  86. */
  87. protected boolean isMyself(final ChannelClientInfo client) {
  88. return getConnection().getParser().map(Parser::getLocalClient)
  89. .map(c -> client.getClient().equals(c)).orElse(false);
  90. }
  91. @Handler
  92. public void onChannelMessage(final com.dmdirc.parser.events.ChannelMessageEvent event) {
  93. if (!checkChannel(event.getChannel())) {
  94. return;
  95. }
  96. eventBus.publishAsync(new ChannelMessageEvent(
  97. event.getDate(),
  98. owner, groupChatUserManager.getUserFromClient(event.getClient(), owner),
  99. event.getMessage()));
  100. }
  101. @Handler
  102. public void onChannelGotNames(final ChannelNamesEvent event) {
  103. if (!checkChannel(event.getChannel())) {
  104. return;
  105. }
  106. owner.setClients(event.getChannel().getChannelClients().stream()
  107. .map(client -> groupChatUserManager.getUserFromClient(client, owner))
  108. .collect(Collectors.toList()));
  109. eventBus.publishAsync(new ChannelGotNamesEvent(event.getDate(), owner));
  110. }
  111. @Handler
  112. public void onChannelTopic(final ChannelTopicEvent event) {
  113. if (!checkChannel(event.getChannel())) {
  114. return;
  115. }
  116. final ChannelInfo channel = event.getChannel();
  117. final LocalDateTime date = event.getDate();
  118. final Topic topic = Topic.create(channel.getTopic(),
  119. owner.getUser(getConnection().getUser(channel.getTopicSetter())).orElse(null),
  120. LocalDateTime.ofEpochSecond(channel.getTopicTime(), 0, ZoneOffset.UTC));
  121. if (event.isJoinTopic()) {
  122. if (Strings.isNullOrEmpty(channel.getTopic())) {
  123. eventBus.publishAsync(new ChannelNoTopicEvent(owner));
  124. } else {
  125. eventBus.publishAsync(new ChannelGotTopicEvent(owner, topic,
  126. owner.getConnection().get().getUser(channel.getTopicSetter())));
  127. }
  128. } else {
  129. if (Strings.isNullOrEmpty(channel.getTopic())) {
  130. eventBus.publishAsync(new ChannelTopicUnsetEvent(date, owner,
  131. owner.getUser(owner.getConnection().get()
  132. .getUser(channel.getTopicSetter())).orElse(null)));
  133. } else {
  134. eventBus.publishAsync(new ChannelTopicChangeEvent(date, owner, topic,
  135. topic.getClient().get()));
  136. }
  137. }
  138. final Optional<Topic> currentTopic = owner.getCurrentTopic();
  139. final boolean hasNewTopic = !Strings.isNullOrEmpty(channel.getTopic());
  140. if (!event.isJoinTopic()
  141. || !currentTopic.isPresent() && hasNewTopic
  142. || currentTopic.isPresent() && !channel.getTopic().equals(
  143. owner.getCurrentTopic().get().getTopic())) {
  144. // Only add the topic if:
  145. // - It's being set while we're in the channel (rather than discovered on join), or
  146. // - We think the current topic is empty and are discovering a new one, or
  147. // - The newly discovered topic is different to what we thought the current topic was.
  148. owner.addTopic(topic);
  149. }
  150. }
  151. @Handler
  152. public void onChannelJoin(final com.dmdirc.parser.events.ChannelJoinEvent event) {
  153. if (!checkChannel(event.getChannel())) {
  154. return;
  155. }
  156. eventBus.publishAsync(new ChannelJoinEvent(
  157. event.getDate(), owner,
  158. groupChatUserManager.getUserFromClient(event.getClient(), owner)));
  159. owner.addClient(groupChatUserManager.getUserFromClient(event.getClient(), owner));
  160. }
  161. @Handler
  162. public void onChannelPart(final com.dmdirc.parser.events.ChannelPartEvent event) {
  163. if (!checkChannel(event.getChannel())) {
  164. return;
  165. }
  166. final ChannelClientInfo client = event.getClient();
  167. final LocalDateTime date = event.getDate();
  168. final String reason = event.getReason();
  169. if (isMyself(client)) {
  170. eventBus.publishAsync(new ChannelSelfPartEvent(date, owner,
  171. groupChatUserManager.getUserFromClient(client, owner), reason));
  172. } else {
  173. eventBus.publishAsync(new ChannelPartEvent(date, owner,
  174. groupChatUserManager.getUserFromClient(client, owner), reason));
  175. }
  176. owner.removeClient(groupChatUserManager.getUserFromClient(client, owner));
  177. }
  178. @Handler
  179. public void onChannelKick(final com.dmdirc.parser.events.ChannelKickEvent event) {
  180. if (!checkChannel(event.getChannel())) {
  181. return;
  182. }
  183. final ChannelClientInfo kickedClient = event.getKickedClient();
  184. eventBus.publishAsync(new ChannelKickEvent(
  185. event.getDate(), owner,
  186. groupChatUserManager.getUserFromClient(event.getClient(), owner),
  187. groupChatUserManager.getUserFromClient(kickedClient, owner), event.getReason()));
  188. owner.removeClient(groupChatUserManager.getUserFromClient(kickedClient, owner));
  189. }
  190. @Handler
  191. public void onChannelQuit(final com.dmdirc.parser.events.ChannelQuitEvent event) {
  192. if (!checkChannel(event.getChannel())) {
  193. return;
  194. }
  195. eventBus.publishAsync(new ChannelQuitEvent(
  196. event.getDate(), owner,
  197. groupChatUserManager.getUserFromClient(event.getClient(), owner),
  198. event.getReason()));
  199. owner.removeClient(groupChatUserManager.getUserFromClient(event.getClient(), owner));
  200. }
  201. @Handler
  202. public void onChannelAction(final com.dmdirc.parser.events.ChannelActionEvent event) {
  203. if (!checkChannel(event.getChannel())) {
  204. return;
  205. }
  206. eventBus.publishAsync(new ChannelActionEvent(
  207. event.getDate(), owner,
  208. groupChatUserManager.getUserFromClient(event.getClient(), owner),
  209. event.getMessage()));
  210. }
  211. @Handler
  212. public void onChannelNickChanged(final com.dmdirc.parser.events.ChannelNickChangeEvent event) {
  213. if (!checkChannel(event.getChannel())) {
  214. return;
  215. }
  216. final String oldNick = event.getOldNick();
  217. final ChannelClientInfo client = event.getClient();
  218. owner.renameClient(oldNick, client.getClient().getNickname());
  219. if (isMyself(client)) {
  220. eventBus.publishAsync(
  221. new ChannelSelfNickChangeEvent(
  222. event.getDate(),
  223. owner, groupChatUserManager.getUserFromClient(client, owner), oldNick));
  224. } else {
  225. eventBus.publishAsync(
  226. new ChannelNickChangeEvent(
  227. event.getDate(),
  228. owner, groupChatUserManager.getUserFromClient(client, owner), oldNick));
  229. }
  230. }
  231. @Handler
  232. public void onChannelModeChanged(final com.dmdirc.parser.events.ChannelModeChangeEvent event) {
  233. if (!checkChannel(event.getChannel())) {
  234. return;
  235. }
  236. final String host = event.getHost();
  237. final String modes = event.getModes();
  238. final ChannelClientInfo client = event.getClient();
  239. final LocalDateTime date = event.getDate();
  240. if (host.isEmpty()) {
  241. if (modes.length() <= 1) {
  242. eventBus.publishAsync(new ChannelNoModesDiscoveredEvent(date, owner));
  243. } else {
  244. eventBus.publishAsync(
  245. new ChannelModesDiscoveredEvent(date, owner,modes));
  246. }
  247. } else if (isMyself(client)) {
  248. eventBus.publishAsync(new ChannelSelfModeChangeEvent(date, owner,
  249. groupChatUserManager.getUserFromClient(client, owner), modes));
  250. } else {
  251. eventBus.publishAsync(new ChannelModeChangeEvent(date, owner,
  252. groupChatUserManager.getUserFromClient(client, owner), modes));
  253. }
  254. owner.refreshClients();
  255. }
  256. @Handler
  257. public void onChannelCTCP(final ChannelCTCPEvent event) {
  258. if (!checkChannel(event.getChannel())) {
  259. return;
  260. }
  261. final ChannelClientInfo client = event.getClient();
  262. final String message = event.getMessage();
  263. final LocalDateTime date = event.getDate();
  264. final String type = event.getType();
  265. final ChannelCtcpEvent coreEvent = new ChannelCtcpEvent(date, owner,
  266. groupChatUserManager.getUserFromClient(client, owner),type, message);
  267. eventBus.publish(coreEvent);
  268. if (!coreEvent.isHandled()) {
  269. getConnection().sendCTCPReply(client.getClient().getNickname(), type, message);
  270. }
  271. }
  272. @Handler
  273. public void onAwayStateOther(final OtherAwayStateEvent event) {
  274. owner.getUser(owner.getConnection().get().getUser(event.getClient().getNickname()))
  275. .ifPresent(c -> {
  276. if (event.getNewState() == AwayState.AWAY) {
  277. eventBus.publishAsync(
  278. new ChannelUserAwayEvent(event.getDate(), owner, c));
  279. } else {
  280. eventBus.publishAsync(
  281. new ChannelUserBackEvent(event.getDate(), owner, c));
  282. }
  283. });
  284. }
  285. @Handler
  286. public void onChannelNotice(final com.dmdirc.parser.events.ChannelNoticeEvent event) {
  287. if (!checkChannel(event.getChannel())) {
  288. return;
  289. }
  290. eventBus.publishAsync(new ChannelNoticeEvent(
  291. event.getDate(), owner,
  292. groupChatUserManager.getUserFromClient(event.getClient(), owner),
  293. event.getMessage()));
  294. }
  295. @Handler
  296. public void onChannelModeNotice(final com.dmdirc.parser.events.ChannelModeNoticeEvent event) {
  297. if (!checkChannel(event.getChannel())) {
  298. return;
  299. }
  300. eventBus.publishAsync(new ChannelModeNoticeEvent(
  301. event.getDate(), owner,
  302. groupChatUserManager.getUserFromClient(event.getClient(), owner), String.valueOf
  303. (event.getPrefix()), event.getMessage()));
  304. }
  305. @Handler
  306. public void onChannelGotListModes(final ChannelListModeEvent event) {
  307. if (!checkChannel(event.getChannel())) {
  308. return;
  309. }
  310. eventBus.publishAsync(new ChannelListModesRetrievedEvent(
  311. event.getDate(), owner, event.getMode()));
  312. }
  313. private boolean checkChannel(final ChannelInfo channelInfo) {
  314. return owner.getChannelInfo().equals(channelInfo);
  315. }
  316. }