Add cache to function response (#22)

* Add cache to function response

* Lowered cache-control maxage to 5 min

* Changed cache control directives to public, max-age

* Added cache-control to pin.js

api/index.js

@@ -1,10 +1,7 @@
 const axios = require("axios");
-const { renderError, kFormatter } = require("../utils");
 require("dotenv").config();
 
 async function fetchStats(username) {
-  if (!username) throw Error("Invalid username");
-
   const res = await axios({
     url: "https://api.github.com/graphql",
     method: "post",
@@ -20,7 +17,7 @@ async function fetchStats(username) {
               totalCount
             }
             contributionsCollection {
-              totalCommitContributions
+              contributionYears
             }
             pullRequests(first: 100) {
               totalCount
@@ -52,17 +49,65 @@ async function fetchStats(username) {
     totalStars: 0,
     contributedTo: 0,
   };
-
-  if (res.data.errors) {
-    console.log(res.data.errors);
-    throw Error("Could not fetch user");
-  }
+  if (res.data.error) return stats;
 
   const user = res.data.data.user;
 
+  // FETCH ALL COMMITS
+  let commitPromises = user.contributionsCollection.contributionYears.map(
+    async (year) => {
+      // Don't fetch contributions older than 2008 (Perf optimization)
+      if (year < 2008) {
+        return {
+          user: { contributionsCollection: { totalCommitContributions: 0 } },
+        };
+      }
+
+      const currentDate = new Date();
+      currentDate.setFullYear(year, 0, 0);
+
+      const res = await axios({
+        url: "https://api.github.com/graphql",
+        method: "post",
+        headers: {
+          Authorization: `bearer ${process.env.GITHUB_TOKEN}`,
+        },
+        data: {
+          query: `
+          query userInfo($login: String!, $from: DateTime!) {
+            user(login: $login) {
+              contributionsCollection(from: $from) {
+                totalCommitContributions
+              }
+            }
+          }
+        `,
+          variables: {
+            login: username,
+            from: currentDate.toISOString(),
+          },
+        },
+      });
+
+      console.log(
+        currentDate.toDateString(),
+        res.data.data.user.contributionsCollection.totalCommitContributions
+      );
+      return res.data.data;
+    }
+  );
+
+  const allCommits = await Promise.all(commitPromises);
+
+  const totalCommits = allCommits.reduce((preYear, currYear) => {
+    return (
+      preYear + currYear.user.contributionsCollection.totalCommitContributions
+    );
+  }, 0);
+
   stats.name = user.name;
   stats.totalIssues = user.issues.totalCount;
-  stats.totalCommits = user.contributionsCollection.totalCommitContributions;
+  stats.totalCommits = totalCommits;
   stats.totalPRs = user.pullRequests.totalCount;
   stats.contributedTo = user.repositoriesContributedTo.totalCount;
 
@@ -74,11 +119,12 @@ async function fetchStats(username) {
 }
 
 const createTextNode = (icon, label, value, lheight) => {
-  const classname = icon === "★" && "star-icon";
   return `
     <tspan x="25" dy="${lheight}" class="stat bold">
-    <tspan class="icon ${classname}" fill="#4C71F2">${icon}</tspan> ${label}:</tspan>
-    <tspan x="155" dy="0" class="stat">${kFormatter(value)}</tspan>
+    <tspan class="icon ${
+      icon === "★" && "star-icon"
+    }" fill="#4C71F2">${icon}</tspan> ${label}:</tspan>
+    <tspan x="155" dy="0" class="stat">${value}</tspan>
   `;
 };
 
@@ -140,15 +186,11 @@ module.exports = async (req, res) => {
   const hide_border = req.query.hide_border;
   const show_icons = req.query.show_icons;
   const line_height = req.query.line_height;
-  let stats;
 
+  const stats = await fetchStats(username);
+
+  res.setHeader('Cache-Control', 'public, max-age=300');
   res.setHeader("Content-Type", "image/svg+xml");
-  try {
-    stats = await fetchStats(username);
-  } catch (err) {
-    return res.send(renderError(err.message));
-  }
-
   res.send(
     renderSVG(stats, {
       hide: JSON.parse(hide || "[]"),

api/pin.js

@@ -1,7 +1,12 @@
 const axios = require("axios");
-const { renderError, kFormatter } = require("../utils");
 require("dotenv").config();
 
+function kFormatter(num) {
+  return Math.abs(num) > 999
+    ? Math.sign(num) * (Math.abs(num) / 1000).toFixed(1) + "k"
+    : Math.sign(num) * Math.abs(num);
+}
+
 async function fetchRepo(username, reponame) {
   const res = await axios({
     url: "https://api.github.com/graphql",
@@ -51,26 +56,14 @@ async function fetchRepo(username, reponame) {
     },
   });
 
-  const data = res.data.data;
+  if (res.data.error && res.data.error.type !== "NOT_FOUND") return {};
+  if (!res.data.data.organization && res.data.data.user)
+    return res.data.data.user.repository;
 
-  console.log(res.data);
-  if (!data.user && !data.organization) {
-    throw new Error("Not found");
-  }
+  const isOrg = res.data.data.organization && !res.data.data.user;
+  if (isOrg) return res.data.data.organization.repository;
 
-  if (data.organization === null && data.user) {
-    if (!data.user.repository) {
-      throw new Error("User Repository Not found");
-    }
-    return data.user.repository;
-  }
-
-  if (data.user === null && data.organization) {
-    if (!data.organization.repository) {
-      throw new Error("Organization Repository Not found");
-    }
-    return data.organization.repository;
-  }
+  return res.data.data.user.repository;
 }
 
 const renderRepoCard = (repo) => {
@@ -79,7 +72,7 @@ const renderRepoCard = (repo) => {
 
   const shiftText = primaryLanguage.name.length > 15 ? 0 : 30;
   return `
-    <svg width="400" height="${height}" viewBox="0 0 400 ${height}" fill="none" xmlns="http://www.w3.org/2000/svg">
+  <svg width="400" height="${height}" viewBox="0 0 400 ${height}" fill="none" xmlns="http://www.w3.org/2000/svg">
       <style>
       .header { font: 600 18px 'Segoe UI', Ubuntu, Sans-Serif; fill: #2F80ED }
       .stat { font: 600 14px 'Segoe UI', Ubuntu, Sans-Serif; fill: #333 }
@@ -125,15 +118,9 @@ module.exports = async (req, res) => {
   const username = req.query.username;
   const repo = req.query.repo;
 
-  let repoData;
+  const repoData = await fetchRepo(username, repo);
+
+  res.setHeader('Cache-Control', 'public, max-age=300');
   res.setHeader("Content-Type", "image/svg+xml");
-
-  try {
-    repoData = await fetchRepo(username, repo);
-  } catch (err) {
-    console.log(err);
-    return res.send(renderError(err.message));
-  }
-
   res.send(renderRepoCard(repoData));
 };

readme.md

@@ -1,6 +1,6 @@
 ## Github Readme stats
 
-Get dynamically generated GitHub stats on your readmes!
+Get dynamically generated github stats on your readmes!
 
 # Features
 
@@ -9,7 +9,7 @@ Get dynamically generated GitHub stats on your readmes!
 
 # Github Stats Card
 
-Copy paste this into your markdown content, and that's it. Simple!
+Copy paste this into your markdown content and thats it. simple!
 
 change the `?username=` value to your GitHubs's username
 
@@ -17,15 +17,9 @@ change the `?username=` value to your GitHubs's username
 ![Anurag's github stats](https://github-readme-stats.vercel.app/api?username=anuraghazra)
 ```
 
-_Psst, you can also use this code so that `github-readme-stats` gets proper credit :D and other people can also try it out!_
+### Hiding certain stats
 
-```md
-[![Anurag's github stats](https://github-readme-stats.vercel.app/api?username=anuraghazra)](https://github.com/anuraghazra/github-readme-stats)
-```
-
-### Hiding individual stats
-
-To hide any specific stats, you can pass a query parameter `?hide=` with an array of items, you wanna hide.
+To hide any specific stats you can pass a query parameter `?hide=` with an array of items you wanna hide.
 
 > Options: `&hide=["stars","prs","issues","contribs"]`
 
@@ -35,7 +29,7 @@ To hide any specific stats, you can pass a query parameter `?hide=` with an arra
 
 ### Showing icons
 
-To enable icons, you can pass `show_icons=true` in the query param like so
+To enable icons you can pass `show_icons=true` in the query param like so
 
 ```md
 ![Anurag's github stats](https://github-readme-stats.vercel.app/api?username=anuraghazra&show_icons=true)
@@ -62,13 +56,13 @@ Other options:
 
 # Github Extra Pins
 
-Github extra pins allow you to pin more than 6 repositories in your profile using a GitHub readme profile.
+Github extra pins allows you to pin more than 6 repositories in your profile by using github readme profile.
 
-Yey! you are no longer limited to 6 pinned repositories.
+Yey! you are no longer limited to 6 pinned repos
 
 ### Usage
 
-Copy-paste this code into your readme and change the links.
+Copy paste this code into your readme and change the links.
 
 Endpoint: `api/pin?username=anuraghazra&repo=github-readme-stats`
 
@@ -82,7 +76,7 @@ Endpoint: `api/pin?username=anuraghazra&repo=github-readme-stats`
 
 ### Quick Tip (Align The Repo Cards)
 
-You usually won't be able to layout the images side by side to do that you can use this approach
+Normally you won't be able to layout the images side by side to do that you can use this approach
 
 ```md
 <a href="https://github.com/anuraghazra/github-readme-stats">

utils.js

@@ -1,21 +0,0 @@
-const renderError = (message) => {
-  return `
-    <svg width="495" height="100" viewBox="0 0 495 100" fill="none" xmlns="http://www.w3.org/2000/svg">
-    <style>
-    .text { font: 600 16px 'Segoe UI', Ubuntu, Sans-Serif; fill: #2F80ED }
-    .small { font: 600 12px 'Segoe UI', Ubuntu, Sans-Serif; fill: #252525 }
-    </style>
-    <rect x="0.5" y="0.5" width="494" height="99%" rx="4.5" fill="#FFFEFE" stroke="#E4E2E2"/>
-    <text x="25" y="45" class="text">Something went wrong! file an issue at https://git.io/JJmN9</text>
-    <text x="25" y="65" class="text small">${message}</text>
-    </svg>
-  `;
-};
-
-function kFormatter(num) {
-  return Math.abs(num) > 999
-    ? Math.sign(num) * (Math.abs(num) / 1000).toFixed(1) + "k"
-    : Math.sign(num) * Math.abs(num);
-}
-
-module.exports = { renderError, kFormatter };