Val Town is a social website to write and deploy JavaScript.
Build APIs and schedule functions from your browser.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
import { sqlite } from "https://esm.town/v/std/sqlite";
import { sql, eq, ne, gt, gte, lt, lte, like, between, and, or, inArray, isNull, isNotNull } from "npm:drizzle-orm";
import { drizzle } from "npm:drizzle-orm/libsql";
import { sqliteTable, text, blob, integer, real, index, uniqueIndex, customType } from "npm:drizzle-orm/sqlite-core";
type ColumnType = 'TEXT' | 'INTEGER' | 'REAL' | 'BLOB' | 'JSON' | 'JSONARRAY' | 'VECTOR' | 'SELF_REFERENCE' | 'BOOLEAN' | 'TIMESTAMP' | 'TIMESTAMP_MS' | 'NUMBER';
const embedding = customType<{
data: ArrayBuffer;
configRequired: true;
config: {
size: number;
};
}>({
dataType(config) {
return `F32_BLOB(${config.size})`;
},
fromDriver(value) {
const isBuffer = Buffer.isBuffer(value);
if (!isBuffer) throw new Error('Expected a Buffer');
const arrayBuffer = value.buffer as ArrayBuffer;
return arrayBuffer;
},
toDriver(value) {
return value;
},
});
interface ColumnDefinition {
name: string;
type: ColumnType;
primaryKey?: boolean;
notNull?: boolean;
size?: number;
autoIncrement?: boolean;
}
interface IndexDefinition {
name: string;
columns: string[];
unique?: boolean;
}
interface FTSIndexDefinition {
name: string;
columns: string[];
}
export class Dobby {
private db: any;
private table: any;
private dbName: string;
private columns: ColumnDefinition[];
private indexes: IndexDefinition[];
private ftsIndexes: FTSIndexDefinition[];
constructor(dbName: string, columns: ColumnDefinition[], indexes: IndexDefinition[] = [], ftsIndexes: FTSIndexDefinition[] = []) {
this.db = drizzle(sqlite as any);
this.dbName = dbName;
this.columns = columns;
this.indexes = indexes;
this.ftsIndexes = ftsIndexes;
this.table = this.createTableSchema();
}
private createTableSchema() {
const schema: Record<string, any> = {};
this.columns.forEach(col => {
switch (col.type) {
case 'TEXT':
schema[col.name] = text(col.name);
break;
case 'INTEGER':
case 'NUMBER':
schema[col.name] = integer(col.name, { mode: 'number' });
break;
case 'BOOLEAN':
schema[col.name] = integer(col.name, { mode: 'boolean' });
break;
case 'TIMESTAMP':
schema[col.name] = integer(col.name, { mode: 'timestamp' });
break;
case 'TIMESTAMP_MS':
schema[col.name] = integer(col.name, { mode: 'timestamp_ms' });
break;
case 'REAL':
schema[col.name] = real(col.name);
break;
case 'JSON':
schema[col.name] = text(col.name, { mode: 'json' });
break;
case 'JSONARRAY':
schema[col.name] = text(col.name, { mode: 'json' }).$type<Record<string, any>[]>();
break;
case 'BLOB':
schema[col.name] = blob(col.name);
break;
case 'VECTOR':
September 10, 2024