diff --git a/packages/core/src/composables/useActiveEditorDecorations.ts b/packages/core/src/composables/useActiveEditorDecorations.ts index 448627b..d51d7e8 100644 --- a/packages/core/src/composables/useActiveEditorDecorations.ts +++ b/packages/core/src/composables/useActiveEditorDecorations.ts @@ -1,5 +1,7 @@ -import type { MaybeRefOrGetter } from '@reactive-vscode/reactivity' -import type { DecorationOptions, DecorationRenderOptions, Range, TextEditorDecorationType } from 'vscode' +import type { MaybeRef, MaybeRefOrGetter } from '@reactive-vscode/reactivity' +import type { DecorationOptions, DecorationRenderOptions, Range, TextEditor, TextEditorDecorationType } from 'vscode' +import type { Awaitable } from '../utils' +import type { UseEditorDecorationsOptions } from './useEditorDecorations' import { useActiveTextEditor } from './useActiveTextEditor' import { useEditorDecorations } from './useEditorDecorations' @@ -10,9 +12,12 @@ import { useEditorDecorations } from './useEditorDecorations' */ export function useActiveEditorDecorations( decorationTypeOrOptions: MaybeRefOrGetter, - rangesOrOptions: MaybeRefOrGetter, + decorations: + | MaybeRef + | ((editor: TextEditor) => Awaitable), + options: UseEditorDecorationsOptions = {}, ) { const activeEditor = useActiveTextEditor() - useEditorDecorations(activeEditor, decorationTypeOrOptions, rangesOrOptions) + return useEditorDecorations(activeEditor, decorationTypeOrOptions, decorations, options) }