@lark-apaas/devtool-kits 1.0.11 → 1.1.0

package/dist/template/types.ts

@@ -0,0 +1,82 @@
+import { sql } from 'drizzle-orm';
+import { customType } from 'drizzle-orm/pg-core';
+
+export type UserProfile = {
+  user_id: string;
+};
+
+export const userProfile = customType<{
+  data: UserProfile;
+  driverData: unknown;
+}>({
+  dataType() {
+    return 'user_profile';
+  },
+  toDriver(value: UserProfile | string) {
+    if(typeof value === 'string') {
+      return sql`ROW(${value})::user_profile`;
+    }
+    return sql`ROW(${value.user_id})::user_profile`;
+  },
+  fromDriver(value: unknown): UserProfile {
+    if (typeof value !== 'string') {
+      throw new Error('Invalid user profile value');
+    }
+    const [userId] = value.slice(1, -1).split(',');
+    return { user_id: userId.trim() };
+  },
+});
+
+export type FileAttachment = {
+  bucket_id: string;
+  file_path: string;
+};
+
+export const fileAttachment = customType<{
+  data: FileAttachment;
+  driverData: unknown;
+}>({
+  dataType() {
+    return 'file_attachment';
+  },
+  toDriver(value: FileAttachment) {
+    return sql`ROW(${value.bucket_id},${value.file_path})::file_attachment`;
+  },
+  fromDriver(value: unknown): FileAttachment {
+    if (typeof value !== 'string') {
+      throw new Error('Invalid file attachment value');
+    }
+    const [bucketId, filePath] = value.slice(1, -1).split(',');
+    return { bucket_id: bucketId.trim(), file_path: filePath.trim() };
+  },
+});
+
+export const customTimestamptz = customType<{
+  data: Date;
+  driverData: string;
+  config: { precision?: number};
+}>({
+  dataType(config) {
+    const precision = typeof config?.precision !== 'undefined'
+      ? ` (${config.precision})`
+      : '';
+    return `timestamptz${precision}`;
+  },
+  toDriver(value: Date | string | number){
+    if(value == null) return value as any;
+    if (typeof value === 'number') {
+      return new Date(value).toISOString();
+    }
+    if(typeof value === 'string') {
+      return value;
+    }
+    if (value instanceof Date) {
+      return value.toISOString();
+    }
+    throw new Error('Invalid timestamp value');
+  },
+  fromDriver(value: string | Date): Date {
+    if(value instanceof Date) return value;
+    return new Date(value);
+  },
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lark-apaas/devtool-kits",
-  "version": "1.0.11",
+  "version": "1.1.0",
   "description": "FullStack Devtool Kits",
   "type": "module",
   "main": "./dist/index.js",