@ai0x0/utils 0.0.8 → 0.0.9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,5 +1,11 @@
1
1
  function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
2
2
  function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return e; }; var t, e = {}, r = Object.prototype, n = r.hasOwnProperty, o = Object.defineProperty || function (t, e, r) { t[e] = r.value; }, i = "function" == typeof Symbol ? Symbol : {}, a = i.iterator || "@@iterator", c = i.asyncIterator || "@@asyncIterator", u = i.toStringTag || "@@toStringTag"; function define(t, e, r) { return Object.defineProperty(t, e, { value: r, enumerable: !0, configurable: !0, writable: !0 }), t[e]; } try { define({}, ""); } catch (t) { define = function define(t, e, r) { return t[e] = r; }; } function wrap(t, e, r, n) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype), c = new Context(n || []); return o(a, "_invoke", { value: makeInvokeMethod(t, r, c) }), a; } function tryCatch(t, e, r) { try { return { type: "normal", arg: t.call(e, r) }; } catch (t) { return { type: "throw", arg: t }; } } e.wrap = wrap; var h = "suspendedStart", l = "suspendedYield", f = "executing", s = "completed", y = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var p = {}; define(p, a, function () { return this; }); var d = Object.getPrototypeOf, v = d && d(d(values([]))); v && v !== r && n.call(v, a) && (p = v); var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p); function defineIteratorMethods(t) { ["next", "throw", "return"].forEach(function (e) { define(t, e, function (t) { return this._invoke(e, t); }); }); } function AsyncIterator(t, e) { function invoke(r, o, i, a) { var c = tryCatch(t[r], t, o); if ("throw" !== c.type) { var u = c.arg, h = u.value; return h && "object" == _typeof(h) && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) { invoke("next", t, i, a); }, function (t) { invoke("throw", t, i, a); }) : e.resolve(h).then(function (t) { u.value = t, i(u); }, function (t) { return invoke("throw", t, i, a); }); } a(c.arg); } var r; o(this, "_invoke", { value: function value(t, n) { function callInvokeWithMethodAndArg() { return new e(function (e, r) { invoke(t, n, e, r); }); } return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(e, r, n) { var o = h; return function (i, a) { if (o === f) throw new Error("Generator is already running"); if (o === s) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var c = n.delegate; if (c) { var u = maybeInvokeDelegate(c, n); if (u) { if (u === y) continue; return u; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (o === h) throw o = s, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = f; var p = tryCatch(e, r, n); if ("normal" === p.type) { if (o = n.done ? s : l, p.arg === y) continue; return { value: p.arg, done: n.done }; } "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg); } }; } function maybeInvokeDelegate(e, r) { var n = r.method, o = e.iterator[n]; if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y; var i = tryCatch(o, e.iterator, r.arg); if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y; var a = i.arg; return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y); } function pushTryEntry(t) { var e = { tryLoc: t[0] }; 1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e); } function resetTryEntry(t) { var e = t.completion || {}; e.type = "normal", delete e.arg, t.completion = e; } function Context(t) { this.tryEntries = [{ tryLoc: "root" }], t.forEach(pushTryEntry, this), this.reset(!0); } function values(e) { if (e || "" === e) { var r = e[a]; if (r) return r.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) { var o = -1, i = function next() { for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next; return next.value = t, next.done = !0, next; }; return i.next = i; } } throw new TypeError(_typeof(e) + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), o(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) { var e = "function" == typeof t && t.constructor; return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name)); }, e.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t; }, e.awrap = function (t) { return { __await: t }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () { return this; }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(wrap(t, r, n, o), i); return e.isGeneratorFunction(r) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () { return this; }), define(g, "toString", function () { return "[object Generator]"; }), e.keys = function (t) { var e = Object(t), r = []; for (var n in e) r.push(n); return r.reverse(), function next() { for (; r.length;) { var t = r.pop(); if (t in e) return next.value = t, next.done = !1, next; } return next.done = !0, next; }; }, e.values = values, Context.prototype = { constructor: Context, reset: function reset(e) { if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0].completion; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(e) { if (this.done) throw e; var r = this; function handle(n, o) { return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o; } for (var o = this.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i.completion; if ("root" === i.tryLoc) return handle("end"); if (i.tryLoc <= this.prev) { var c = n.call(i, "catchLoc"), u = n.call(i, "finallyLoc"); if (c && u) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } else if (c) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); } else { if (!u) throw new Error("try statement without catch or finally"); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } } } }, abrupt: function abrupt(t, e) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var o = this.tryEntries[r]; if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) { var i = o; break; } } i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null); var a = i ? i.completion : {}; return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a); }, complete: function complete(t, e) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y; }, finish: function finish(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y; } }, catch: function _catch(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.tryLoc === t) { var n = r.completion; if ("throw" === n.type) { var o = n.arg; resetTryEntry(r); } return o; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(e, r, n) { return this.delegate = { iterator: values(e), resultName: r, nextLoc: n }, "next" === this.method && (this.arg = t), y; } }, e; }
3
+ function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
4
+ function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
5
+ function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
6
+ function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
7
+ function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
8
+ function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
3
9
  function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
4
10
  function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
5
11
  import { z } from "zod";
@@ -14,6 +20,9 @@ export function createPutAction(_ref) {
14
20
  var _ref3,
15
21
  _ref3$byCreator,
16
22
  byCreator,
23
+ _data,
24
+ _yield$db$update$set$,
25
+ _yield$db$update$set$2,
17
26
  data,
18
27
  _args = arguments;
19
28
  return _regeneratorRuntime().wrap(function _callee$(_context) {
@@ -37,15 +46,21 @@ export function createPutAction(_ref) {
37
46
  id: body.id
38
47
  });
39
48
  case 4:
40
- data = _context.sent;
41
- if (data) {
49
+ _data = _context.sent;
50
+ if (_data) {
42
51
  _context.next = 7;
43
52
  break;
44
53
  }
45
54
  throw Error("未找到编辑对象,或没有权限");
46
55
  case 7:
47
- return _context.abrupt("return", db.update(table).set(transformBody(body)).where(eq(table.id, body.id)).returning());
48
- case 8:
56
+ _context.next = 9;
57
+ return db.update(table).set(transformBody(body)).where(eq(table.id, body.id)).returning();
58
+ case 9:
59
+ _yield$db$update$set$ = _context.sent;
60
+ _yield$db$update$set$2 = _slicedToArray(_yield$db$update$set$, 1);
61
+ data = _yield$db$update$set$2[0];
62
+ return _context.abrupt("return", data);
63
+ case 13:
49
64
  case "end":
50
65
  return _context.stop();
51
66
  }
@@ -1,12 +1,15 @@
1
1
  import { z, ZodSchema } from "zod";
2
2
  import { PgTable } from "drizzle-orm/pg-core";
3
3
  import { NextRequest } from "next/server";
4
+ import { TypedNextResponse } from "next-rest-framework";
4
5
  import { NodePgDatabase } from "drizzle-orm/node-postgres";
5
- export interface PostOperationOptions<T extends ZodSchema, TTable extends PgTable> {
6
- bodySchema: T;
7
- setBody?: (req: NextRequest) => Promise<Partial<z.infer<T>>>;
6
+ export interface PostOperationOptions<IB extends ZodSchema, OB extends ZodSchema, TTable extends PgTable> {
7
+ bodySchema: IB;
8
+ outputBodySchema: IB;
9
+ setBody?: (req: NextRequest) => Promise<Partial<z.infer<IB>>>;
8
10
  summary?: string;
9
- onSuccess?: (data: z.infer<T>) => Promise<z.infer<T>>;
11
+ onSuccess?: (data: z.infer<OB>) => Promise<z.infer<OB>>;
12
+ onError?: (error: Error) => Promise<ReturnType<(typeof TypedNextResponse)["json"]> | undefined>;
10
13
  table: TTable;
11
14
  }
12
15
  export declare const createPostOperation: ({ getSession, db, }: {
@@ -14,4 +17,4 @@ export declare const createPostOperation: ({ getSession, db, }: {
14
17
  getSession: (req: NextRequest) => Promise<{
15
18
  userId?: string;
16
19
  } | undefined>;
17
- }) => <T extends z.ZodType<any, z.ZodTypeDef, any>, TTable extends PgTable<import("drizzle-orm/pg-core").TableConfig>>({ bodySchema, setBody, summary, table, onSuccess, }: PostOperationOptions<T, TTable>) => any;
20
+ }) => <IB extends z.ZodType<any, z.ZodTypeDef, any>, OB extends z.ZodType<any, z.ZodTypeDef, any>, TTable extends PgTable<import("drizzle-orm/pg-core").TableConfig>>({ bodySchema, outputBodySchema, setBody, summary, table, onSuccess, onError, }: PostOperationOptions<IB, OB, TTable>) => any;
@@ -22,10 +22,12 @@ export var createPostOperation = function createPostOperation(_ref) {
22
22
  db = _ref.db;
23
23
  return function (_ref2) {
24
24
  var bodySchema = _ref2.bodySchema,
25
+ outputBodySchema = _ref2.outputBodySchema,
25
26
  setBody = _ref2.setBody,
26
27
  summary = _ref2.summary,
27
28
  table = _ref2.table,
28
- onSuccess = _ref2.onSuccess;
29
+ onSuccess = _ref2.onSuccess,
30
+ onError = _ref2.onError;
29
31
  return routeOperation({
30
32
  method: "POST",
31
33
  openApiOperation: {
@@ -36,47 +38,48 @@ export var createPostOperation = function createPostOperation(_ref) {
36
38
  body: bodySchema,
37
39
  contentType: "application/json"
38
40
  }).outputs([{
39
- body: z.object({
41
+ body: outputBodySchema || z.object({
40
42
  id: z.string()
41
43
  }),
42
44
  contentType: "application/json",
43
45
  status: 200
44
46
  }]).handler( /*#__PURE__*/function () {
45
47
  var _ref3 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(req) {
46
- var _ref4, userId, body, _yield$createPostActi, _yield$createPostActi2, data;
48
+ var _ref4, userId, body, _yield$createPostActi, _yield$createPostActi2, _data, response;
47
49
  return _regeneratorRuntime().wrap(function _callee$(_context) {
48
50
  while (1) switch (_context.prev = _context.next) {
49
51
  case 0:
50
- _context.next = 2;
52
+ _context.prev = 0;
53
+ _context.next = 3;
51
54
  return getSession(req);
52
- case 2:
55
+ case 3:
53
56
  _context.t0 = _context.sent;
54
57
  if (_context.t0) {
55
- _context.next = 5;
58
+ _context.next = 6;
56
59
  break;
57
60
  }
58
61
  _context.t0 = {};
59
- case 5:
62
+ case 6:
60
63
  _ref4 = _context.t0;
61
64
  userId = _ref4.userId;
62
65
  _context.t1 = Object;
63
- _context.next = 10;
66
+ _context.next = 11;
64
67
  return req.json();
65
- case 10:
68
+ case 11:
66
69
  _context.t2 = _context.sent;
67
- _context.next = 13;
70
+ _context.next = 14;
68
71
  return setBody === null || setBody === void 0 ? void 0 : setBody(req);
69
- case 13:
72
+ case 14:
70
73
  _context.t3 = _context.sent;
71
74
  if (_context.t3) {
72
- _context.next = 16;
75
+ _context.next = 17;
73
76
  break;
74
77
  }
75
78
  _context.t3 = {};
76
- case 16:
79
+ case 17:
77
80
  _context.t4 = _context.t3;
78
81
  body = _context.t1.assign.call(_context.t1, _context.t2, _context.t4);
79
- _context.next = 20;
82
+ _context.next = 21;
80
83
  return createPostAction({
81
84
  bodySchema: bodySchema,
82
85
  db: db,
@@ -84,27 +87,41 @@ export var createPostOperation = function createPostOperation(_ref) {
84
87
  })(_objectSpread(_objectSpread({}, body), {}, {
85
88
  creatorId: userId
86
89
  }));
87
- case 20:
90
+ case 21:
88
91
  _yield$createPostActi = _context.sent;
89
92
  _yield$createPostActi2 = _slicedToArray(_yield$createPostActi, 1);
90
- data = _yield$createPostActi2[0];
93
+ _data = _yield$createPostActi2[0];
91
94
  if (!onSuccess) {
92
- _context.next = 27;
95
+ _context.next = 28;
93
96
  break;
94
97
  }
95
- _context.next = 26;
96
- return onSuccess(data);
97
- case 26:
98
- data = _context.sent;
98
+ _context.next = 27;
99
+ return onSuccess(_data);
99
100
  case 27:
100
- return _context.abrupt("return", TypedNextResponse.json(data, {
101
+ _data = _context.sent;
102
+ case 28:
103
+ return _context.abrupt("return", TypedNextResponse.json(_data, {
101
104
  status: 200
102
105
  }));
103
- case 28:
106
+ case 31:
107
+ _context.prev = 31;
108
+ _context.t5 = _context["catch"](0);
109
+ _context.next = 35;
110
+ return onError === null || onError === void 0 ? void 0 : onError(_context.t5);
111
+ case 35:
112
+ response = _context.sent;
113
+ if (!response) {
114
+ _context.next = 40;
115
+ break;
116
+ }
117
+ return _context.abrupt("return", response);
118
+ case 40:
119
+ throw _context.t5;
120
+ case 41:
104
121
  case "end":
105
122
  return _context.stop();
106
123
  }
107
- }, _callee);
124
+ }, _callee, null, [[0, 31]]);
108
125
  }));
109
126
  return function (_x) {
110
127
  return _ref3.apply(this, arguments);
@@ -3,12 +3,13 @@ import { PgTable } from "drizzle-orm/pg-core";
3
3
  import { NodePgDatabase } from "drizzle-orm/node-postgres";
4
4
  import { NextRequest } from "next/server";
5
5
  import { BaseTable } from "../types";
6
- export interface PutOperationOptions<T extends ZodSchema, TTable extends PgTable> {
7
- bodySchema: T;
6
+ export interface PutOperationOptions<IB extends ZodSchema, OB extends ZodSchema, TTable extends PgTable> {
7
+ bodySchema: IB;
8
+ outputBodySchema?: OB;
8
9
  table: TTable;
9
10
  summary?: string;
10
- setBody?: (req: NextRequest) => Promise<Partial<z.infer<T>>>;
11
- onSuccess?: (data: z.infer<T>) => Promise<z.infer<T>>;
11
+ setBody?: (req: NextRequest) => Promise<Partial<z.infer<IB>>>;
12
+ onSuccess?: (data: z.infer<OB>) => Promise<z.infer<OB>>;
12
13
  byCreator?: boolean;
13
14
  }
14
15
  export declare const createPutOperation: ({ getSession, db, }: {
@@ -16,4 +17,4 @@ export declare const createPutOperation: ({ getSession, db, }: {
16
17
  userId?: string;
17
18
  } | undefined>;
18
19
  db: NodePgDatabase<any>;
19
- }) => <T extends z.ZodType<any, z.ZodTypeDef, any>, TTable extends BaseTable>({ bodySchema, table, summary, setBody, onSuccess, byCreator, }: PutOperationOptions<T, TTable>) => any;
20
+ }) => <IB extends z.ZodType<any, z.ZodTypeDef, any>, OB extends z.ZodType<any, z.ZodTypeDef, any>, TTable extends BaseTable>({ bodySchema, outputBodySchema, table, summary, setBody, onSuccess, byCreator, }: PutOperationOptions<IB, OB, TTable>) => any;
@@ -7,6 +7,7 @@ function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol"
7
7
  function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
8
8
  function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
9
9
  function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
10
+ import { z } from "zod";
10
11
  import { routeOperation, TypedNextResponse } from "next-rest-framework";
11
12
  import getTableName from "./get-table-name";
12
13
  import { createPutAction } from "../actions";
@@ -15,6 +16,7 @@ export var createPutOperation = function createPutOperation(_ref) {
15
16
  db = _ref.db;
16
17
  return function (_ref2) {
17
18
  var bodySchema = _ref2.bodySchema,
19
+ outputBodySchema = _ref2.outputBodySchema,
18
20
  table = _ref2.table,
19
21
  summary = _ref2.summary,
20
22
  setBody = _ref2.setBody,
@@ -30,7 +32,11 @@ export var createPutOperation = function createPutOperation(_ref) {
30
32
  }).input({
31
33
  body: bodySchema,
32
34
  contentType: "application/json"
33
- }).handler( /*#__PURE__*/function () {
35
+ }).outputs([{
36
+ body: outputBodySchema || z.void(),
37
+ status: 200,
38
+ contentType: "application/json"
39
+ }]).handler( /*#__PURE__*/function () {
34
40
  var _ref3 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(req) {
35
41
  var _ref4, userId, body, data;
36
42
  return _regeneratorRuntime().wrap(function _callee$(_context) {
@@ -34,7 +34,7 @@ function createPutAction({
34
34
  byCreator = true
35
35
  } = {}) => {
36
36
  if (byCreator) {
37
- const data = await (0, import_backend.createGetAction)({
37
+ const data2 = await (0, import_backend.createGetAction)({
38
38
  db,
39
39
  table,
40
40
  bodySchema: import_zod.z.object({
@@ -42,11 +42,12 @@ function createPutAction({
42
42
  id: import_zod.z.string()
43
43
  })
44
44
  })({ creatorId: body.editorId, id: body.id });
45
- if (!data) {
45
+ if (!data2) {
46
46
  throw Error("未找到编辑对象,或没有权限");
47
47
  }
48
48
  }
49
- return db.update(table).set((0, import_transform_body.transformBody)(body)).where((0, import_drizzle_orm.eq)(table.id, body.id)).returning();
49
+ const [data] = await db.update(table).set((0, import_transform_body.transformBody)(body)).where((0, import_drizzle_orm.eq)(table.id, body.id)).returning();
50
+ return data;
50
51
  };
51
52
  }
52
53
  // Annotate the CommonJS export names for ESM import in node:
@@ -1,12 +1,15 @@
1
1
  import { z, ZodSchema } from "zod";
2
2
  import { PgTable } from "drizzle-orm/pg-core";
3
3
  import { NextRequest } from "next/server";
4
+ import { TypedNextResponse } from "next-rest-framework";
4
5
  import { NodePgDatabase } from "drizzle-orm/node-postgres";
5
- export interface PostOperationOptions<T extends ZodSchema, TTable extends PgTable> {
6
- bodySchema: T;
7
- setBody?: (req: NextRequest) => Promise<Partial<z.infer<T>>>;
6
+ export interface PostOperationOptions<IB extends ZodSchema, OB extends ZodSchema, TTable extends PgTable> {
7
+ bodySchema: IB;
8
+ outputBodySchema: IB;
9
+ setBody?: (req: NextRequest) => Promise<Partial<z.infer<IB>>>;
8
10
  summary?: string;
9
- onSuccess?: (data: z.infer<T>) => Promise<z.infer<T>>;
11
+ onSuccess?: (data: z.infer<OB>) => Promise<z.infer<OB>>;
12
+ onError?: (error: Error) => Promise<ReturnType<(typeof TypedNextResponse)["json"]> | undefined>;
10
13
  table: TTable;
11
14
  }
12
15
  export declare const createPostOperation: ({ getSession, db, }: {
@@ -14,4 +17,4 @@ export declare const createPostOperation: ({ getSession, db, }: {
14
17
  getSession: (req: NextRequest) => Promise<{
15
18
  userId?: string;
16
19
  } | undefined>;
17
- }) => <T extends z.ZodType<any, z.ZodTypeDef, any>, TTable extends PgTable<import("drizzle-orm/pg-core").TableConfig>>({ bodySchema, setBody, summary, table, onSuccess, }: PostOperationOptions<T, TTable>) => any;
20
+ }) => <IB extends z.ZodType<any, z.ZodTypeDef, any>, OB extends z.ZodType<any, z.ZodTypeDef, any>, TTable extends PgTable<import("drizzle-orm/pg-core").TableConfig>>({ bodySchema, outputBodySchema, setBody, summary, table, onSuccess, onError, }: PostOperationOptions<IB, OB, TTable>) => any;
@@ -41,10 +41,12 @@ var createPostOperation = ({
41
41
  db
42
42
  }) => ({
43
43
  bodySchema,
44
+ outputBodySchema,
44
45
  setBody,
45
46
  summary,
46
47
  table,
47
- onSuccess
48
+ onSuccess,
49
+ onError
48
50
  }) => (0, import_next_rest_framework.routeOperation)({
49
51
  method: "POST",
50
52
  openApiOperation: {
@@ -56,24 +58,33 @@ var createPostOperation = ({
56
58
  contentType: "application/json"
57
59
  }).outputs([
58
60
  {
59
- body: import_zod.z.object({ id: import_zod.z.string() }),
61
+ body: outputBodySchema || import_zod.z.object({ id: import_zod.z.string() }),
60
62
  contentType: "application/json",
61
63
  status: 200
62
64
  }
63
65
  ]).handler(async (req) => {
64
- const { userId } = await getSession(req) || {};
65
- const body = Object.assign(
66
- await req.json(),
67
- await (setBody == null ? void 0 : setBody(req)) || {}
68
- );
69
- let [data] = await (0, import_actions.createPostAction)({ bodySchema, db, table })({
70
- ...body,
71
- creatorId: userId
72
- });
73
- if (onSuccess) {
74
- data = await onSuccess(data);
66
+ try {
67
+ const { userId } = await getSession(req) || {};
68
+ const body = Object.assign(
69
+ await req.json(),
70
+ await (setBody == null ? void 0 : setBody(req)) || {}
71
+ );
72
+ let [data] = await (0, import_actions.createPostAction)({ bodySchema, db, table })({
73
+ ...body,
74
+ creatorId: userId
75
+ });
76
+ if (onSuccess) {
77
+ data = await onSuccess(data);
78
+ }
79
+ return import_next_rest_framework.TypedNextResponse.json(data, { status: 200 });
80
+ } catch (e) {
81
+ const response = await (onError == null ? void 0 : onError(e));
82
+ if (response) {
83
+ return response;
84
+ } else {
85
+ throw e;
86
+ }
75
87
  }
76
- return import_next_rest_framework.TypedNextResponse.json(data, { status: 200 });
77
88
  });
78
89
  // Annotate the CommonJS export names for ESM import in node:
79
90
  0 && (module.exports = {
@@ -3,12 +3,13 @@ import { PgTable } from "drizzle-orm/pg-core";
3
3
  import { NodePgDatabase } from "drizzle-orm/node-postgres";
4
4
  import { NextRequest } from "next/server";
5
5
  import { BaseTable } from "../types";
6
- export interface PutOperationOptions<T extends ZodSchema, TTable extends PgTable> {
7
- bodySchema: T;
6
+ export interface PutOperationOptions<IB extends ZodSchema, OB extends ZodSchema, TTable extends PgTable> {
7
+ bodySchema: IB;
8
+ outputBodySchema?: OB;
8
9
  table: TTable;
9
10
  summary?: string;
10
- setBody?: (req: NextRequest) => Promise<Partial<z.infer<T>>>;
11
- onSuccess?: (data: z.infer<T>) => Promise<z.infer<T>>;
11
+ setBody?: (req: NextRequest) => Promise<Partial<z.infer<IB>>>;
12
+ onSuccess?: (data: z.infer<OB>) => Promise<z.infer<OB>>;
12
13
  byCreator?: boolean;
13
14
  }
14
15
  export declare const createPutOperation: ({ getSession, db, }: {
@@ -16,4 +17,4 @@ export declare const createPutOperation: ({ getSession, db, }: {
16
17
  userId?: string;
17
18
  } | undefined>;
18
19
  db: NodePgDatabase<any>;
19
- }) => <T extends z.ZodType<any, z.ZodTypeDef, any>, TTable extends BaseTable>({ bodySchema, table, summary, setBody, onSuccess, byCreator, }: PutOperationOptions<T, TTable>) => any;
20
+ }) => <IB extends z.ZodType<any, z.ZodTypeDef, any>, OB extends z.ZodType<any, z.ZodTypeDef, any>, TTable extends BaseTable>({ bodySchema, outputBodySchema, table, summary, setBody, onSuccess, byCreator, }: PutOperationOptions<IB, OB, TTable>) => any;
@@ -32,6 +32,7 @@ __export(put_operation_exports, {
32
32
  createPutOperation: () => createPutOperation
33
33
  });
34
34
  module.exports = __toCommonJS(put_operation_exports);
35
+ var import_zod = require("zod");
35
36
  var import_next_rest_framework = require("next-rest-framework");
36
37
  var import_get_table_name = __toESM(require("./get-table-name"));
37
38
  var import_actions = require("../actions");
@@ -40,6 +41,7 @@ var createPutOperation = ({
40
41
  db
41
42
  }) => ({
42
43
  bodySchema,
44
+ outputBodySchema,
43
45
  table,
44
46
  summary,
45
47
  setBody,
@@ -54,7 +56,13 @@ var createPutOperation = ({
54
56
  }).input({
55
57
  body: bodySchema,
56
58
  contentType: "application/json"
57
- }).handler(async (req) => {
59
+ }).outputs([
60
+ {
61
+ body: outputBodySchema || import_zod.z.void(),
62
+ status: 200,
63
+ contentType: "application/json"
64
+ }
65
+ ]).handler(async (req) => {
58
66
  const { userId } = await getSession(req) || {};
59
67
  const body = Object.assign(
60
68
  await req.json(),
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@ai0x0/utils",
3
- "version": "0.0.8",
3
+ "version": "0.0.9",
4
4
  "description": "AI0x0 utils",
5
5
  "keywords": [
6
6
  "ai0x0"