feat(plugins/stargazers): add worldmap (#1137) [skip ci]

source/plugins/stargazers/worldmap/index.mjs

@@ -0,0 +1,60 @@
+//Imports
+import * as d3 from "d3"
+import D3Node from "d3-node"
+import color from "color"
+import {Client as Gmap} from "@googlemaps/google-maps-services-js"
+
+/**
+ * Worldmap
+ * Mostly ported from https://github.com/dyatko/worldstar
+ * License: https://raw.githubusercontent.com/dyatko/worldstar/master/LICENSE
+ */
+export default async function (login, {locations, sample, imports, token}) {
+  //Parse geocodes
+  let stars = new Map()
+  if (token) {
+    const cache = new Map()
+    const get = new Gmap()
+    locations = imports.shuffle(locations.filter(string => string).map(string => string.toLocaleLowerCase())).slice(0, sample || Infinity)
+    for (const location of locations) {
+      console.debug(`metrics/compute/${login}/plugins > stargazers > worldmap > looking for ${location}`)
+      if (!cache.has(location)) {
+        try {
+          const {data:{results}} = await get.geocode({params:{address:location, key:token}})
+          const country = results.at(0).address_components.find(({types}) => types.includes("country"))
+          cache.set(location, country.short_name ?? country.long_name)
+          console.debug(`metrics/compute/${login}/plugins > stargazers > worldmap > ${location} resolved to ${cache.get(location)}`)
+        }
+        catch (error) {
+          console.debug(`metrics/compute/${login}/plugins > stargazers > worldmap > failed to resolve ${location}: ${error}`)
+        }
+      }
+      const code = cache.get(location)
+      stars.set(code, (stars.get(code) ?? 0) + 1)
+    }
+  }
+  else throw {error:{message:"Google Maps API token is not set"}}
+
+  //Generate SVG
+  const d3n = new D3Node()
+  const svg = d3n.createSVG(480, 315)
+  const countries = JSON.parse(await imports.fs.readFile(imports.paths.join(imports.__module(import.meta.url), "atlas/50m_countries.geojson")))
+  const geopath = d3.geoPath(d3.geoMercator().fitWidth(svg.attr("width"), countries))
+  const splits = [...new Set(stars.values())].sort((a, b) => a - b)
+  svg
+    .append("g")
+    .selectAll("path")
+    .data(countries.features)
+    .join("path")
+    .attr("id", ({id}) => id)
+    .style("fill", ({properties:{iso_a2, wb_a2, sov_a3}}) => {
+      const code = iso_a2?.match(/[A-Z]{2}/) ? iso_a2 : wb_a2?.match(/[A-Z]{2}/) ? wb_a2 : sov_a3?.substr(0, 2) ?? ""
+      const value = stars.get(code)
+      return color("#216e39").mix(color("#ffffff"), 1 - Math.max(0, splits.indexOf(value))/splits.length).hex()
+    })
+    .style("stroke", "#afafaf")
+    .style("stroke-width", "0.6px")
+    .attr("d", geopath)
+
+  return d3n.svgString()
+}