-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
SourceModel.ts
104 lines (87 loc) · 2.25 KB
/
SourceModel.ts
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
101
102
103
104
import {
checkConnectionToSource,
checkConnectionToSourceForUpdate,
deleteSource,
discoverSchemaForSource,
getSource,
updateSource,
} from '../generated/AirbyteApi';
import {
CheckConnectionRead,
SourceConfiguration,
SourceDiscoverSchemaRead,
SourceId,
SourceRead,
SourceUpdate,
} from '../generated/AirbyteApi.schemas';
export class SourceModel {
meta: SourceRead;
constructor(source: SourceRead) {
this.meta = source;
}
get(key: keyof SourceRead) {
const metaKey: keyof SourceRead = key;
return this.meta[metaKey];
}
async getLatest() {
const newValues = await getSource({
sourceId: this.meta.sourceId,
});
this.meta = newValues;
}
async save() {
const sourceUpdate = this.meta as SourceUpdate;
await updateSource(sourceUpdate);
}
async update(updateBody: Partial<SourceRead>) {
this.meta = { ...this.meta, ...updateBody };
this.save();
}
async hasValidConfig(): Promise<CheckConnectionRead> {
return await checkConnectionToSource({
sourceId: this.meta.sourceId,
});
}
async validateNewConfig(
newConfig: Partial<SourceConfiguration> = {},
name?: string,
) {
const configToUpdate = {
...this.getConnectionConfiguration(),
...newConfig,
};
const nameToUpdate = name ? name : this.meta.name;
return await checkConnectionToSourceForUpdate({
sourceId: this.meta.sourceId,
name: nameToUpdate,
connectionConfiguration: configToUpdate,
});
}
async discoverSchema(): Promise<SourceDiscoverSchemaRead> {
return await discoverSchemaForSource({
sourceId: this.meta.sourceId,
});
}
async delete() {
return await deleteSource({
sourceId: this.meta.sourceId,
});
}
// Static methods
static async createSourceInstanceFromId(
sourceId: SourceId,
): Promise<SourceModel> {
const source = await getSource({ sourceId });
return new SourceModel(source);
}
static createSourceInstance(source: SourceRead): SourceModel {
return new SourceModel(source);
}
// Utility functions
toJSON(): SourceRead {
return this.meta;
}
getConnectionConfiguration(): Map<string, unknown> {
return (this.meta.connectionConfiguration as Map<string, unknown>) || {};
}
}