Add PyPI integration

Author: rossmacarthur

README.md

@@ -26,6 +26,7 @@
 - GitHub
 - crates.io
 - npm
+- PyPI
 
 <small>More integrations coming soon!</small>
 
@@ -43,6 +44,12 @@ Building SpaceBadgers has been a labor of love, aiming to offer a superior, reli
 
 ## Development
 
+Clone the repository and make sure to initialize the submodules
+
+```bash
+git submodule update --init --recursive
+```
+
 ### Environment Variables
 > Paste this template into `badgers-web/.env.local`
 

badgers-web/src/app/page.tsx

@@ -180,6 +180,14 @@ export default function Home() {
                     <Row name="Types" path="/npm/types/:pkg" inject={['react']} />
                     <Row name="Types (scoped)" path="/npm/types/:org/:pkg" inject={['@octokit', 'rest']} />
                 </div>
+              </Section>
+              <Section name="PyPI">
+                <div className="grid grid-cols-[auto_1fr_auto] items-center gap-x-8">
+                    <Row name="Name" path="/pypi/name/:pkg" inject={['janus']} />
+                    <Row name="Version" path="/pypi/version/:pkg" inject={['janus']} />
+                    <Row name="Name and Version" path="/pypi/info/:pkg" inject={['janus']} />
+                    <Row name="License" path="/pypi/license/:pkg" inject={['janus']} />
+                </div>
             </Section>
         </div>
     </main>

badgers-web/src/app/pypi/info/[pkg]/route.ts

@@ -0,0 +1,18 @@
+import { NextRequest } from "next/server"
+
+import Badge from '@/utils/Badge'
+import PyPI from '@/utils/PyPI'
+
+interface Params {
+    params: {
+        pkg: string
+    }
+}
+
+export async function GET(request: NextRequest, { params: { pkg } }: Params) {
+    const data = await PyPI.getPackage(pkg, 'latest')
+    if (data === null) return await Badge.error(request, 'pypi')
+    return await Badge.generate(request, 'pypi', `${data.name} v${data.version}`)
+}
+
+export const runtime = 'edge'

badgers-web/src/app/pypi/license/[pkg]/route.ts

@@ -0,0 +1,20 @@
+import { NextRequest } from "next/server"
+
+import Badge from '@/utils/Badge'
+import PyPI from '@/utils/PyPI'
+
+interface Params {
+    params: {
+        pkg: string
+    }
+}
+
+export async function GET(request: NextRequest, { params: { pkg } }: Params) {
+    const data = await PyPI.getPackage(pkg, 'latest')
+    if (data === null) return await Badge.error(request, 'pypi')
+    const license = data.license || 'unknown'
+    const color = data.license ? 'blue' : 'gray'
+    return await Badge.generate(request, 'license', `${license}`, { color })
+}
+
+export const runtime = 'edge'

badgers-web/src/app/pypi/name/[pkg]/route.ts

@@ -0,0 +1,18 @@
+import { NextRequest } from "next/server"
+
+import Badge from '@/utils/Badge'
+import PyPI from '@/utils/PyPI'
+
+interface Params {
+    params: {
+        pkg: string
+    }
+}
+
+export async function GET(request: NextRequest, { params: { pkg } }: Params) {
+    const data = await PyPI.getPackage(pkg, 'latest')
+    if (data === null) return await Badge.error(request, 'pypi')
+    return await Badge.generate(request, 'pypi', `${data.name}`)
+}
+
+export const runtime = 'edge'

badgers-web/src/app/pypi/version/[pkg]/route.ts

@@ -0,0 +1,18 @@
+import { NextRequest } from "next/server"
+
+import Badge from '@/utils/Badge'
+import PyPI from '@/utils/PyPI'
+
+interface Params {
+    params: {
+        pkg: string
+    }
+}
+
+export async function GET(request: NextRequest, { params: { pkg } }: Params) {
+    const data = await PyPI.getPackage(pkg, 'latest')
+    if (data === null) return await Badge.error(request, 'pypi')
+    return await Badge.generate(request, 'pypi', `v${data.version}`)
+}
+
+export const runtime = 'edge'

badgers-web/src/utils/PyPI.ts

@@ -0,0 +1,52 @@
+// Docs: https://warehouse.pypa.io/api-reference/json.html
+
+const BASE_URL = 'https://pypi.org/pypi'
+
+type VersionIdentifier = string | "latest"
+
+type Package = {
+    name: string
+    version: string
+    license: string
+    // ...
+}
+
+const fetchOptions = {
+    next: {
+        revalidate: 300, // 5m
+    }
+}
+
+export default class PyPI {
+    /**
+     * Get the package information for the given version.
+     *
+     * @param packageName The package name
+     * @param version The version or `'latest'`
+     * @returns The package information for the given version, or `null`
+     *
+     * @example
+     * ```ts
+     * await PyPI.getPackageVersion('requests', 'latest')
+     * await PyPI.getPackageVersion('numpy', '1.24.3')
+     * ```
+     */
+    static async getPackage(packageName: string, version: VersionIdentifier): Promise<Package | null> {
+        const url = (version === 'latest') ?
+            `${BASE_URL}/${packageName}/json` :
+            `${BASE_URL}/${packageName}/${version}/json`
+
+        try {
+            const response = await fetch(url, fetchOptions)
+
+            if (response.status === 404) {
+                return null
+            }
+
+            const resp = await response.json() as { info: Package }
+            return resp.info
+        } catch {
+            return null
+        }
+    }
+}