unstorage/drivers/upstash.cjs

Universal Storage Layer

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _redis = require("@upstash/redis");
var _utils = require("./utils/index.cjs");
const DRIVER_NAME = "upstash";
module.exports = (0, _utils.defineDriver)((options = {}) => {
  const base = (0, _utils.normalizeKey)(options?.base);
  const r = (...keys) => (0, _utils.joinKeys)(base, ...keys);
  let redisClient;
  const getClient = () => {
    if (redisClient) {
      return redisClient;
    }
    const url = options.url || globalThis.process?.env?.UPSTASH_REDIS_REST_URL;
    const token = options.token || globalThis.process?.env?.UPSTASH_REDIS_REST_TOKEN;
    redisClient = new _redis.Redis({
      url,
      token,
      ...options
    });
    return redisClient;
  };
  return {
    name: DRIVER_NAME,
    getInstance: getClient,
    hasItem(key) {
      return getClient().exists(r(key)).then(Boolean);
    },
    getItem(key) {
      return getClient().get(r(key));
    },
    setItem(key, value, tOptions) {
      const ttl = tOptions?.ttl || options.ttl;
      return getClient().set(r(key), value, ttl ? {
        ex: ttl
      } : void 0).then(() => {});
    },
    removeItem(key) {
      return getClient().del(r(key)).then(() => {});
    },
    getKeys(_base) {
      return getClient().keys(r(_base, "*")).then(keys => base ? keys.map(key => key.slice(base.length + 1)) : keys);
    },
    async clear(base2) {
      const keys = await getClient().keys(r(base2, "*"));
      if (keys.length === 0) {
        return;
      }
      return getClient().del(...keys).then(() => {});
    }
  };
});