Allow POST to PURGE for Cloudflare

Author: smashedr

src/api.js

@@ -199,7 +199,7 @@ export async function getVTStats(hash) {
         stats = data?.data?.attributes?.last_analysis_stats
     }
     if (!stats) await cacheError(key, 'VT Stats Not Found')
-    await cacheSet(key, stats, 60 * 60 * 48)
+    await cacheSet(key, stats, 60 * 60 * 24 * 5)
     return stats
 }
 
@@ -279,6 +279,16 @@ export async function incrKey(key) {
     await client.incr(key)
 }
 
+// export async function incrPurge(result) {
+//     const multi = client.multi().incr('purge_count')
+//     // multi.incr('purge_count')
+//     if (result) {
+//         multi.incr('purge_hit')
+//     } else {
+//         multi.incr('purge_miss')
+//     }
+// }
+
 export async function sendInflux() {
     if (!influxClient) return debug('InfluxDB Not Configured.')
     debug(`Processing Influx: ${new Date().toLocaleString()}`)

src/app.js

@@ -133,8 +133,8 @@ app.get('/colors{/:index}', async (req, res) => {
 // })
 
 app.all('/vt/:type/:hash', async (req, res, next) => {
-    if (req.method === 'PURGE') {
-        debug('PURGE:', req.originalUrl)
+    if (['PURGE', 'POST'].includes(req.method)) {
+        debug(`PURGE: ${req.method}`, req.originalUrl)
         if (!['id', 'sha'].includes(req.params.type)) return next()
         const hash = req.params.hash.includes(':')
             ? req.params.hash.split(':')[1]
@@ -167,8 +167,8 @@ app.get('/vt/:type/:hash', async (req, res) => {
 })
 
 app.all('/vt/:owner/:repo/:asset{/:tag}', async (req, res, next) => {
-    if (req.method === 'PURGE') {
-        debug('PURGE:', req.originalUrl)
+    if (['PURGE', 'POST'].includes(req.method)) {
+        debug(`PURGE: ${req.method}`, req.originalUrl)
         const tag = req.params.tag || 'latest'
         const key = `${req.params.owner}/${req.params.repo}/${req.params.asset}/${tag}`
         return purgeKey(res, key)
@@ -189,8 +189,8 @@ app.get('/vt/:owner/:repo/:asset{/:tag}', async (req, res) => {
 })
 
 app.all('/ghcr/tags/:owner/:package{/:latest}', async (req, res, next) => {
-    if (req.method === 'PURGE') {
-        debug('PURGE:', req.originalUrl)
+    if (['PURGE', 'POST'].includes(req.method)) {
+        debug(`PURGE: ${req.method}`, req.originalUrl)
         const key = `ghcr/tags/${req.params.owner}/${req.params.package}/tags/list`
         return purgeKey(res, key)
     }
@@ -233,8 +233,8 @@ app.get('/ghcr/tags/:owner/:package{/:latest}', async (req, res) => {
 })
 
 app.all('/ghcr/size/:owner/:package{/:tag}', async (req, res, next) => {
-    if (req.method === 'PURGE') {
-        debug('PURGE:', req.originalUrl)
+    if (['PURGE', 'POST'].includes(req.method)) {
+        debug(`PURGE: ${req.method}`, req.originalUrl)
         const tag = req.params.tag ? req.params.tag : 'latest'
         const key = `ghcr/size/${req.params.owner}/${req.params.package}/${tag}`
         return purgeKey(res, key)
@@ -270,8 +270,8 @@ app.get('/static/:message{/:label}', async (req, res) => {
 
 app.all('/:type/:url/:path', async (req, res, next) => {
     if (!['yaml', 'json'].includes(req.params.type)) return next()
-    if (req.method === 'PURGE') {
-        debug('PURGE:', req.originalUrl)
+    if (['PURGE', 'POST'].includes(req.method)) {
+        debug(`PURGE: ${req.method}`, req.originalUrl)
         return purgeKey(res, req.path)
     }
     next()
@@ -431,6 +431,11 @@ async function purgeKey(res, key) {
     const result = await cacheDelete(key)
     debug('result:', result)
     res.send(result.toString())
+    if (result) {
+        incrKey('purge_hit').catch(console.error)
+    } else {
+        incrKey('purge_miss').catch(console.error)
+    }
 }
 
 /**