io.dangernoodle.slack.client.SlackConnectionSession Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of slack-bot-api-core Show documentation
The newest version!
package io.dangernoodle.slack.client;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicLong;

import io.dangernoodle.slack.objects.SlackIntegration;
import io.dangernoodle.slack.objects.SlackMessageable;
import io.dangernoodle.slack.objects.SlackSelf;
import io.dangernoodle.slack.objects.SlackTeam;
import io.dangernoodle.slack.objects.SlackUser;
import io.dangernoodle.slack.objects.api.SlackStartRtmResponse;


public class SlackConnectionSession
{
    // visible for testing
    final Map channels;

    // visible for testing
    final Map integrations;

    // visible for testing
    final Map users;

    private final AtomicLong lastSentPingId;

    private SlackSelf self;

    private SlackTeam team;

    private SlackUser user;

    public SlackConnectionSession()
    {
        this.lastSentPingId = new AtomicLong();

        this.integrations = new ConcurrentHashMap<>(16, 0.9f, 1);
        this.users = new ConcurrentHashMap<>(16, 0.9f, 1);
        this.channels = new ConcurrentHashMap<>(16, 0.9f, 1);
    }

    public SlackMessageable findChannel(SlackMessageable.Id id) throws IllegalArgumentException
    {
        return findById(id, channels);
    }

    public SlackIntegration findIntegration(SlackIntegration.Id id)
    {
        return findById(id, integrations);
    }

    public SlackUser findUser(SlackUser.Id id) throws IllegalArgumentException
    {
        return findById(id, users);
    }

    public SlackUser getSelfUser()
    {
        return user;
    }

    public SlackTeam getTeam()
    {
        return team;
    }

    long getLastPingId()
    {
        return lastSentPingId.get();
    }

    void removeChannel(SlackMessageable.Id channelId)
    {
        channels.remove(channelId);
    }

    void updateChannels(SlackMessageable channel)
    {
        channels.put(channel.getId(), channel);
    }

    void updateIntegrations(SlackIntegration integration)
    {
        integrations.put(integration.getId(), integration);
    }

    void updateLastPingId(long id)
    {
        lastSentPingId.set(id);
    }

    synchronized void updateSession(SlackStartRtmResponse session)
    {
        self = session.getSelf();
        team = session.getTeam();

        channels.clear();

        // for now these are all all lumped together as a 'SlackMessageable'
        session.getGroups().forEach(this::updateChannels);
        session.getChannels().forEach(this::updateChannels);
        session.getDirectMessages().forEach(this::updateChannels);

        session.getUsers().forEach(this::updateUsers);
        session.getIntegrations().forEach(this::updateIntegrations);

        user = findUser(self.getId());
    }

    void updateUsers(SlackUser user)
    {
        users.put(user.getId(), user);
    }

    private  T findById(I id, Map map)
    {
        return map.get(id);
    }
}