e621: fix order for pools
[fanfix.git] / src / jexer / ttree / TDirectoryTreeItem.java
... / ...
CommitLineData
1/*
2 * Jexer - Java Text User Interface
3 *
4 * The MIT License (MIT)
5 *
6 * Copyright (C) 2019 Kevin Lamonte
7 *
8 * Permission is hereby granted, free of charge, to any person obtaining a
9 * copy of this software and associated documentation files (the "Software"),
10 * to deal in the Software without restriction, including without limitation
11 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
12 * and/or sell copies of the Software, and to permit persons to whom the
13 * Software is furnished to do so, subject to the following conditions:
14 *
15 * The above copyright notice and this permission notice shall be included in
16 * all copies or substantial portions of the Software.
17 *
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
19 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
20 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
21 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
22 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
23 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
24 * DEALINGS IN THE SOFTWARE.
25 *
26 * @author Kevin Lamonte [kevin.lamonte@gmail.com]
27 * @version 1
28 */
29package jexer.ttree;
30
31import java.io.File;
32import java.io.IOException;
33import java.util.Collections;
34import java.util.List;
35import java.util.LinkedList;
36
37import jexer.TWidget;
38
39/**
40 * TDirectoryTreeItem is a single item in a disk directory tree view.
41 */
42public class TDirectoryTreeItem extends TTreeItem {
43
44 // ------------------------------------------------------------------------
45 // Variables --------------------------------------------------------------
46 // ------------------------------------------------------------------------
47
48 /**
49 * File corresponding to this list item.
50 */
51 private File file;
52
53 /**
54 * The TTreeViewWidget containing this directory tree.
55 */
56 private TTreeViewWidget treeViewWidget;
57
58 // ------------------------------------------------------------------------
59 // Constructors -----------------------------------------------------------
60 // ------------------------------------------------------------------------
61
62 /**
63 * Public constructor.
64 *
65 * @param view root TTreeViewWidget
66 * @param text text for this item
67 * @param expanded if true, have it expanded immediately
68 * @throws IOException if a java.io operation throws
69 */
70 public TDirectoryTreeItem(final TTreeViewWidget view, final String text,
71 final boolean expanded) throws IOException {
72
73 this(view, text, expanded, true);
74 }
75
76 /**
77 * Public constructor.
78 *
79 * @param view root TTreeViewWidget
80 * @param text text for this item
81 * @param expanded if true, have it expanded immediately
82 * @param openParents if true, expand all paths up the root path and
83 * return the root path entry
84 * @throws IOException if a java.io operation throws
85 */
86 public TDirectoryTreeItem(final TTreeViewWidget view, final String text,
87 final boolean expanded, final boolean openParents) throws IOException {
88
89 super(view.getTreeView(), text, false);
90
91 this.treeViewWidget = view;
92
93 List<String> parentFiles = new LinkedList<String>();
94 boolean oldExpanded = expanded;
95
96 // Convert to canonical path
97 File rootFile = new File(text);
98 rootFile = rootFile.getCanonicalFile();
99
100 if (openParents) {
101 setExpanded(true);
102
103 // Go up the directory tree
104 File parent = rootFile.getParentFile();
105 while (parent != null) {
106 parentFiles.add(rootFile.getName());
107 rootFile = rootFile.getParentFile();
108 parent = rootFile.getParentFile();
109 }
110 }
111 file = rootFile;
112 if (rootFile.getParentFile() == null) {
113 // This is a filesystem root, use its full name
114 setText(rootFile.getCanonicalPath());
115 } else {
116 // This is a relative path. We got here because openParents was
117 // false.
118 assert (!openParents);
119 setText(rootFile.getName());
120 }
121 onExpand();
122
123 if (openParents) {
124 TDirectoryTreeItem childFile = this;
125 Collections.reverse(parentFiles);
126 for (String p: parentFiles) {
127 for (TWidget widget: childFile.getChildren()) {
128 TDirectoryTreeItem child = (TDirectoryTreeItem) widget;
129 if (child.getText().equals(p)) {
130 childFile = child;
131 childFile.setExpanded(true);
132 childFile.onExpand();
133 break;
134 }
135 }
136 }
137 unselect();
138 getTreeView().setSelected(childFile, true);
139 setExpanded(oldExpanded);
140 }
141
142 view.reflowData();
143 }
144
145 // ------------------------------------------------------------------------
146 // TTreeItem --------------------------------------------------------------
147 // ------------------------------------------------------------------------
148
149 /**
150 * Get the File corresponding to this list item.
151 *
152 * @return the File
153 */
154 public final File getFile() {
155 return file;
156 }
157
158 /**
159 * Called when this item is expanded or collapsed. this.expanded will be
160 * true if this item was just expanded from a mouse click or keypress.
161 */
162 @Override
163 public final void onExpand() {
164 // System.err.printf("onExpand() %s\n", file);
165
166 if (file == null) {
167 return;
168 }
169 getChildren().clear();
170
171 // Make sure we can read it before trying to.
172 if (file.canRead()) {
173 setSelectable(true);
174 } else {
175 setSelectable(false);
176 }
177 assert (file.isDirectory());
178 setExpandable(true);
179
180 if (!isExpanded() || !isExpandable()) {
181 return;
182 }
183
184 File [] listFiles = file.listFiles();
185 if (listFiles != null) {
186 for (File f: listFiles) {
187 // System.err.printf(" -> file %s %s\n", file, file.getName());
188
189 if (f.getName().startsWith(".")) {
190 // Hide dot-files
191 continue;
192 }
193 if (!f.isDirectory()) {
194 continue;
195 }
196
197 try {
198 TDirectoryTreeItem item = new TDirectoryTreeItem(treeViewWidget,
199 f.getCanonicalPath(), false, false);
200
201 item.level = this.level + 1;
202 getChildren().add(item);
203 } catch (IOException e) {
204 continue;
205 }
206 }
207 }
208 Collections.sort(getChildren());
209 }
210
211}