1 | var __assign = (this && this.__assign) || function () {
|
2 | __assign = Object.assign || function(t) {
|
3 | for (var s, i = 1, n = arguments.length; i < n; i++) {
|
4 | s = arguments[i];
|
5 | for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
|
6 | t[p] = s[p];
|
7 | }
|
8 | return t;
|
9 | };
|
10 | return __assign.apply(this, arguments);
|
11 | };
|
12 | Object.defineProperty(exports, "__esModule", { value: true });
|
13 | var graphql_1 = require("graphql");
|
14 | var WrapQuery = (function () {
|
15 | function WrapQuery(path, wrapper, extractor) {
|
16 | this.path = path;
|
17 | this.wrapper = wrapper;
|
18 | this.extractor = extractor;
|
19 | }
|
20 | WrapQuery.prototype.transformRequest = function (originalRequest) {
|
21 | var _this = this;
|
22 | var _a;
|
23 | var document = originalRequest.document;
|
24 | var fieldPath = [];
|
25 | var ourPath = JSON.stringify(this.path);
|
26 | var newDocument = graphql_1.visit(document, (_a = {},
|
27 | _a[graphql_1.Kind.FIELD] = {
|
28 | enter: function (node) {
|
29 | fieldPath.push(node.name.value);
|
30 | if (ourPath === JSON.stringify(fieldPath)) {
|
31 | var wrapResult = _this.wrapper(node.selectionSet);
|
32 |
|
33 |
|
34 | var selectionSet = wrapResult.kind === graphql_1.Kind.SELECTION_SET
|
35 | ? wrapResult
|
36 | : {
|
37 | kind: graphql_1.Kind.SELECTION_SET,
|
38 | selections: [wrapResult]
|
39 | };
|
40 | return __assign({}, node, { selectionSet: selectionSet });
|
41 | }
|
42 | },
|
43 | leave: function (node) {
|
44 | fieldPath.pop();
|
45 | }
|
46 | },
|
47 | _a));
|
48 | return __assign({}, originalRequest, { document: newDocument });
|
49 | };
|
50 | WrapQuery.prototype.transformResult = function (originalResult) {
|
51 | var rootData = originalResult.data;
|
52 | if (rootData) {
|
53 | var data = rootData;
|
54 | var path = this.path.slice();
|
55 | while (path.length > 1) {
|
56 | var next = path.shift();
|
57 | if (data[next]) {
|
58 | data = data[next];
|
59 | }
|
60 | }
|
61 | data[path[0]] = this.extractor(data[path[0]]);
|
62 | }
|
63 | return {
|
64 | data: rootData,
|
65 | errors: originalResult.errors
|
66 | };
|
67 | };
|
68 | return WrapQuery;
|
69 | }());
|
70 | exports.default = WrapQuery;
|
71 |
|
\ | No newline at end of file |