diff --git a/site.js b/site.js index d124487..4d11b19 100644 --- a/site.js +++ b/site.js @@ -75,17 +75,22 @@ async function siteCollector(user, path, site, useLimit) { let penv = process.env.user || 'paradock'; penv = penv.split(','); - let users = []; + let users = penv; let legal = []; - for (let u of penv) { - let ca = await siteCollector(u, 'followers', site); - users = users.concat(ca); - legal = legal.concat(ca); - users = users.concat(await siteCollector(u, 'following', site)); - } + let depth = process.env.depth || 1; - users = [...new Set(users)]; + for (let i = 0; i < depth; i++) { + for (let u of penv) { + let ca = await siteCollector(u, 'followers', site); + users = users.concat(ca); + legal = legal.concat(ca); + users = users.concat(await siteCollector(u, 'following', site)); + } + + users = [...new Set(users)]; + penv = [...users]; + } let data = {}; let p = [];