From dfc44b63b8906f7de653ab3c7be92513efcf0179 Mon Sep 17 00:00:00 2001
From: devStorm <59678453+developStorm@users.noreply.github.com>
Date: Thu, 10 Jun 2021 01:01:57 -0700
Subject: [PATCH] Add default affiliation

---
 src/fetchers/stats-fetcher.js         | 7 ++++++-
 src/fetchers/top-languages-fetcher.js | 5 +++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/fetchers/stats-fetcher.js b/src/fetchers/stats-fetcher.js
index 73dc9690af60b..3ee0964ee270b 100644
--- a/src/fetchers/stats-fetcher.js
+++ b/src/fetchers/stats-fetcher.js
@@ -105,7 +105,12 @@ async function fetchStats(
     rank: { level: "C", score: 0 },
   };
 
-  let res = await retryer(fetcher, { login: username, ownerAffiliations});
+  // Set default value for ownerAffiliations in GraphQL query won't work because
+  // parseArray() will always return an empty array even nothing was specified
+  // and GraphQL would consider that empty arr as a valid value. Nothing will be
+  // queried in that case as no affiliation is presented.
+  ownerAffiliations = ownerAffiliations.length > 0 ? ownerAffiliations : ["OWNER"];
+  let res = await retryer(fetcher, { login: username, ownerAffiliations });
 
   if (res.data.errors) {
     logger.error(res.data.errors);
diff --git a/src/fetchers/top-languages-fetcher.js b/src/fetchers/top-languages-fetcher.js
index d3e150f142ce7..99a25f9eb5d27 100644
--- a/src/fetchers/top-languages-fetcher.js
+++ b/src/fetchers/top-languages-fetcher.js
@@ -37,6 +37,11 @@ const fetcher = (variables, token) => {
 async function fetchTopLanguages(username, exclude_repo = [], ownerAffiliations) {
   if (!username) throw Error("Invalid username");
 
+  // Set default value for ownerAffiliations in GraphQL query won't work because
+  // parseArray() will always return an empty array even nothing was specified
+  // and GraphQL would consider that empty arr as a valid value. Nothing will be
+  // queried in that case as no affiliation is presented.
+  ownerAffiliations = ownerAffiliations.length > 0 ? ownerAffiliations : ["OWNER"];
   const res = await retryer(fetcher, { login: username, ownerAffiliations });
 
   if (res.data.errors) {