1/* Copyright (c) 2013-2015 Jeffrey Pfau
  2 *
  3 * This Source Code Form is subject to the terms of the Mozilla Public
  4 * License, v. 2.0. If a copy of the MPL was not distributed with this
  5 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
  6#include "file-select.h"
  7
  8#include "util/gui/font.h"
  9#include "util/gui/menu.h"
 10#include "util/vfs.h"
 11
 12#include <stdlib.h>
 13
 14#define ITERATION_SIZE 5
 15#define SCANNING_THRESHOLD_1 50
 16#ifdef _3DS
 17// 3DS is slooooow at opening files
 18#define SCANNING_THRESHOLD_2 10
 19#else
 20#define SCANNING_THRESHOLD_2 50
 21#endif
 22
 23static void _cleanFiles(struct GUIMenuItemList* currentFiles) {
 24	size_t size = GUIMenuItemListSize(currentFiles);
 25	size_t i;
 26	for (i = 1; i < size; ++i) {
 27		free(GUIMenuItemListGetPointer(currentFiles, i)->title);
 28	}
 29	GUIMenuItemListClear(currentFiles);
 30}
 31
 32static void _upDirectory(char* currentPath) {
 33	char* end = strrchr(currentPath, '/');
 34	if (!end) {
 35		return;
 36	}
 37	if (end == currentPath) {
 38		end[1] = '\0';
 39		return;
 40	}
 41	end[0] = '\0';
 42	if (end[1]) {
 43		return;
 44	}
 45	// TODO: What if there was a trailing slash?
 46}
 47
 48static int _strpcmp(const void* a, const void* b) {
 49	return strcasecmp(((const struct GUIMenuItem*) a)->title, ((const struct GUIMenuItem*) b)->title);
 50}
 51
 52static bool _refreshDirectory(struct GUIParams* params, const char* currentPath, struct GUIMenuItemList* currentFiles, bool (*filter)(struct VFile*)) {
 53	_cleanFiles(currentFiles);
 54
 55	struct VDir* dir = VDirOpen(currentPath);
 56	if (!dir) {
 57		return false;
 58	}
 59	*GUIMenuItemListAppend(currentFiles) = (struct GUIMenuItem) { .title = "(Up)" };
 60	size_t i = 0;
 61	size_t items = 0;
 62	struct VDirEntry* de;
 63	while ((de = dir->listNext(dir))) {
 64		++i;
 65		if (!(i % SCANNING_THRESHOLD_1)) {
 66			uint32_t input = 0;
 67			GUIPollInput(params, &input, 0);
 68			if (input & (1 << GUI_INPUT_CANCEL)) {
 69				return false;
 70			}
 71
 72			params->drawStart();
 73			if (params->guiPrepare) {
 74				params->guiPrepare();
 75			}
 76			GUIFontPrintf(params->font, 0, GUIFontHeight(params->font), GUI_TEXT_LEFT, 0xFFFFFFFF, "(scanning for items: %zu)", i);
 77			GUIFontPrintf(params->font, 0, GUIFontHeight(params->font) * 2, GUI_TEXT_LEFT, 0xFFFFFFFF, "%s", currentPath);
 78			if (params->guiFinish) {
 79				params->guiFinish();
 80			}
 81			params->drawEnd();
 82		}
 83		const char* name = de->name(de);
 84		if (name[0] == '.') {
 85			continue;
 86		}
 87		*GUIMenuItemListAppend(currentFiles) = (struct GUIMenuItem) { .title = strdup(name) };
 88		++items;
 89	}
 90	qsort(GUIMenuItemListGetPointer(currentFiles, 1), GUIMenuItemListSize(currentFiles) - 1, sizeof(struct GUIMenuItem), _strpcmp);
 91	i = 0;
 92	size_t item = 0;
 93	while (item < GUIMenuItemListSize(currentFiles)) {
 94		++i;
 95		if (!(i % SCANNING_THRESHOLD_2)) {
 96			uint32_t input = 0;
 97			GUIPollInput(params, &input, 0);
 98			if (input & (1 << GUI_INPUT_CANCEL)) {
 99				return false;
100			}
101
102			params->drawStart();
103			if (params->guiPrepare) {
104				params->guiPrepare();
105			}
106			GUIFontPrintf(params->font, 0, GUIFontHeight(params->font), GUI_TEXT_LEFT, 0xFFFFFFFF, "(scanning item %zu of %zu)", i, items);
107			GUIFontPrintf(params->font, 0, GUIFontHeight(params->font) * 2, GUI_TEXT_LEFT, 0xFFFFFFFF, "%s", currentPath);
108			if (params->guiFinish) {
109				params->guiFinish();
110			}
111			params->drawEnd();
112		}
113		if (!filter) {
114			++item;
115			continue;
116		}
117		struct VDir* vd = dir->openDir(dir, GUIMenuItemListGetPointer(currentFiles, item)->title);
118		if (vd) {
119			vd->close(vd);
120			++item;
121			continue;
122		}
123		struct VFile* vf = dir->openFile(dir, GUIMenuItemListGetPointer(currentFiles, item)->title, O_RDONLY);
124		if (vf) {
125			if (filter(vf)) {
126				++item;
127			} else {
128				free(GUIMenuItemListGetPointer(currentFiles, item)->title);
129				GUIMenuItemListShift(currentFiles, item, 1);
130			}
131			vf->close(vf);
132			continue;
133		}
134		++item;
135	}
136	dir->close(dir);
137
138	return true;
139}
140
141bool GUISelectFile(struct GUIParams* params, char* outPath, size_t outLen, bool (*filter)(struct VFile*)) {
142	struct GUIMenu menu = {
143		.title = "Select file",
144		.subtitle = params->currentPath,
145		.index = params->fileIndex,
146	};
147	GUIMenuItemListInit(&menu.items, 0);
148	_refreshDirectory(params, params->currentPath, &menu.items, filter);
149
150	while (true) {
151		struct GUIMenuItem* item;
152		enum GUIMenuExitReason reason = GUIShowMenu(params, &menu, &item);
153		params->fileIndex = menu.index;
154		if (reason == GUI_MENU_EXIT_CANCEL) {
155			break;
156		}
157		if (reason == GUI_MENU_EXIT_ACCEPT) {
158			if (params->fileIndex == 0) {
159				_upDirectory(params->currentPath);
160				if (!_refreshDirectory(params, params->currentPath, &menu.items, filter)) {
161					break;
162				}
163			} else {
164				size_t len = strlen(params->currentPath);
165				const char* sep = PATH_SEP;
166				if (params->currentPath[len - 1] == *sep) {
167					sep = "";
168				}
169				snprintf(outPath, outLen, "%s%s%s", params->currentPath, sep, item->title);
170
171				struct GUIMenuItemList newFiles;
172				GUIMenuItemListInit(&newFiles, 0);
173				if (!_refreshDirectory(params, outPath, &newFiles, filter)) {
174					_cleanFiles(&newFiles);
175					GUIMenuItemListDeinit(&newFiles);
176					_cleanFiles(&menu.items);
177					GUIMenuItemListDeinit(&menu.items);
178					return true;
179				} else {
180					_cleanFiles(&menu.items);
181					GUIMenuItemListDeinit(&menu.items);
182					menu.items = newFiles;
183					strncpy(params->currentPath, outPath, PATH_MAX);
184				}
185			}
186			params->fileIndex = 0;
187			menu.index = 0;
188		}
189		if (reason == GUI_MENU_EXIT_BACK) {
190			if (strncmp(params->currentPath, params->basePath, PATH_MAX) == 0) {
191				break;
192			}
193			_upDirectory(params->currentPath);
194			if (!_refreshDirectory(params, params->currentPath, &menu.items, filter)) {
195				break;
196			}
197			params->fileIndex = 0;
198			menu.index = 0;
199		}
200	}
201
202	_cleanFiles(&menu.items);
203	GUIMenuItemListDeinit(&menu.items);
204	return false;
205}