read-urls-extension/treeProvider.js at master · muremwa/read-urls-extension · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
const vscode = require('vscode');
const path = require('path');

class TreeDataProvider {
    constructor(urlsTreeItems) {
		this.data = urlsTreeItems;
	};

    getTreeItem(element) {
        return element;
	};

    getChildren(element) {
        if (element === undefined) {
            return this.data;
        };
        return element.children;
	};

};

class ArgumentTreeItem extends vscode.TreeItem {
    /**
     *
     * @param {string} argumentName
     * @param {string} argumentType
     * @param {string} parent
     */
    constructor(argumentName, argumentType, parent) {
        super(`${argumentName} <${argumentType === 'NULL'? 'type_undeclared': argumentType}>`, vscode.TreeItemCollapsibleState.None);
        this.contextValue = 'args'
        this.tooltip = `URL config argument for ${parent}`;
        this.argumentName = argumentName;
    };
};


class URLConfigTreeItem extends vscode.TreeItem {
    /**
     *
     * @param {string} label
     * @param {ArgumentTreeItem[]} children
     * @param {string} urlConfigName
     * @param {boolean} keywords
     */
    constructor (label, children, urlConfigName, keywords) {
        super(urlConfigName, children && children.length > 0? vscode.TreeItemCollapsibleState.Collapsed: vscode.TreeItemCollapsibleState.None);
        this.children = children;
        this.contextValue = 'urlName';
        this.ogLabel = label;
        this.keywords = keywords;
        this.tooltip = `URL config named ${urlConfigName}`;
    };
};


class AppTreeItem extends vscode.TreeItem {
    /**
     *
     * @param {string} label
     * @param {URLConfigTreeItem[]} children
     * @param {boolean} isExtraApp
     * @param {{force: boolean, collapsed: boolean}} forceCollapse
     */
    constructor (label, children, isExtraApp, forceCollapse = {force: false, collapsed: false}) {
        const collapsedState = () => {
            let state = vscode.TreeItemCollapsibleState.None;

            if (children.length > 0) {
                if (forceCollapse.force) {
                    state = forceCollapse.collapse? vscode.TreeItemCollapsibleState.Collapsed: vscode.TreeItemCollapsibleState.Expanded;
                } else {
                    state = isExtraApp? vscode.TreeItemCollapsibleState.Collapsed: vscode.TreeItemCollapsibleState.Expanded;
                };
            };
            return state;
        };
        const labelToFeed = ((() => {
            let processedLabel = label;

            if (label.includes('READER_FILE_PATH_')) {
                const homeMatch = label.match(/READER_FILE_PATH_.*?\\(?<home>[^\\/:*?"<>|]+\\urls.py)$/);
                processedLabel = homeMatch && homeMatch.groups.home? homeMatch.groups.home: 'UNKNOWN_APP\\urls.py';
            };

            return processedLabel.toUpperCase();
        })());

        super(labelToFeed, collapsedState());
        this.children = children;
        this.contextValue = 'app'
        this.ogLabel = label;
        this.tooltip = `App called ${labelToFeed}`;
    }
}


class ProjectTreeItem extends vscode.TreeItem {
    /**
     *
     * @param {string} projectTitle
     * @param {AppTreeItem[]} children
     */
    constructor (projectTitle, children) {
        super(`${projectTitle} (PROJECT)`, vscode.TreeItemCollapsibleState.Collapsed);
        this.contextValue = 'project';
        this.tooltip = `Project called ${projectTitle}`;
        this.ogLabel = projectTitle;
        this.children = children;
        this.iconPath = {
            light: path.join(__dirname, 'media', 'folder_light.png'),
            dark: path.join(__dirname, 'media', 'folder_dark.png')
        };
    };
}


module.exports = {
    TreeDataProvider,
    ArgumentTreeItem,
    URLConfigTreeItem,
    AppTreeItem,
    ProjectTreeItem,
};