mayhem-firmware/firmware/application/apps/ui_fileman.cpp

/*
 * Copyright (C) 2015 Jared Boone, ShareBrained Technology, Inc.
 * Copyright (C) 2017 Furrtek
 *
 * This file is part of PortaPack.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#include "ui_fileman.hpp"
#include "string_format.hpp"
#include "portapack.hpp"
#include "event_m0.hpp"

using namespace portapack;

namespace ui {

void FileManBaseView::load_directory_contents(const std::filesystem::path& dir_path) {
	current_path = dir_path;
	
	text_current.set(dir_path.string().length()? dir_path.string().substr(0, 30 - 6):"(sd root)");

	entry_list.clear();
	
	auto filtering = (bool)extension_filter.size();
	
	// List directories and files, put directories up top
	if (dir_path.string().length())
		entry_list.push_back({ u"..", 0, true });

	for (const auto& entry : std::filesystem::directory_iterator(dir_path, u"*")) {

		// do not display dir / files starting with '.' (hidden / tmp)
		if (entry.path().string().length() && entry.path().filename().string()[0] != '.') {
			if (std::filesystem::is_regular_file(entry.status())) {
				bool matched = true;
				if (filtering) {
					auto entry_extension = entry.path().extension().string();
				
					for (auto &c: entry_extension)
						c = toupper(c);
					
					if (entry_extension != extension_filter)
						matched = false;
				}
				
				if (matched)
					entry_list.push_back({ entry.path(), (uint32_t)entry.size(), false });
			} else if (std::filesystem::is_directory(entry.status())) {
				entry_list.insert(entry_list.begin(), { entry.path(), 0, true });
			}
		}
	}
}

std::filesystem::path FileManBaseView::get_selected_path() {
	auto selected_path_str = current_path.string();
	auto entry_path = entry_list[menu_view.highlighted_index()].entry_path.string();
	
	if (entry_path == "..") {
		selected_path_str = get_parent_dir().string();
	} else {
		if (selected_path_str.back() != '/')
			selected_path_str += '/';
		
		selected_path_str += entry_path;
	}
	
	return selected_path_str;
}

std::filesystem::path FileManBaseView::get_parent_dir() {
	auto current_path_str = current_path.string();
	return current_path.string().substr(0, current_path_str.find_last_of('/'));
}

FileManBaseView::FileManBaseView(
	NavigationView& nav,
	std::string filter
) : nav_ (nav),
	extension_filter { filter }
{
	add_children({
		&labels,
		&text_current,
		&button_exit
	});

	button_exit.on_select = [this, &nav](Button&) {
		nav.pop();
	};

	if (!sdcIsCardInserted(&SDCD1)) {
		empty_root=true;
		text_current.set("NO SD CARD!");
	} else {
		load_directory_contents(current_path);
		if (!entry_list.size())
		{
			empty_root = true;
			text_current.set("EMPTY SD CARD!");
		} else {
			menu_view.on_left = [&nav, this]() {
				load_directory_contents(get_parent_dir());
				refresh_list();
			};
		}
	}
}

void FileManBaseView::focus() {
	if (empty_root) {
		button_exit.focus();
	} else {
		menu_view.focus();
	}
}

void FileManBaseView::refresh_list() {
	if (on_refresh_widgets)
		on_refresh_widgets(false);

	menu_view.clear();
	
	for (size_t n = 0; n < entry_list.size(); n++) {
		auto entry = &entry_list[n];
		auto entry_name = entry->entry_path.filename().string().substr(0, 20);
		
		if (entry->is_directory) {
			
			menu_view.add_item({
				entry_name,
				ui::Color::yellow(),
				&bitmap_icon_dir,
				[this](){
					if (on_select_entry)
						on_select_entry();
				}
			});
			
		} else {
			
			auto file_size = entry->size;
			size_t suffix_index = 0;
			
			while (file_size >= 1024) {
				file_size /= 1024;
				suffix_index++;
			}
			if (suffix_index > 4)
				suffix_index = 4;
			
			std::string size_str = to_string_dec_uint(file_size) + suffix[suffix_index];
			
			auto entry_extension = entry->entry_path.extension().string();
			for (auto &c: entry_extension)
				c = toupper(c);
			
			// Associate extension to icon and color
			size_t c;
			for (c = 0; c < file_types.size() - 1; c++) {
				if (entry_extension == file_types[c].extension)
					break;
			}
			
			menu_view.add_item({
				entry_name + std::string(21 - entry_name.length(), ' ') + size_str,
				file_types[c].color,
				file_types[c].icon,
				[this](){
					if (on_select_entry)
						on_select_entry();
				}
			});
			
		}
	}
	
	menu_view.set_highlighted(0);	// Refresh
}

/*void FileSaveView::on_save_name() {
	text_prompt(nav_, &filename_buffer, 8, [this](std::string * buffer) {
		nav_.pop();
	});
}
FileSaveView::FileSaveView(
	NavigationView& nav
) : FileManBaseView(nav)
{
	name_buffer.clear();
	
	add_children({
		&text_save,
		&button_save_name,
		&live_timestamp
	});
	
	button_save_name.on_select = [this, &nav](Button&) {
		on_save_name();
	};
}*/

void FileLoadView::refresh_widgets(const bool v) {
	(void)v; //avoid unused warning
	set_dirty();
}

FileLoadView::FileLoadView(
	NavigationView& nav,
	std::string filter
) : FileManBaseView(nav, filter)
{
	on_refresh_widgets = [this](bool v) {
		refresh_widgets(v);
	};
	
	add_children({
		&menu_view
	});
	
	// Resize menu view to fill screen
	menu_view.set_parent_rect({ 0, 3 * 8, 240, 29 * 8 });
	
	refresh_list();
	
	on_select_entry = [&nav, this]() {
		if (entry_list[menu_view.highlighted_index()].is_directory) {
			load_directory_contents(get_selected_path());
			refresh_list();
		} else {
			nav_.pop();
			if (on_changed)
				on_changed(current_path.string() + '/' + entry_list[menu_view.highlighted_index()].entry_path.string());
		}
	};
}

void FileManagerView::on_rename(NavigationView& nav) {
	text_prompt(nav, name_buffer, max_filename_length, [this](std::string& buffer) {
		std::string destination_path = current_path.string();
		if (destination_path.back() != '/')
			destination_path += '/';
		destination_path = destination_path + buffer;
		rename_file(get_selected_path(), destination_path);
		load_directory_contents(current_path);
		refresh_list();
	});
}

void FileManagerView::on_refactor(NavigationView& nav) {
	text_prompt(nav, name_buffer, max_filename_length, [this](std::string& buffer) {

		std::string destination_path = current_path.string();
		if (destination_path.back() != '/')//if the path is not ended with '/', add '/'
			destination_path += '/';

		auto selected_path = get_selected_path();
		auto extension = selected_path.extension().string();

		if(extension.empty()){// Is Dir
			destination_path = destination_path + buffer;
			extension_buffer = "";
		}else{//is File
			destination_path = destination_path + buffer + extension_buffer;
		}

		rename_file(get_selected_path(), destination_path);  //rename the selected file

		if (!extension.empty() && selected_path.string().back() != '/' && extension.substr(1) == "C16") { //substr(1) is for ignore the dot
			// Rename its partner ( C16 <-> TXT ) file.
			auto partner_file_path = selected_path.string().substr(0, selected_path.string().size() - 4) + ".TXT";
			destination_path = destination_path.substr(0, destination_path.size() - 4) + ".TXT";
			rename_file(partner_file_path, destination_path);
		} else if (!extension.empty() && selected_path.string().back() != '/' && extension.substr(1) == "TXT") {
			// If the file user choose is a TXT file.
			auto partner_file_path = selected_path.string().substr(0, selected_path.string().size() - 4) + ".C16";
			destination_path = destination_path.substr(0, destination_path.size() - 4) + ".C16";
			rename_file(partner_file_path, destination_path);
		}

		load_directory_contents(current_path);
		refresh_list();

	});

}

void FileManagerView::on_delete() {
	delete_file(get_selected_path());
	load_directory_contents(current_path);
	refresh_list();
}

void FileManagerView::refresh_widgets(const bool v) {
	button_rename.hidden(v);
	button_new_dir.hidden(v);
	button_refactor.hidden(v);
	button_delete.hidden(v);
	set_dirty();
}

FileManagerView::~FileManagerView() {
	// Flush ?
}

FileManagerView::FileManagerView(
	NavigationView& nav
) : FileManBaseView(nav, "")
{
	if (!empty_root) {
		on_refresh_widgets = [this](bool v) {
			refresh_widgets(v);
		};
		
		add_children({
			&menu_view,
			&labels,
			&text_date,
			&button_rename,
			&button_refactor,
			&button_new_dir,
			&button_delete
		});
		
		menu_view.on_highlight = [this]() {
			text_date.set(to_string_FAT_timestamp(file_created_date(get_selected_path())));
		};
		
		refresh_list();
		
		on_select_entry = [this]() {
			if (entry_list[menu_view.highlighted_index()].is_directory) {
				load_directory_contents(get_selected_path());
				refresh_list();
			} else
				button_rename.focus();
		};
		
		button_new_dir.on_select = [this, &nav](Button&) {
			name_buffer.clear();
			
			text_prompt(nav, name_buffer, max_filename_length, [this](std::string& buffer) {
				make_new_directory(current_path.string() + '/' + buffer);
				load_directory_contents(current_path);
				refresh_list();
			});
		};
		
		button_rename.on_select = [this, &nav](Button&) {
			name_buffer = entry_list[menu_view.highlighted_index()].entry_path.filename().string().substr(0, max_filename_length);
			on_rename(nav);
		};

		button_refactor.on_select = [this, &nav](Button&) {
			name_buffer = entry_list[menu_view.highlighted_index()].entry_path.filename().string().substr(0, max_filename_length);
			size_t pos = name_buffer.find_last_of(".");

			if (pos != std::string::npos) {
				extension_buffer = name_buffer.substr(pos);
				name_buffer = name_buffer.substr(0, pos);
			}

			on_refactor(nav);
		};

		button_delete.on_select = [this, &nav](Button&) {
			// Use display_modal ?
			nav.push<ModalMessageView>("Delete", "Delete " + entry_list[menu_view.highlighted_index()].entry_path.filename().string() + "\nAre you sure?", YESNO,
				[this](bool choice) {
					if (choice)
						on_delete();
				}
			);
		};
	} 
}

}