jarv.is/functions/stats.js

"use strict";

const faunadb = require("faunadb"),
  q = faunadb.query;
const numeral = require("numeral");
const pluralize = require("pluralize");
const rssParser = require("rss-parser");
require("dotenv").config();

const baseUrl = "https://jarv.is/";

exports.handler = async (event) => {
  try {
    // some rudimentary error handling
    if (!process.env.FAUNADB_SERVER_SECRET) {
      throw new Error("Database credentials aren't set.");
    }
    if (event.httpMethod !== "GET") {
      throw new Error(`Method ${event.httpMethod} not allowed.`);
    }

    const parser = new rssParser({
      timeout: 3000,
    });
    const client = new faunadb.Client({
      secret: process.env.FAUNADB_SERVER_SECRET,
    });

    // get database and RSS results asynchronously
    const [feed, result] = await Promise.all([
      parser.parseURL(baseUrl + "feed.xml"),
      client.query(
        q.Map(
          q.Paginate(q.Documents(q.Collection("hits"))),
          q.Lambda((x) => q.Select("data", q.Get(x)))
        )
      ),
    ]);

    let stats = {
      total: {
        hits: 0,
      },
      pages: result.data,
    };

    stats.pages.map((p) => {
      // match URLs from RSS feed with db to populate some metadata
      let match = feed.items.find((x) => x.link === baseUrl + p.slug + "/");
      if (match) {
        p.title = match.title;
        p.url = match.link;
        p.date = match.isoDate;
        delete p.slug;
      }

      // it's easier to add comma-separated numbers and proper pluralization here on the backend
      p.pretty_hits = numeral(p.hits).format("0,0");
      p.pretty_unit = pluralize("hit", p.hits);

      // add these hits to running tally
      stats.total.hits += p.hits;

      return p;
    });

    // sort by hits (descending)
    stats.pages.sort((a, b) => {
      return a.hits > b.hits ? -1 : 1;
    });

    // do same prettification as above to totals
    stats.total.pretty_hits = numeral(stats.total.hits).format("0,0");
    stats.total.pretty_unit = pluralize("hit", stats.total.hits);

    return {
      statusCode: 200,
      headers: {
        "Content-Type": "application/json; charset=utf-8",
        "Access-Control-Allow-Methods": "GET",
        "Access-Control-Allow-Origin": "*",
      },
      body: JSON.stringify(stats),
    };
  } catch (error) {
    console.error(error);

    return {
      statusCode: 400,
      body: JSON.stringify({
        message: error.message,
      }),
    };
  }
};