GFilePicker.cpp 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274
  1. #include <AK/FileSystemPath.h>
  2. #include <AK/Function.h>
  3. #include <LibDraw/PNGLoader.h>
  4. #include <LibGUI/GAction.h>
  5. #include <LibGUI/GBoxLayout.h>
  6. #include <LibGUI/GButton.h>
  7. #include <LibGUI/GDirectoryModel.h>
  8. #include <LibGUI/GFilePicker.h>
  9. #include <LibGUI/GInputBox.h>
  10. #include <LibGUI/GLabel.h>
  11. #include <LibGUI/GMessageBox.h>
  12. #include <LibGUI/GSortingProxyModel.h>
  13. #include <LibGUI/GTextBox.h>
  14. #include <LibGUI/GToolBar.h>
  15. Optional<String> GFilePicker::get_open_filepath(const String& window_title)
  16. {
  17. auto picker = GFilePicker::construct(Mode::Open);
  18. if (!window_title.is_null())
  19. picker->set_title(window_title);
  20. if (picker->exec() == GDialog::ExecOK) {
  21. String file_path = picker->selected_file().string();
  22. if (file_path.is_null())
  23. return {};
  24. return file_path;
  25. }
  26. return {};
  27. }
  28. Optional<String> GFilePicker::get_save_filepath(const String& title, const String& extension)
  29. {
  30. auto picker = GFilePicker::construct(Mode::Save, String::format("%s.%s", title.characters(), extension.characters()));
  31. if (picker->exec() == GDialog::ExecOK) {
  32. String file_path = picker->selected_file().string();
  33. if (file_path.is_null())
  34. return {};
  35. return file_path;
  36. }
  37. return {};
  38. }
  39. GFilePicker::GFilePicker(Mode mode, const StringView& file_name, const StringView& path, CObject* parent)
  40. : GDialog(parent)
  41. , m_model(GDirectoryModel::create())
  42. , m_mode(mode)
  43. {
  44. set_title(m_mode == Mode::Open ? "Open File" : "Save File");
  45. set_rect(200, 200, 700, 400);
  46. auto horizontal_container = GWidget::construct();
  47. set_main_widget(horizontal_container);
  48. horizontal_container->set_layout(make<GBoxLayout>(Orientation::Horizontal));
  49. horizontal_container->layout()->set_margins({ 4, 4, 4, 4 });
  50. horizontal_container->set_fill_with_background_color(true);
  51. auto vertical_container = GWidget::construct(horizontal_container.ptr());
  52. vertical_container->set_layout(make<GBoxLayout>(Orientation::Vertical));
  53. vertical_container->layout()->set_spacing(4);
  54. auto upper_container = GWidget::construct(vertical_container.ptr());
  55. upper_container->set_layout(make<GBoxLayout>(Orientation::Horizontal));
  56. upper_container->layout()->set_spacing(4);
  57. upper_container->set_size_policy(SizePolicy::Fill, SizePolicy::Fixed);
  58. upper_container->set_preferred_size(0, 26);
  59. auto toolbar = GToolBar::construct(upper_container);
  60. toolbar->set_size_policy(SizePolicy::Fixed, SizePolicy::Fill);
  61. toolbar->set_preferred_size(85, 0);
  62. toolbar->set_has_frame(false);
  63. auto location_textbox = GTextBox::construct(upper_container);
  64. location_textbox->set_size_policy(SizePolicy::Fill, SizePolicy::Fixed);
  65. location_textbox->set_preferred_size(0, 20);
  66. m_view = GTableView::construct(vertical_container);
  67. m_view->set_model(GSortingProxyModel::create(*m_model));
  68. m_view->set_column_hidden(GDirectoryModel::Column::Owner, true);
  69. m_view->set_column_hidden(GDirectoryModel::Column::Group, true);
  70. m_view->set_column_hidden(GDirectoryModel::Column::Permissions, true);
  71. m_view->set_column_hidden(GDirectoryModel::Column::Inode, true);
  72. m_model->open(path);
  73. location_textbox->on_return_pressed = [&] {
  74. m_model->open(location_textbox->text());
  75. clear_preview();
  76. };
  77. auto open_parent_directory_action = GAction::create("Open parent directory", { Mod_Alt, Key_Up }, GraphicsBitmap::load_from_file("/res/icons/16x16/open-parent-directory.png"), [this](const GAction&) {
  78. m_model->open(String::format("%s/..", m_model->path().characters()));
  79. clear_preview();
  80. });
  81. toolbar->add_action(*open_parent_directory_action);
  82. auto go_home_action = GCommonActions::make_go_home_action([this](auto&) {
  83. m_model->open(get_current_user_home_path());
  84. });
  85. toolbar->add_action(go_home_action);
  86. toolbar->add_separator();
  87. auto mkdir_action = GAction::create("New directory...", GraphicsBitmap::load_from_file("/res/icons/16x16/mkdir.png"), [this](const GAction&) {
  88. auto input_box = GInputBox::construct("Enter name:", "New directory", this);
  89. if (input_box->exec() == GInputBox::ExecOK && !input_box->text_value().is_empty()) {
  90. auto new_dir_path = FileSystemPath(String::format("%s/%s",
  91. m_model->path().characters(),
  92. input_box->text_value().characters()))
  93. .string();
  94. int rc = mkdir(new_dir_path.characters(), 0777);
  95. if (rc < 0) {
  96. GMessageBox::show(String::format("mkdir(\"%s\") failed: %s", new_dir_path.characters(), strerror(errno)), "Error", GMessageBox::Type::Error, GMessageBox::InputType::OK, this);
  97. } else {
  98. m_model->update();
  99. }
  100. }
  101. });
  102. toolbar->add_action(*mkdir_action);
  103. auto lower_container = GWidget::construct(vertical_container.ptr());
  104. lower_container->set_layout(make<GBoxLayout>(Orientation::Vertical));
  105. lower_container->layout()->set_spacing(4);
  106. lower_container->set_size_policy(SizePolicy::Fill, SizePolicy::Fixed);
  107. lower_container->set_preferred_size(0, 60);
  108. auto filename_container = GWidget::construct(lower_container.ptr());
  109. filename_container->set_size_policy(SizePolicy::Fill, SizePolicy::Fixed);
  110. filename_container->set_preferred_size(0, 20);
  111. filename_container->set_layout(make<GBoxLayout>(Orientation::Horizontal));
  112. auto filename_label = GLabel::construct("File name:", filename_container);
  113. filename_label->set_text_alignment(TextAlignment::CenterLeft);
  114. filename_label->set_size_policy(SizePolicy::Fixed, SizePolicy::Fill);
  115. filename_label->set_preferred_size(60, 0);
  116. m_filename_textbox = GTextBox::construct(filename_container.ptr());
  117. if (m_mode == Mode::Save) {
  118. m_filename_textbox->set_text(file_name);
  119. m_filename_textbox->set_focus(true);
  120. m_filename_textbox->select_all();
  121. }
  122. m_filename_textbox->on_return_pressed = [&] {
  123. on_file_return();
  124. };
  125. m_view->on_selection = [this](auto& index) {
  126. auto& filter_model = (GSortingProxyModel&)*m_view->model();
  127. auto local_index = filter_model.map_to_target(index);
  128. const GDirectoryModel::Entry& entry = m_model->entry(local_index.row());
  129. FileSystemPath path(String::format("%s/%s", m_model->path().characters(), entry.name.characters()));
  130. clear_preview();
  131. if (!entry.is_directory())
  132. m_filename_textbox->set_text(entry.name);
  133. set_preview(path);
  134. };
  135. auto button_container = GWidget::construct(lower_container.ptr());
  136. button_container->set_size_policy(SizePolicy::Fill, SizePolicy::Fixed);
  137. button_container->set_preferred_size(0, 20);
  138. button_container->set_layout(make<GBoxLayout>(Orientation::Horizontal));
  139. button_container->layout()->set_spacing(4);
  140. button_container->layout()->add_spacer();
  141. auto cancel_button = GButton::construct(button_container);
  142. cancel_button->set_size_policy(SizePolicy::Fixed, SizePolicy::Fill);
  143. cancel_button->set_preferred_size(80, 0);
  144. cancel_button->set_text("Cancel");
  145. cancel_button->on_click = [this](auto&) {
  146. done(ExecCancel);
  147. };
  148. auto ok_button = GButton::construct(button_container);
  149. ok_button->set_size_policy(SizePolicy::Fixed, SizePolicy::Fill);
  150. ok_button->set_preferred_size(80, 0);
  151. ok_button->set_text(ok_button_name(m_mode));
  152. ok_button->on_click = [this](auto&) {
  153. on_file_return();
  154. };
  155. m_view->on_activation = [this](auto& index) {
  156. auto& filter_model = (GSortingProxyModel&)*m_view->model();
  157. auto local_index = filter_model.map_to_target(index);
  158. const GDirectoryModel::Entry& entry = m_model->entry(local_index.row());
  159. FileSystemPath path(String::format("%s/%s", m_model->path().characters(), entry.name.characters()));
  160. if (entry.is_directory()) {
  161. m_model->open(path.string());
  162. // NOTE: 'entry' is invalid from here on
  163. } else {
  164. on_file_return();
  165. }
  166. };
  167. auto preview_container = GFrame::construct(horizontal_container);
  168. preview_container->set_size_policy(SizePolicy::Fixed, SizePolicy::Fill);
  169. preview_container->set_preferred_size(180, 0);
  170. preview_container->set_frame_shape(FrameShape::Container);
  171. preview_container->set_frame_shadow(FrameShadow::Sunken);
  172. preview_container->set_frame_thickness(2);
  173. preview_container->set_layout(make<GBoxLayout>(Orientation::Vertical));
  174. preview_container->layout()->set_margins({ 8, 8, 8, 8 });
  175. m_preview_image_label = GLabel::construct(preview_container);
  176. m_preview_image_label->set_should_stretch_icon(true);
  177. m_preview_image_label->set_size_policy(SizePolicy::Fixed, SizePolicy::Fixed);
  178. m_preview_image_label->set_preferred_size(160, 160);
  179. m_preview_name_label = GLabel::construct(preview_container);
  180. m_preview_name_label->set_font(Font::default_bold_font());
  181. m_preview_name_label->set_size_policy(SizePolicy::Fill, SizePolicy::Fixed);
  182. m_preview_name_label->set_preferred_size(0, m_preview_name_label->font().glyph_height());
  183. m_preview_geometry_label = GLabel::construct(preview_container);
  184. m_preview_geometry_label->set_size_policy(SizePolicy::Fill, SizePolicy::Fixed);
  185. m_preview_geometry_label->set_preferred_size(0, m_preview_name_label->font().glyph_height());
  186. }
  187. GFilePicker::~GFilePicker()
  188. {
  189. }
  190. void GFilePicker::set_preview(const FileSystemPath& path)
  191. {
  192. if (path.has_extension(".png")) {
  193. auto bitmap = load_png(path.string());
  194. if (!bitmap) {
  195. clear_preview();
  196. return;
  197. }
  198. bool should_stretch = bitmap->width() > m_preview_image_label->width() || bitmap->height() > m_preview_image_label->height();
  199. m_preview_name_label->set_text(path.basename());
  200. m_preview_geometry_label->set_text(bitmap->size().to_string());
  201. m_preview_image_label->set_should_stretch_icon(should_stretch);
  202. m_preview_image_label->set_icon(move(bitmap));
  203. }
  204. }
  205. void GFilePicker::clear_preview()
  206. {
  207. m_preview_image_label->set_icon(nullptr);
  208. m_preview_name_label->set_text(String::empty());
  209. m_preview_geometry_label->set_text(String::empty());
  210. }
  211. void GFilePicker::on_file_return()
  212. {
  213. FileSystemPath path(String::format("%s/%s", m_model->path().characters(), m_filename_textbox->text().characters()));
  214. if (GFilePicker::file_exists(path.string()) && m_mode == Mode::Save) {
  215. auto result = GMessageBox::show("File already exists, overwrite?", "Existing File", GMessageBox::Type::Warning, GMessageBox::InputType::OKCancel);
  216. if (result == GMessageBox::ExecCancel)
  217. return;
  218. }
  219. m_selected_file = path;
  220. done(ExecOK);
  221. }
  222. bool GFilePicker::file_exists(const StringView& path)
  223. {
  224. struct stat st;
  225. int rc = stat(String(path).characters(), &st);
  226. if (rc < 0) {
  227. if (errno == ENOENT)
  228. return false;
  229. }
  230. if (rc == 0) {
  231. return true;
  232. }
  233. return false;
  234. }