var nodes; var edges; var options = { nodes: { borderWidth:4, size:30, color: { border: '#222222', background: '#666666' }, font:{color:'#eeeeee'} }, edges: { color: 'lightgray' } }; /** * Checks if a user is a node in the graph * * @param userID * @returns {boolean} */ function alreadyInGraph(userID) { for(var i = 0; i < nodes.length; i++) { if(nodes[i].id === userID) { return true; } } return false; } /** * adds a person to the nodes list * * @param profileData */ function addPersonToGraph(profileData) { nodes.push( { id:profileData.id, name:profileData.login, shape: 'circularImage', image:profileData.avatar_url }); } /** * Adds the followers/following of a person * to the graph * * @param username * @param apiPath * @returns {Promise} */ function addFriends(username, apiPath) { return new Promise(function(resolve, reject) { queryAPIByUser(apiPath, username, function(data) { for(var i = 0; i < data.length; i++) { if(!alreadyInGraph(data[i].id)) { addPersonToGraph(data[i]); } } resolve(); }, function(error) { reject(error); }) }); } /** * Greedy function which checks to see if a edge is in the graphs * * @param id1 * @param id2 * @returns {boolean} */ function edgeInGraph(id1, id2) { for(var i = 0;i < edges.length; i++) { if(edges[i].from === id1 && edges[i].to === id2) { return true; } if(edges[i].to === id1 && edges[i].from === id2) { return true; } } return false; } /** * Adds a connection to the graph * * @param person1 * @param person2 */ function addConnection(person1, person2) { if(person1.id !== person2.id) { if(alreadyInGraph(person2.id) && !edgeInGraph(person1.id, person2.id)) { edges.push( { from: person1.id, to: person2.id }); } } } /** * Processes all the connections of a user and adds them to the graph * * @param user has .id and .name * @returns {Promise} */ function processUserConnections(user) { return new Promise(function(resolve, reject) { queryAPIByUser(API_FOLLOWING, user.name, function(data) { for(var i = 0; i < data.length; i++) { addConnection(user, data[i]) } queryAPIByUser(API_FOLLOWERS, user.name, function(data2) { for(var i = 0; i < data2.length; i++) { addConnection(user, data2[i]); } resolve(); }, function(error) { reject(error); }); }, function(error) { reject(error); }) }); } /** * Creates connections between all the nodes in * the graph. * * @returns {Promise} */ function createConnections() { return new Promise(function(resolve, reject) { var prom = []; for(var i = 0; i < nodes.length; i++) { prom.push(processUserConnections(nodes[i])); } Promise.all(prom).then(function() { resolve(); }).catch(function(error) { reject(error); }); }); } /** * Adds the base person to the graph. * * @param username * @returns {Promise} */ function addSelfToGraph(username) { return new Promise(function(resolve, reject) { queryAPIByUser("", username, function(data) { addPersonToGraph(data); resolve(); }, function(error) { reject(error); }); }); } /** * Creates a graph * @param username * @param containerName * @param graphsTitle */ function createFriendsGraph(username, containerName, graphsTitle) { nodes = []; edges = []; addSelfToGraph(username).then(function() { addFriends(username, API_FOLLOWERS).then(function() { addFriends(username, API_FOLLOWING).then(function() { createConnections().then(function() { var container = document.getElementById(containerName); var data = { nodes: nodes, edges: edges }; var network = new vis.Network(container, data, options); }); }); }) }).catch(function(error) { console.log(error); $("#" + graphsTitle).html("Error Fetching Data From API"); }); }