init migration poc

packages/server/src/database/entities/ChatFlow.ts

@@ -0,0 +1,33 @@
+/* eslint-disable */
+import { Entity, Column, CreateDateColumn, UpdateDateColumn, PrimaryGeneratedColumn } from 'typeorm'
+import { IChatFlow } from '../../Interface'
+
+@Entity()
+export class ChatFlow implements IChatFlow {
+    @PrimaryGeneratedColumn('uuid')
+    id: string
+
+    @Column()
+    name: string
+
+    @Column({ type: 'text' })
+    flowData: string
+
+    @Column({ nullable: true })
+    deployed?: boolean
+
+    @Column({ nullable: true })
+    isPublic?: boolean
+
+    @Column({ nullable: true })
+    apikeyid?: string
+
+    @Column({ nullable: true })
+    chatbotConfig?: string
+
+    @CreateDateColumn()
+    createdDate: Date
+
+    @UpdateDateColumn()
+    updatedDate: Date
+}

packages/server/src/database/entities/ChatMessage.ts

@@ -0,0 +1,25 @@
+/* eslint-disable */
+import { Entity, Column, CreateDateColumn, PrimaryGeneratedColumn, Index } from 'typeorm'
+import { IChatMessage, MessageType } from '../../Interface'
+
+@Entity()
+export class ChatMessage implements IChatMessage {
+    @PrimaryGeneratedColumn('uuid')
+    id: string
+
+    @Column()
+    role: MessageType
+
+    @Index()
+    @Column()
+    chatflowid: string
+
+    @Column({ type: 'text' })
+    content: string
+
+    @Column({ nullable: true })
+    sourceDocuments?: string
+
+    @CreateDateColumn()
+    createdDate: Date
+}

packages/server/src/database/entities/Credential.ts

@@ -0,0 +1,24 @@
+/* eslint-disable */
+import { Entity, Column, PrimaryGeneratedColumn, Index, CreateDateColumn, UpdateDateColumn } from 'typeorm'
+import { ICredential } from '../../Interface'
+
+@Entity()
+export class Credential implements ICredential {
+    @PrimaryGeneratedColumn('uuid')
+    id: string
+
+    @Column()
+    name: string
+
+    @Column()
+    credentialName: string
+
+    @Column()
+    encryptedData: string
+
+    @CreateDateColumn()
+    createdDate: Date
+
+    @UpdateDateColumn()
+    updatedDate: Date
+}

packages/server/src/database/entities/Tool.ts

@@ -0,0 +1,33 @@
+/* eslint-disable */
+import { Entity, Column, CreateDateColumn, UpdateDateColumn, PrimaryGeneratedColumn } from 'typeorm'
+import { ITool } from '../../Interface'
+
+@Entity()
+export class Tool implements ITool {
+    @PrimaryGeneratedColumn('uuid')
+    id: string
+
+    @Column()
+    name: string
+
+    @Column({ type: 'text' })
+    description: string
+
+    @Column()
+    color: string
+
+    @Column({ nullable: true })
+    iconSrc?: string
+
+    @Column({ nullable: true })
+    schema?: string
+
+    @Column({ nullable: true })
+    func?: string
+
+    @CreateDateColumn()
+    createdDate: Date
+
+    @UpdateDateColumn()
+    updatedDate: Date
+}

packages/server/src/database/entities/index.ts

@@ -0,0 +1,11 @@
+import { ChatFlow } from './ChatFlow'
+import { ChatMessage } from './ChatMessage'
+import { Credential } from './Credential'
+import { Tool } from './Tool'
+
+export const entities = {
+    ChatFlow,
+    ChatMessage,
+    Credential,
+    Tool
+}

packages/server/src/database/migrations/1693809869231-initSqlite.ts

@@ -0,0 +1,29 @@
+import { MigrationInterface, QueryRunner } from 'typeorm'
+
+export class InitSqlite1693809869231 implements MigrationInterface {
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        console.info('started migration query')
+        await queryRunner.query(
+            `CREATE TABLE "chat_flow" ("id" varchar PRIMARY KEY NOT NULL, "name" varchar NOT NULL, "flowData" text NOT NULL, "deployed" boolean, "isPublic" boolean, "apikeyid" varchar, "chatbotConfig" varchar, "createdDate" datetime NOT NULL DEFAULT (datetime('now')), "updatedDate" datetime NOT NULL DEFAULT (datetime('now')));`
+        )
+        await queryRunner.query(
+            `CREATE TABLE "chat_message" ("id" varchar PRIMARY KEY NOT NULL, "role" varchar NOT NULL, "chatflowid" varchar NOT NULL, "content" text NOT NULL, "sourceDocuments" varchar, "createdDate" datetime NOT NULL DEFAULT (datetime('now')));`
+        )
+        await queryRunner.query(`CREATE INDEX "IDX_e574527322272fd838f4f0f3d3" ON "chat_message" ("chatflowid") ;`)
+        await queryRunner.query(
+            `CREATE TABLE "credential" ("id" varchar PRIMARY KEY NOT NULL, "name" varchar NOT NULL, "credentialName" varchar NOT NULL, "encryptedData" varchar NOT NULL, "createdDate" datetime NOT NULL DEFAULT (datetime('now')), "updatedDate" datetime NOT NULL DEFAULT (datetime('now')));`
+        )
+        await queryRunner.query(
+            `CREATE TABLE "tool" ("id" varchar PRIMARY KEY NOT NULL, "name" varchar NOT NULL, "description" text NOT NULL, "color" varchar NOT NULL, "iconSrc" varchar, "schema" varchar, "func" varchar, "createdDate" datetime NOT NULL DEFAULT (datetime('now')), "updatedDate" datetime NOT NULL DEFAULT (datetime('now')));`
+        )
+        console.info('Finish migration query')
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`DROP TABLE "chat_flow"`)
+        await queryRunner.query(`DROP INDEX "IDX_chat_messagee574527322272fd838f4f0f3d3"`)
+        await queryRunner.query(`DROP TABLE "chat_message"`)
+        await queryRunner.query(`DROP TABLE "credential"`)
+        await queryRunner.query(`DROP TABLE "tool"`)
+    }
+}