relaks-wordpress-example/server/nginx-cache.js

const Bluebird = require('bluebird');
const FS = Bluebird.promisifyAll(require('fs'));
const Crypto = require('crypto');

const NGINX_CACHE = process.env.NGINX_CACHE;

async function purge(pattern) {
  console.log(`Purging: ${pattern}`);
  const purged = [];
  if (typeof(pattern) === 'string') {
    const url = pattern;
    const md5 = Crypto.createHash('md5').update(url).digest('hex');
    const success = await removeCacheEntry({ url, md5 });
    if (success) {
      purged.push(url);
    }
  } else if (pattern instanceof RegExp) {
    const cacheEntries = await loadCacheEntries();
    for (let cacheEntry of cacheEntries) {
      if (pattern.test(cacheEntry.url)) {
        const success = await removeCacheEntry(cacheEntry);
        if (success) {
          purged.push(cacheEntry.url);
        }
      }
    }
  }
  return purged;
}

let cacheEntriesPromise = null;

async function loadCacheEntries() {
  if (!cacheEntriesPromise) {
    cacheEntriesPromise = loadCacheEntriesUncached();
  }
  const entries = await cacheEntriesPromise;
  cacheEntriesPromise = null;
  return entries;
}

async function loadCacheEntriesUncached() {
  const files = await FS.readdirAsync(NGINX_CACHE);
  const entries = [];
  for (let file of files) {
    if (/^[0-9a-f]{32}$/.test(file)) {
      const entry = await loadCacheEntry(file);
      if (entry) {
        entries.push(entry);
      }
    }
  }
  return entries;
}

const cacheEntryCache = {};

async function loadCacheEntry(md5) {
  try {
    const path = `${NGINX_CACHE}/${md5}`;
    const { mtime, size } = await FS.statAsync(path);
    const entry = cacheEntryCache[md5];
    if (!entry || entry.mtime !== mtime) {
      const url = await loadCacheEntryKey(path);
      entry = cacheEntryCache[md5] = { url, md5, mtime, size };
    }
    return entry;
  } catch (err) {
    delete cacheEntryCache[md5];
    return null;
  }
}

async function loadCacheEntryKey(path) {
  const buf = Buffer.alloc(1024);
  const fd = await FS.openAsync(path, 'r');
  try {
    await FS.readAsync(fd, buf, 0, 1024, 0);
  } finally {
    FS.closeAsync(fd);
  }
  const si = buf.indexOf('KEY:');
  const ei = buf.indexOf('\n', si);
  if (si !== -1 && ei !== -1) {
    const s = buf.toString('utf-8', si + 4, ei).trim();;
    return s;
  } else {
    throw new Error('Unable to find key');
  }
}

async function removeCacheEntry(entry) {
  try {
    delete cacheEntryCache[entry.md5];
    await FS.unlinkAsync(`${NGINX_CACHE}/${entry.md5}`);
    console.log(`Purged: ${entry.url}`);
    return true;
  } catch (err){
    return false;
  }
}

module.exports = {
  purge,
  read: loadCacheEntries,
};