|
| 1 | +import type { TreeViewNodeTypes } from '../../../../constants.views'; |
| 2 | +import { treeViewFileNodeTypes, treeViewRefFileNodeTypes, treeViewRefNodeTypes } from '../../../../constants.views'; |
| 3 | +import type { LaunchpadItemNode } from '../../../launchpadView'; |
| 4 | +import type { RepositoryFolderNode } from '../../abstract/repositoryFolderNode'; |
| 5 | +import { ViewNode } from '../../abstract/viewNode'; |
| 6 | +import type { BranchNode } from '../../branchNode'; |
| 7 | +import type { BranchTrackingStatusFilesNode } from '../../branchTrackingStatusFilesNode'; |
| 8 | +import type { BranchTrackingStatusNode } from '../../branchTrackingStatusNode'; |
| 9 | +import type { CodeSuggestionsNode } from '../../codeSuggestionsNode'; |
| 10 | +import type { CommitFileNode } from '../../commitFileNode'; |
| 11 | +import type { CommitNode } from '../../commitNode'; |
| 12 | +import type { CompareBranchNode } from '../../compareBranchNode'; |
| 13 | +import type { CompareResultsNode } from '../../compareResultsNode'; |
| 14 | +import type { FileRevisionAsCommitNode } from '../../fileRevisionAsCommitNode'; |
| 15 | +import type { FolderNode } from '../../folderNode'; |
| 16 | +import type { LineHistoryTrackerNode } from '../../lineHistoryTrackerNode'; |
| 17 | +import type { MergeConflictFileNode } from '../../mergeConflictFileNode'; |
| 18 | +import type { PullRequestNode } from '../../pullRequestNode'; |
| 19 | +import type { RepositoryNode } from '../../repositoryNode'; |
| 20 | +import type { ResultsCommitsNode } from '../../resultsCommitsNode'; |
| 21 | +import type { ResultsFileNode } from '../../resultsFileNode'; |
| 22 | +import type { ResultsFilesNode } from '../../resultsFilesNode'; |
| 23 | +import type { StashFileNode } from '../../stashFileNode'; |
| 24 | +import type { StashNode } from '../../stashNode'; |
| 25 | +import type { StatusFileNode } from '../../statusFileNode'; |
| 26 | +import type { TagNode } from '../../tagNode'; |
| 27 | +import type { UncommittedFileNode } from '../../UncommittedFileNode'; |
| 28 | + |
| 29 | +// prettier-ignore |
| 30 | +export type TreeViewNodesByType = { |
| 31 | + [T in TreeViewNodeTypes]: T extends 'branch' |
| 32 | + ? BranchNode |
| 33 | + : T extends 'commit' |
| 34 | + ? CommitNode |
| 35 | + : T extends 'commit-file' |
| 36 | + ? CommitFileNode |
| 37 | + : T extends 'compare-branch' |
| 38 | + ? CompareBranchNode |
| 39 | + : T extends 'compare-results' |
| 40 | + ? CompareResultsNode |
| 41 | + : T extends 'conflict-file' |
| 42 | + ? MergeConflictFileNode |
| 43 | + : T extends 'drafts-code-suggestions' |
| 44 | + ? CodeSuggestionsNode |
| 45 | + : T extends 'file-commit' |
| 46 | + ? FileRevisionAsCommitNode |
| 47 | + : T extends 'folder' |
| 48 | + ? FolderNode |
| 49 | + : T extends 'launchpad-item' |
| 50 | + ? LaunchpadItemNode |
| 51 | + : T extends 'line-history-tracker' |
| 52 | + ? LineHistoryTrackerNode |
| 53 | + : T extends 'pullrequest' |
| 54 | + ? PullRequestNode |
| 55 | + : T extends 'repository' |
| 56 | + ? RepositoryNode |
| 57 | + : T extends 'repo-folder' |
| 58 | + ? RepositoryFolderNode |
| 59 | + : T extends 'results-commits' |
| 60 | + ? ResultsCommitsNode |
| 61 | + : T extends 'results-file' |
| 62 | + ? ResultsFileNode |
| 63 | + : T extends 'results-files' |
| 64 | + ? ResultsFilesNode |
| 65 | + : T extends 'stash' |
| 66 | + ? StashNode |
| 67 | + : T extends 'stash-file' |
| 68 | + ? StashFileNode |
| 69 | + : T extends 'status-file' |
| 70 | + ? StatusFileNode |
| 71 | + : T extends 'tag' |
| 72 | + ? TagNode |
| 73 | + : T extends 'tracking-status' |
| 74 | + ? BranchTrackingStatusNode |
| 75 | + : T extends 'tracking-status-files' |
| 76 | + ? BranchTrackingStatusFilesNode |
| 77 | + : T extends 'uncommitted-file' |
| 78 | + ? UncommittedFileNode |
| 79 | + : ViewNode<T>; |
| 80 | +}; |
| 81 | + |
| 82 | +type FilterNodesByType<T extends keyof TreeViewNodesByType | (keyof TreeViewNodesByType)[]> = |
| 83 | + T extends keyof TreeViewNodesByType |
| 84 | + ? TreeViewNodesByType[T] |
| 85 | + : T extends (keyof TreeViewNodesByType)[] |
| 86 | + ? TreeViewNodesByType[T[number]] |
| 87 | + : never; |
| 88 | + |
| 89 | +export function isViewNode(node: unknown): node is ViewNode; |
| 90 | +export function isViewNode<T extends keyof TreeViewNodesByType | (keyof TreeViewNodesByType)[]>( |
| 91 | + node: unknown, |
| 92 | + type: T, |
| 93 | +): node is FilterNodesByType<T>; |
| 94 | +export function isViewNode<T extends keyof TreeViewNodesByType>(node: unknown, type?: T | T[]): node is ViewNode { |
| 95 | + if (node == null || !(node instanceof ViewNode)) return false; |
| 96 | + |
| 97 | + if (type == null) return true; |
| 98 | + if (Array.isArray(type)) { |
| 99 | + return type.includes(node.type); |
| 100 | + } |
| 101 | + return node.type === type; |
| 102 | +} |
| 103 | + |
| 104 | +export function isViewFileNode(node: unknown): node is FilterNodesByType<typeof treeViewFileNodeTypes> { |
| 105 | + return isViewNode(node, treeViewFileNodeTypes); |
| 106 | +} |
| 107 | + |
| 108 | +export function isViewFileOrFolderNode( |
| 109 | + node: unknown, |
| 110 | +): node is FilterNodesByType<'folder' | (typeof treeViewFileNodeTypes)[number]> { |
| 111 | + return isViewNode(node, 'folder') || isViewNode(node, treeViewFileNodeTypes); |
| 112 | +} |
| 113 | + |
| 114 | +export function isViewRefFileNode(node: unknown): node is FilterNodesByType<typeof treeViewRefFileNodeTypes> { |
| 115 | + return isViewNode(node, treeViewRefFileNodeTypes); |
| 116 | +} |
| 117 | + |
| 118 | +export function isViewRefNode(node: unknown): node is FilterNodesByType<typeof treeViewRefNodeTypes> { |
| 119 | + return isViewNode(node, treeViewRefNodeTypes); |
| 120 | +} |
0 commit comments