|
45 | 45 | import org.kitteh.irc.client.library.element.mode.ModeStatus; |
46 | 46 | import org.kitteh.irc.client.library.element.mode.ModeStatusList; |
47 | 47 | import org.kitteh.irc.client.library.element.mode.UserMode; |
| 48 | +import org.kitteh.irc.client.library.event.abstractbase.ClientReceiveServerMessageEventBase; |
48 | 49 | import org.kitteh.irc.client.library.event.batch.ClientBatchEndEvent; |
49 | 50 | import org.kitteh.irc.client.library.event.batch.ClientBatchMessageEvent; |
50 | 51 | import org.kitteh.irc.client.library.event.batch.ClientBatchStartEvent; |
@@ -950,7 +951,7 @@ private void handleLine(final @NonNull String line) { |
950 | 951 | BatchReferenceTag tag = this.batchHold.get(batch); |
951 | 952 | if (tag != null) { |
952 | 953 | tag.addEvent(event); |
953 | | - this.eventManager.callEvent(new ClientBatchMessageEvent(this, event.getOriginalMessage(), tag)); |
| 954 | + this.eventManager.callEvent(new ClientBatchMessageEvent(this, event.getSource(), tag)); |
954 | 955 | return; |
955 | 956 | } |
956 | 957 | // else improper batch |
@@ -980,15 +981,15 @@ private void sendLineEvent(@NonNull ClientReceiveServerMessageEvent event) { |
980 | 981 | String type = parameters.get(1); |
981 | 982 | List<String> batchParameters = new ArrayList<>(parameters.subList(2, parameters.size())); |
982 | 983 | BatchReferenceTag tag = new BatchReferenceTag(refTag, type, batchParameters); |
983 | | - ClientBatchStartEvent batchEvent = new ClientBatchStartEvent(this, event.getOriginalMessage(), tag); |
| 984 | + ClientBatchStartEvent batchEvent = new ClientBatchStartEvent(this, event.getSource(), tag); |
984 | 985 | this.eventManager.callEvent(batchEvent); |
985 | 986 | if (!batchEvent.isReferenceTagIgnored()) { |
986 | 987 | this.batchHold.put(refTag, tag); |
987 | 988 | } |
988 | 989 | } else if (plusOrMinus == '-') { |
989 | 990 | BatchReferenceTag tag = this.batchHold.remove(refTag); |
990 | 991 | if (tag != null) { |
991 | | - this.eventManager.callEvent(new ClientBatchEndEvent(this, event.getOriginalMessage(), tag)); |
| 992 | + this.eventManager.callEvent(new ClientBatchEndEvent(this, event.getSource(), tag)); |
992 | 993 | tag.getEvents().forEach(this::sendLineEvent); |
993 | 994 | } |
994 | 995 | } else { |
|
0 commit comments