edu.rit.chrisbitler.ritcraft.slackintegration.rtm.UserList.java Source code

Introduction

Here is the source code for edu.rit.chrisbitler.ritcraft.slackintegration.rtm.UserList.java

Source

/*
 * Copyright 2015 RITcraft & Chris Bitler
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/**
 * @author Chris Bitler
 */
package edu.rit.chrisbitler.ritcraft.slackintegration.rtm;

import edu.rit.chrisbitler.ritcraft.slackintegration.SlackIntegration;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.JSONValue;

import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

/**
 * Class to help maintain a list of user ids to usernames since
 * slack returns user ids in chat messages instead of real names
 */
public class UserList {
    private static HashMap<String, String> users = new HashMap<>(); //User id -> username map

    /**
     * Get a user's name based on their slack user id
     * @param id The user's slack id
     * @return Their 'real name' as per slack
     */
    public static String getName(String id) {
        if (users.get(id) != null) {
            return users.get(id);
        } else {
            queryUsers();
            return users.get(id);
        }
    }

    /**
     * Query the slack api to get the list of users and their real names. This is done whenever we can't find an ID mapping,
     * and when the plugin loads
     */
    public static void queryUsers() {
        try {
            users.clear();
            //Contact the slack api to get the user list
            HttpURLConnection conn = (HttpURLConnection) new URL(
                    "https://www.slack.com/api/users.list?token=" + SlackIntegration.BOT_TOKEN).openConnection();
            conn.connect();
            JSONObject retVal = (JSONObject) JSONValue
                    .parse(new InputStreamReader((InputStream) conn.getContent()));
            JSONArray members = (JSONArray) retVal.get("members");

            //Loop through the members and add them to the map
            Iterator<JSONObject> iter = members.iterator();
            while (iter.hasNext()) {
                JSONObject obj = iter.next();
                JSONObject profile = (JSONObject) obj.get("profile");
                String id = (String) obj.get("id");
                String realName = (String) profile.get("real_name");
                users.put(id, realName);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static String replaceAllUserIDs(String text) {
        for (Map.Entry<String, String> id : users.entrySet()) {
            text = text.replace(id.getKey(), id.getValue());
        }
        return text;
    }
}