agentic-studio/network-poc/frontend/node_modules/unstorage/drivers/planetscale.cjs

"use strict";

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

var _utils = require("./utils/index.cjs");
var _database = require("@planetscale/database");
var _nodeFetchNative = require("node-fetch-native");
const DRIVER_NAME = "planetscale";
module.exports = (0, _utils.defineDriver)((opts = {}) => {
  opts.table = opts.table || "storage";
  let _connection;
  const getConnection = () => {
    if (!_connection) {
      if (!opts.url) {
        throw (0, _utils.createRequiredError)(DRIVER_NAME, "url");
      }
      _connection = (0, _database.connect)({
        url: opts.url,
        fetch: _nodeFetchNative.fetch
      });
      if (opts.boostCache) {
        _connection.execute("SET @@boost_cached_queries = true;").catch(error => {
          console.error("[unstorage] [planetscale] Failed to enable cached queries:", error);
        });
      }
    }
    return _connection;
  };
  return {
    name: DRIVER_NAME,
    options: opts,
    getInstance: getConnection,
    hasItem: async key => {
      const res = await getConnection().execute(`SELECT EXISTS (SELECT 1 FROM ${opts.table} WHERE id = :key) as value;`, {
        key
      });
      return rows(res)[0]?.value == "1";
    },
    getItem: async key => {
      const res = await getConnection().execute(`SELECT value from ${opts.table} WHERE id=:key;`, {
        key
      });
      return rows(res)[0]?.value ?? null;
    },
    setItem: async (key, value) => {
      await getConnection().execute(`INSERT INTO ${opts.table} (id, value) VALUES (:key, :value) ON DUPLICATE KEY UPDATE value = :value;`, {
        key,
        value
      });
    },
    removeItem: async key => {
      await getConnection().execute(`DELETE FROM ${opts.table} WHERE id=:key;`, {
        key
      });
    },
    getMeta: async key => {
      const res = await getConnection().execute(`SELECT created_at, updated_at from ${opts.table} WHERE id=:key;`, {
        key
      });
      return {
        birthtime: rows(res)[0]?.created_at,
        mtime: rows(res)[0]?.updated_at
      };
    },
    getKeys: async (base = "") => {
      const res = await getConnection().execute(`SELECT id from ${opts.table} WHERE id LIKE :base;`, {
        base: `${base}%`
      });
      return rows(res).map(r => r.id);
    },
    clear: async () => {
      await getConnection().execute(`DELETE FROM ${opts.table};`);
    }
  };
});
function rows(res) {
  return res.rows || [];
}