jotai/esm/urql.mjs

import { createClient } from '@urql/core';
import { atom } from 'jotai';
import { pipe, subscribe } from 'wonka';

const DEFAULT_URL = (() => {
  try {
    return process.env.JOTAI_URQL_DEFAULT_URL;
  } catch {
    return void 0;
  }
})() || "/graphql";
const clientAtom = atom(createClient({ url: DEFAULT_URL }));

const isOperationResultWithData$1 = (result) => "data" in result && !result.error;
function atomWithQuery(createQueryArgs, getClient = (get) => get(clientAtom)) {
  const queryResultAtom = atom((get) => {
    const args = createQueryArgs(get);
    if (args.pause) {
      return null;
    }
    const client = getClient(get);
    let resolve = null;
    const baseResultAtom = atom(new Promise((r) => {
      resolve = r;
    }));
    let setResult = () => {
      throw new Error("setting result without mount");
    };
    const listener = (result) => {
      if (result instanceof Promise) {
        setResult(result);
        return;
      }
      if (!isOperationResultWithData$1(result)) {
        throw new Error("result does not have data");
      }
      if (resolve) {
        resolve(result);
        resolve = null;
      } else {
        setResult(result);
      }
    };
    client.query(args.query, args.variables, {
      ...args.requestPolicy && { requestPolicy: args.requestPolicy },
      ...args.context
    }).toPromise().then(listener).catch(() => {
    });
    baseResultAtom.onMount = (update) => {
      setResult = update;
    };
    const subscriptionAtom = atom(null);
    const resultAtom = atom((get2) => get2(baseResultAtom), (get2, set, callback) => {
      const subscription = pipe(client.query(args.query, args.variables, {
        ...args.requestPolicy && { requestPolicy: args.requestPolicy },
        ...args.context
      }), subscribe(listener));
      set(subscriptionAtom, subscription);
      callback(() => {
        var _a;
        return (_a = get2(subscriptionAtom)) == null ? void 0 : _a.unsubscribe();
      });
    });
    resultAtom.onMount = (init) => {
      let cleanup;
      init((c) => {
        cleanup = c;
      });
      return cleanup;
    };
    return { args, client, resultAtom, subscriptionAtom, listener };
  });
  const queryAtom = atom((get) => {
    const queryResult = get(queryResultAtom);
    if (!queryResult) {
      return null;
    }
    const { resultAtom } = queryResult;
    return get(resultAtom);
  }, (get, set, action) => {
    switch (action.type) {
      case "reexecute": {
        const queryResult = get(queryResultAtom);
        if (!queryResult) {
          throw new Error("query is paused");
        }
        const { args, client, subscriptionAtom, listener } = queryResult;
        listener(new Promise(() => {
        }));
        const newSubscription = pipe(client.query(args.query, args.variables, {
          ...args.requestPolicy && { requestPolicy: args.requestPolicy },
          ...args.context,
          ...action.opts
        }), subscribe(listener));
        const oldSubscription = get(subscriptionAtom);
        oldSubscription == null ? void 0 : oldSubscription.unsubscribe();
        set(subscriptionAtom, newSubscription);
      }
    }
  });
  return queryAtom;
}

function atomWithMutation(createQuery, getClient = (get) => get(clientAtom)) {
  const operationResultAtom = atom(new Promise(() => {
  }));
  const queryResultAtom = atom((get) => get(operationResultAtom), (get, set, action) => {
    set(operationResultAtom, new Promise(() => {
    }));
    const client = getClient(get);
    const query = createQuery(get);
    client.mutation(query, action.variables, action.context).toPromise().then((result) => {
      var _a;
      set(operationResultAtom, result);
      (_a = action.callback) == null ? void 0 : _a.call(action, result);
    }).catch(() => {
    });
  });
  return queryResultAtom;
}

const isOperationResultWithData = (result) => "data" in result;
function atomWithSubscription(createSubscriptionArgs, getClient = (get) => get(clientAtom)) {
  const queryResultAtom = atom((get) => {
    const args = createSubscriptionArgs(get);
    if (args.pause) {
      return { args };
    }
    const client = getClient(get);
    let resolve = null;
    const resultAtom = atom(new Promise((r) => {
      resolve = r;
    }));
    let setResult = () => {
      throw new Error("setting result without mount");
    };
    let isMounted = false;
    const listener = (result) => {
      if (!isOperationResultWithData(result)) {
        throw new Error("result does not have data");
      }
      if (resolve) {
        if (!isMounted) {
          subscription == null ? void 0 : subscription.unsubscribe();
          subscription = null;
        }
        resolve(result);
        resolve = null;
      } else {
        setResult(result);
      }
    };
    let subscription = pipe(client.subscription(args.query, args.variables, args.context), subscribe(listener));
    resultAtom.onMount = (update) => {
      setResult = update;
      isMounted = true;
      if (!subscription) {
        subscription = pipe(client.subscription(args.query, args.variables, args.context), subscribe(listener));
      }
      return () => subscription == null ? void 0 : subscription.unsubscribe();
    };
    return { resultAtom, args };
  });
  const queryAtom = atom((get) => {
    const { resultAtom } = get(queryResultAtom);
    return resultAtom ? get(resultAtom) : null;
  });
  return queryAtom;
}

export { atomWithMutation, atomWithQuery, atomWithSubscription, clientAtom };