You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
259 lines
5.1 KiB
259 lines
5.1 KiB
|
|
var nodes;
|
|
|
|
var edges;
|
|
|
|
|
|
const 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<any>}
|
|
*/
|
|
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<any>}
|
|
*/
|
|
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<any>}
|
|
*/
|
|
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<any>}
|
|
*/
|
|
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");
|
|
});
|
|
}
|