documentSymbols.js 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. /*---------------------------------------------------------------------------------------------
  2. * Copyright (c) Microsoft Corporation. All rights reserved.
  3. * Licensed under the MIT License. See License.txt in the project root for license information.
  4. *--------------------------------------------------------------------------------------------*/
  5. var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
  6. function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
  7. return new (P || (P = Promise))(function (resolve, reject) {
  8. function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
  9. function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
  10. function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
  11. step((generator = generator.apply(thisArg, _arguments || [])).next());
  12. });
  13. };
  14. import { CancellationToken } from '../../../base/common/cancellation.js';
  15. import { assertType } from '../../../base/common/types.js';
  16. import { URI } from '../../../base/common/uri.js';
  17. import { IModelService } from '../../common/services/modelService.js';
  18. import { ITextModelService } from '../../common/services/resolverService.js';
  19. import { OutlineModel } from './outlineModel.js';
  20. import { CommandsRegistry } from '../../../platform/commands/common/commands.js';
  21. export function getDocumentSymbols(document, flat, token) {
  22. return __awaiter(this, void 0, void 0, function* () {
  23. const model = yield OutlineModel.create(document, token);
  24. return flat
  25. ? model.asListOfDocumentSymbols()
  26. : model.getTopLevelSymbols();
  27. });
  28. }
  29. CommandsRegistry.registerCommand('_executeDocumentSymbolProvider', function (accessor, ...args) {
  30. return __awaiter(this, void 0, void 0, function* () {
  31. const [resource] = args;
  32. assertType(URI.isUri(resource));
  33. const model = accessor.get(IModelService).getModel(resource);
  34. if (model) {
  35. return getDocumentSymbols(model, false, CancellationToken.None);
  36. }
  37. const reference = yield accessor.get(ITextModelService).createModelReference(resource);
  38. try {
  39. return yield getDocumentSymbols(reference.object.textEditorModel, false, CancellationToken.None);
  40. }
  41. finally {
  42. reference.dispose();
  43. }
  44. });
  45. });