12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970 |
- package com.dmdirc.ktirc.events
-
- import com.dmdirc.ktirc.IrcClient
- import com.dmdirc.ktirc.model.UserState
-
- internal class UserStateHandler : EventHandler {
-
- override fun processEvent(client: IrcClient, event: IrcEvent): List<IrcEvent> {
- when (event) {
- is ChannelJoined -> handleJoin(client.userState, event)
- is ChannelParted -> handlePart(client, event)
- is ChannelUserKicked -> handleKick(client, event)
- is ChannelNamesReceived -> handleNamesReceived(client, event)
- is UserAccountChanged -> handleAccountChanged(client, event)
- is UserQuit -> handleQuit(client.userState, event)
- }
- return emptyList()
- }
-
- private fun handleJoin(state: UserState, event: ChannelJoined) {
- state.addToChannel(event.user, event.channel)
- state.update(event.user)
- }
-
- private fun handlePart(client: IrcClient, event: ChannelParted) {
- if (client.isLocalUser(event.user)) {
- // Remove channel from all users
- client.userState.forEach { it.channels -= event.channel }
- client.userState.removeIf { it.channels.isEmpty() && !client.isLocalUser(it.details) }
- } else {
- client.userState[event.user]?.channels?.let {
- it -= event.channel
- if (it.isEmpty()) {
- client.userState -= event.user
- }
- }
- }
- }
-
- private fun handleKick(client: IrcClient, event: ChannelUserKicked) {
- if (client.isLocalUser(event.victim)) {
- // Remove channel from all users
- client.userState.forEach { it.channels -= event.channel }
- client.userState.removeIf { it.channels.isEmpty() && !client.isLocalUser(it.details) }
- } else {
- client.userState[event.victim]?.channels?.let {
- it -= event.channel
- if (it.isEmpty()) {
- client.userState -= event.victim
- }
- }
- }
- }
-
- private fun handleNamesReceived(client: IrcClient, event: ChannelNamesReceived) {
- event.toModesAndUsers(client).forEach { (_, user) ->
- client.userState.addToChannel(user, event.channel)
- client.userState.update(user)
- }
- }
-
- private fun handleAccountChanged(client: IrcClient, event: UserAccountChanged) {
- client.userState[event.user]?.details?.account = event.newAccount
- }
-
- private fun handleQuit(state: UserState, event: UserQuit) {
- state -= event.user
- }
-
- }
|