-
Notifications
You must be signed in to change notification settings - Fork 255
/
utils.ts
87 lines (78 loc) · 2.52 KB
/
utils.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
/* eslint-disable @typescript-eslint/no-explicit-any */
/**
* Copyright 2016-2019 IBM Corp. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
const debug = require('debug')('watson-middleware:utils');
import { Storage } from 'botbuilder';
import AssistantV1 = require('ibm-watson/assistant/v1');
import { Context } from 'ibm-watson/assistant/v1';
import { Payload } from './index';
const storagePrefix = 'user.';
export async function readContext(
userId: string,
storage: Storage,
): Promise<Context | null> {
const itemId = storagePrefix + userId;
try {
const result = await storage.read([itemId]);
if (typeof result[itemId] !== 'undefined' && result[itemId].context) {
debug(
'User: %s, Context: %s',
userId,
JSON.stringify(result[itemId].context, null, 2),
);
return result[itemId].context;
}
} catch (err) {
debug('User: %s, read context error: %s', userId, err);
}
return null;
}
export async function updateContext(
userId: string,
storage: Storage,
watsonResponse: { context: Context },
): Promise<{ context: Context }> {
const itemId = storagePrefix + userId;
let userData: any = {};
try {
const result = await storage.read([itemId]);
if (typeof result[itemId] !== 'undefined') {
debug(
'User: %s, Data: %s',
userId,
JSON.stringify(result[itemId], null, 2),
);
userData = result[itemId];
}
} catch (err) {
debug('User: %s, read context error: %s', userId, err);
}
userData.id = userId;
userData.context = watsonResponse.context;
let changes = {};
changes[itemId] = userData;
await storage.write(changes);
return watsonResponse;
}
export async function postMessage(
conversation: AssistantV1,
payload: Payload,
): Promise<AssistantV1.MessageResponse> {
debug('Assistant Request: %s', JSON.stringify(payload, null, 2));
const response = await conversation.message(payload);
debug('Assistant Response: %s', JSON.stringify(response, null, 2));
return response;
}