query_troubleshooter.ts 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188
  1. import _ from 'lodash';
  2. import appEvents from 'app/core/app_events';
  3. import { coreModule, JsonExplorer } from 'app/core/core';
  4. const template = `
  5. <div class="query-troubleshooter" ng-if="ctrl.isOpen">
  6. <div class="query-troubleshooter__header">
  7. <a class="pointer" ng-click="ctrl.toggleMocking()">Mock Response</a>
  8. <a class="pointer" ng-click="ctrl.toggleExpand()" ng-hide="ctrl.allNodesExpanded">
  9. <i class="fa fa-plus-square-o"></i> Expand All
  10. </a>
  11. <a class="pointer" ng-click="ctrl.toggleExpand()" ng-show="ctrl.allNodesExpanded">
  12. <i class="fa fa-minus-square-o"></i> Collapse All
  13. </a>
  14. <a class="pointer" clipboard-button="ctrl.getClipboardText()"><i class="fa fa-clipboard"></i> Copy to Clipboard</a>
  15. </div>
  16. <div class="query-troubleshooter__body" ng-hide="ctrl.isMocking">
  17. <i class="fa fa-spinner fa-spin" ng-show="ctrl.isLoading"></i>
  18. <div class="query-troubleshooter-json"></div>
  19. </div>
  20. <div class="query-troubleshooter__body" ng-show="ctrl.isMocking">
  21. <div class="gf-form p-l-1 gf-form--v-stretch">
  22. <textarea class="gf-form-input" style="width: 95%" rows="10" ng-model="ctrl.mockedResponse" placeholder="JSON"></textarea>
  23. </div>
  24. </div>
  25. </div>
  26. `;
  27. export class QueryTroubleshooterCtrl {
  28. isOpen: any;
  29. isLoading: boolean;
  30. showResponse: boolean;
  31. panelCtrl: any;
  32. renderJsonExplorer: (data) => void;
  33. onRequestErrorEventListener: any;
  34. onRequestResponseEventListener: any;
  35. hasError: boolean;
  36. allNodesExpanded: boolean;
  37. isMocking: boolean;
  38. mockedResponse: string;
  39. jsonExplorer: JsonExplorer;
  40. /** @ngInject **/
  41. constructor($scope, private $timeout) {
  42. this.onRequestErrorEventListener = this.onRequestError.bind(this);
  43. this.onRequestResponseEventListener = this.onRequestResponse.bind(this);
  44. appEvents.on('ds-request-response', this.onRequestResponseEventListener);
  45. appEvents.on('ds-request-error', this.onRequestErrorEventListener);
  46. $scope.$on('$destroy', this.removeEventsListeners.bind(this));
  47. $scope.$watch('ctrl.isOpen', this.stateChanged.bind(this));
  48. }
  49. removeEventsListeners() {
  50. appEvents.off('ds-request-response', this.onRequestResponseEventListener);
  51. appEvents.off('ds-request-error', this.onRequestErrorEventListener);
  52. }
  53. toggleMocking() {
  54. this.isMocking = !this.isMocking;
  55. }
  56. onRequestError(err) {
  57. // ignore if closed
  58. if (!this.isOpen) {
  59. return;
  60. }
  61. this.isOpen = true;
  62. this.hasError = true;
  63. this.onRequestResponse(err);
  64. }
  65. stateChanged() {
  66. if (this.isOpen) {
  67. this.panelCtrl.refresh();
  68. this.isLoading = true;
  69. }
  70. }
  71. getClipboardText(): string {
  72. if (this.jsonExplorer) {
  73. return JSON.stringify(this.jsonExplorer.json, null, 2);
  74. }
  75. return '';
  76. }
  77. handleMocking(data) {
  78. var mockedData;
  79. try {
  80. mockedData = JSON.parse(this.mockedResponse);
  81. } catch (err) {
  82. appEvents.emit('alert-error', ['Failed to parse mocked response']);
  83. return;
  84. }
  85. data.data = mockedData;
  86. }
  87. onRequestResponse(data) {
  88. // ignore if closed
  89. if (!this.isOpen) {
  90. return;
  91. }
  92. if (this.isMocking) {
  93. this.handleMocking(data);
  94. return;
  95. }
  96. this.isLoading = false;
  97. data = _.cloneDeep(data);
  98. if (data.headers) {
  99. delete data.headers;
  100. }
  101. if (data.config) {
  102. data.request = data.config;
  103. delete data.config;
  104. delete data.request.transformRequest;
  105. delete data.request.transformResponse;
  106. delete data.request.paramSerializer;
  107. delete data.request.jsonpCallbackParam;
  108. delete data.request.headers;
  109. delete data.request.requestId;
  110. delete data.request.inspect;
  111. delete data.request.retry;
  112. delete data.request.timeout;
  113. }
  114. if (data.data) {
  115. data.response = data.data;
  116. if (data.status === 200) {
  117. // if we are in error state, assume we automatically opened
  118. // and auto close it again
  119. if (this.hasError) {
  120. this.hasError = false;
  121. this.isOpen = false;
  122. }
  123. }
  124. delete data.data;
  125. delete data.status;
  126. delete data.statusText;
  127. delete data.$$config;
  128. }
  129. this.$timeout(_.partial(this.renderJsonExplorer, data));
  130. }
  131. toggleExpand(depth) {
  132. if (this.jsonExplorer) {
  133. this.allNodesExpanded = !this.allNodesExpanded;
  134. this.jsonExplorer.openAtDepth(this.allNodesExpanded ? 20 : 1);
  135. }
  136. }
  137. }
  138. export function queryTroubleshooter() {
  139. return {
  140. restrict: 'E',
  141. template: template,
  142. controller: QueryTroubleshooterCtrl,
  143. bindToController: true,
  144. controllerAs: 'ctrl',
  145. scope: {
  146. panelCtrl: '=',
  147. isOpen: '=',
  148. },
  149. link: function(scope, elem, attrs, ctrl) {
  150. ctrl.renderJsonExplorer = function(data) {
  151. var jsonElem = elem.find('.query-troubleshooter-json');
  152. ctrl.jsonExplorer = new JsonExplorer(data, 3, {
  153. animateOpen: true,
  154. });
  155. const html = ctrl.jsonExplorer.render(true);
  156. jsonElem.html(html);
  157. };
  158. },
  159. };
  160. }
  161. coreModule.directive('queryTroubleshooter', queryTroubleshooter);