find_in_files.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918
  1. /*************************************************************************/
  2. /* find_in_files.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2019 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2019 Godot Engine contributors (cf. AUTHORS.md) */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /*************************************************************************/
  30. #include "find_in_files.h"
  31. #include "core/os/dir_access.h"
  32. #include "core/os/os.h"
  33. #include "editor_node.h"
  34. #include "editor_scale.h"
  35. #include "scene/gui/box_container.h"
  36. #include "scene/gui/button.h"
  37. #include "scene/gui/check_box.h"
  38. #include "scene/gui/file_dialog.h"
  39. #include "scene/gui/grid_container.h"
  40. #include "scene/gui/label.h"
  41. #include "scene/gui/line_edit.h"
  42. #include "scene/gui/progress_bar.h"
  43. #include "scene/gui/tree.h"
  44. const char *FindInFiles::SIGNAL_RESULT_FOUND = "result_found";
  45. const char *FindInFiles::SIGNAL_FINISHED = "finished";
  46. // TODO Would be nice in Vector and PoolVectors
  47. template <typename T>
  48. inline void pop_back(T &container) {
  49. container.resize(container.size() - 1);
  50. }
  51. // TODO Copied from TextEdit private, would be nice to extract it in a single place
  52. static bool is_text_char(CharType c) {
  53. return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || (c >= '0' && c <= '9') || c == '_';
  54. }
  55. static bool find_next(const String &line, String pattern, int from, bool match_case, bool whole_words, int &out_begin, int &out_end) {
  56. int end = from;
  57. while (true) {
  58. int begin = match_case ? line.find(pattern, end) : line.findn(pattern, end);
  59. if (begin == -1)
  60. return false;
  61. end = begin + pattern.length();
  62. out_begin = begin;
  63. out_end = end;
  64. if (whole_words) {
  65. if (begin > 0 && is_text_char(line[begin - 1])) {
  66. continue;
  67. }
  68. if (end < line.size() && is_text_char(line[end])) {
  69. continue;
  70. }
  71. }
  72. return true;
  73. }
  74. }
  75. //--------------------------------------------------------------------------------
  76. FindInFiles::FindInFiles() {
  77. _searching = false;
  78. _whole_words = true;
  79. _match_case = true;
  80. }
  81. void FindInFiles::set_search_text(String p_pattern) {
  82. _pattern = p_pattern;
  83. }
  84. void FindInFiles::set_whole_words(bool p_whole_word) {
  85. _whole_words = p_whole_word;
  86. }
  87. void FindInFiles::set_match_case(bool p_match_case) {
  88. _match_case = p_match_case;
  89. }
  90. void FindInFiles::set_folder(String folder) {
  91. _root_dir = folder;
  92. }
  93. void FindInFiles::set_filter(const Set<String> &exts) {
  94. _extension_filter = exts;
  95. }
  96. void FindInFiles::_notification(int p_notification) {
  97. if (p_notification == NOTIFICATION_PROCESS) {
  98. _process();
  99. }
  100. }
  101. void FindInFiles::start() {
  102. if (_pattern == "") {
  103. print_verbose("Nothing to search, pattern is empty");
  104. emit_signal(SIGNAL_FINISHED);
  105. return;
  106. }
  107. if (_extension_filter.size() == 0) {
  108. print_verbose("Nothing to search, filter matches no files");
  109. emit_signal(SIGNAL_FINISHED);
  110. return;
  111. }
  112. // Init search
  113. _current_dir = "";
  114. PoolStringArray init_folder;
  115. init_folder.append(_root_dir);
  116. _folders_stack.clear();
  117. _folders_stack.push_back(init_folder);
  118. _initial_files_count = 0;
  119. _searching = true;
  120. set_process(true);
  121. }
  122. void FindInFiles::stop() {
  123. _searching = false;
  124. _current_dir = "";
  125. set_process(false);
  126. }
  127. void FindInFiles::_process() {
  128. // This part can be moved to a thread if needed
  129. OS &os = *OS::get_singleton();
  130. float time_before = os.get_ticks_msec();
  131. while (is_processing()) {
  132. _iterate();
  133. float elapsed = (os.get_ticks_msec() - time_before);
  134. if (elapsed > 1000.0 / 120.0)
  135. break;
  136. }
  137. }
  138. void FindInFiles::_iterate() {
  139. if (_folders_stack.size() != 0) {
  140. // Scan folders first so we can build a list of files and have progress info later
  141. PoolStringArray &folders_to_scan = _folders_stack.write[_folders_stack.size() - 1];
  142. if (folders_to_scan.size() != 0) {
  143. // Scan one folder below
  144. String folder_name = folders_to_scan[folders_to_scan.size() - 1];
  145. pop_back(folders_to_scan);
  146. _current_dir = _current_dir.plus_file(folder_name);
  147. PoolStringArray sub_dirs;
  148. _scan_dir("res://" + _current_dir, sub_dirs);
  149. _folders_stack.push_back(sub_dirs);
  150. } else {
  151. // Go back one level
  152. pop_back(_folders_stack);
  153. _current_dir = _current_dir.get_base_dir();
  154. if (_folders_stack.size() == 0) {
  155. // All folders scanned
  156. _initial_files_count = _files_to_scan.size();
  157. }
  158. }
  159. } else if (_files_to_scan.size() != 0) {
  160. // Then scan files
  161. String fpath = _files_to_scan[_files_to_scan.size() - 1];
  162. pop_back(_files_to_scan);
  163. _scan_file(fpath);
  164. } else {
  165. print_verbose("Search complete");
  166. set_process(false);
  167. _current_dir = "";
  168. _searching = false;
  169. emit_signal(SIGNAL_FINISHED);
  170. }
  171. }
  172. float FindInFiles::get_progress() const {
  173. if (_initial_files_count != 0) {
  174. return static_cast<float>(_initial_files_count - _files_to_scan.size()) / static_cast<float>(_initial_files_count);
  175. }
  176. return 0;
  177. }
  178. void FindInFiles::_scan_dir(String path, PoolStringArray &out_folders) {
  179. DirAccess *dir = DirAccess::open(path);
  180. if (dir == NULL) {
  181. print_verbose("Cannot open directory! " + path);
  182. return;
  183. }
  184. dir->list_dir_begin();
  185. for (int i = 0; i < 1000; ++i) {
  186. String file = dir->get_next();
  187. if (file == "")
  188. break;
  189. // Ignore special dirs and hidden dirs (such as .git and .import)
  190. if (file == "." || file == ".." || file.begins_with("."))
  191. continue;
  192. if (dir->current_is_dir())
  193. out_folders.append(file);
  194. else {
  195. String file_ext = file.get_extension();
  196. if (_extension_filter.has(file_ext)) {
  197. _files_to_scan.push_back(path.plus_file(file));
  198. }
  199. }
  200. }
  201. }
  202. void FindInFiles::_scan_file(String fpath) {
  203. FileAccess *f = FileAccess::open(fpath, FileAccess::READ);
  204. if (f == NULL) {
  205. print_verbose(String("Cannot open file ") + fpath);
  206. return;
  207. }
  208. int line_number = 0;
  209. while (!f->eof_reached()) {
  210. // line number starts at 1
  211. ++line_number;
  212. int begin = 0;
  213. int end = 0;
  214. String line = f->get_line();
  215. while (find_next(line, _pattern, end, _match_case, _whole_words, begin, end)) {
  216. emit_signal(SIGNAL_RESULT_FOUND, fpath, line_number, begin, end, line);
  217. }
  218. }
  219. f->close();
  220. }
  221. void FindInFiles::_bind_methods() {
  222. ADD_SIGNAL(MethodInfo(SIGNAL_RESULT_FOUND,
  223. PropertyInfo(Variant::STRING, "path"),
  224. PropertyInfo(Variant::INT, "line_number"),
  225. PropertyInfo(Variant::INT, "begin"),
  226. PropertyInfo(Variant::INT, "end"),
  227. PropertyInfo(Variant::STRING, "text")));
  228. ADD_SIGNAL(MethodInfo(SIGNAL_FINISHED));
  229. }
  230. //-----------------------------------------------------------------------------
  231. const char *FindInFilesDialog::SIGNAL_FIND_REQUESTED = "find_requested";
  232. const char *FindInFilesDialog::SIGNAL_REPLACE_REQUESTED = "replace_requested";
  233. FindInFilesDialog::FindInFilesDialog() {
  234. set_custom_minimum_size(Size2(500 * EDSCALE, 0));
  235. set_title(TTR("Find in Files"));
  236. VBoxContainer *vbc = memnew(VBoxContainer);
  237. vbc->set_anchor_and_margin(MARGIN_LEFT, ANCHOR_BEGIN, 8 * EDSCALE);
  238. vbc->set_anchor_and_margin(MARGIN_TOP, ANCHOR_BEGIN, 8 * EDSCALE);
  239. vbc->set_anchor_and_margin(MARGIN_RIGHT, ANCHOR_END, -8 * EDSCALE);
  240. vbc->set_anchor_and_margin(MARGIN_BOTTOM, ANCHOR_END, -8 * EDSCALE);
  241. add_child(vbc);
  242. GridContainer *gc = memnew(GridContainer);
  243. gc->set_columns(2);
  244. vbc->add_child(gc);
  245. Label *find_label = memnew(Label);
  246. find_label->set_text(TTR("Find:"));
  247. gc->add_child(find_label);
  248. _search_text_line_edit = memnew(LineEdit);
  249. _search_text_line_edit->set_h_size_flags(SIZE_EXPAND_FILL);
  250. _search_text_line_edit->connect("text_changed", this, "_on_search_text_modified");
  251. _search_text_line_edit->connect("text_entered", this, "_on_search_text_entered");
  252. gc->add_child(_search_text_line_edit);
  253. gc->add_child(memnew(Control)); // Space to mantain the grid aligned.
  254. {
  255. HBoxContainer *hbc = memnew(HBoxContainer);
  256. _whole_words_checkbox = memnew(CheckBox);
  257. _whole_words_checkbox->set_text(TTR("Whole Words"));
  258. hbc->add_child(_whole_words_checkbox);
  259. _match_case_checkbox = memnew(CheckBox);
  260. _match_case_checkbox->set_text(TTR("Match Case"));
  261. hbc->add_child(_match_case_checkbox);
  262. gc->add_child(hbc);
  263. }
  264. Label *folder_label = memnew(Label);
  265. folder_label->set_text(TTR("Folder:"));
  266. gc->add_child(folder_label);
  267. {
  268. HBoxContainer *hbc = memnew(HBoxContainer);
  269. Label *prefix_label = memnew(Label);
  270. prefix_label->set_text("res://");
  271. hbc->add_child(prefix_label);
  272. _folder_line_edit = memnew(LineEdit);
  273. _folder_line_edit->set_h_size_flags(SIZE_EXPAND_FILL);
  274. hbc->add_child(_folder_line_edit);
  275. Button *folder_button = memnew(Button);
  276. folder_button->set_text("...");
  277. folder_button->connect("pressed", this, "_on_folder_button_pressed");
  278. hbc->add_child(folder_button);
  279. _folder_dialog = memnew(FileDialog);
  280. _folder_dialog->set_mode(FileDialog::MODE_OPEN_DIR);
  281. _folder_dialog->connect("dir_selected", this, "_on_folder_selected");
  282. add_child(_folder_dialog);
  283. gc->add_child(hbc);
  284. }
  285. Label *filter_label = memnew(Label);
  286. filter_label->set_text(TTR("Filters:"));
  287. gc->add_child(filter_label);
  288. {
  289. HBoxContainer *hbc = memnew(HBoxContainer);
  290. // TODO: Unhardcode this.
  291. Vector<String> exts;
  292. exts.push_back("gd");
  293. if (Engine::get_singleton()->has_singleton("GodotSharp"))
  294. exts.push_back("cs");
  295. exts.push_back("shader");
  296. for (int i = 0; i < exts.size(); ++i) {
  297. CheckBox *cb = memnew(CheckBox);
  298. cb->set_text(exts[i]);
  299. cb->set_pressed(true);
  300. hbc->add_child(cb);
  301. _filters.push_back(cb);
  302. }
  303. gc->add_child(hbc);
  304. }
  305. _find_button = add_button(TTR("Find..."), false, "find");
  306. _find_button->set_disabled(true);
  307. _replace_button = add_button(TTR("Replace..."), false, "replace");
  308. _replace_button->set_disabled(true);
  309. Button *cancel_button = get_ok();
  310. cancel_button->set_text(TTR("Cancel"));
  311. }
  312. void FindInFilesDialog::set_search_text(String text) {
  313. _search_text_line_edit->set_text(text);
  314. _on_search_text_modified(text);
  315. }
  316. String FindInFilesDialog::get_search_text() const {
  317. String text = _search_text_line_edit->get_text();
  318. return text.strip_edges();
  319. }
  320. bool FindInFilesDialog::is_match_case() const {
  321. return _match_case_checkbox->is_pressed();
  322. }
  323. bool FindInFilesDialog::is_whole_words() const {
  324. return _whole_words_checkbox->is_pressed();
  325. }
  326. String FindInFilesDialog::get_folder() const {
  327. String text = _folder_line_edit->get_text();
  328. return text.strip_edges();
  329. }
  330. Set<String> FindInFilesDialog::get_filter() const {
  331. Set<String> filters;
  332. for (int i = 0; i < _filters.size(); ++i) {
  333. CheckBox *cb = _filters[i];
  334. if (cb->is_pressed()) {
  335. filters.insert(_filters[i]->get_text());
  336. }
  337. }
  338. return filters;
  339. }
  340. void FindInFilesDialog::_notification(int p_what) {
  341. if (p_what == NOTIFICATION_VISIBILITY_CHANGED) {
  342. if (is_visible()) {
  343. // Doesn't work more than once if not deferred...
  344. _search_text_line_edit->call_deferred("grab_focus");
  345. _search_text_line_edit->select_all();
  346. }
  347. }
  348. }
  349. void FindInFilesDialog::_on_folder_button_pressed() {
  350. _folder_dialog->popup_centered_ratio();
  351. }
  352. void FindInFilesDialog::custom_action(const String &p_action) {
  353. if (p_action == "find") {
  354. emit_signal(SIGNAL_FIND_REQUESTED);
  355. hide();
  356. } else if (p_action == "replace") {
  357. emit_signal(SIGNAL_REPLACE_REQUESTED);
  358. hide();
  359. }
  360. }
  361. void FindInFilesDialog::_on_search_text_modified(String text) {
  362. ERR_FAIL_COND(!_find_button);
  363. ERR_FAIL_COND(!_replace_button);
  364. _find_button->set_disabled(get_search_text().empty());
  365. _replace_button->set_disabled(get_search_text().empty());
  366. }
  367. void FindInFilesDialog::_on_search_text_entered(String text) {
  368. // This allows to trigger a global search without leaving the keyboard
  369. if (!_find_button->is_disabled())
  370. custom_action("find");
  371. }
  372. void FindInFilesDialog::_on_folder_selected(String path) {
  373. int i = path.find("://");
  374. if (i != -1)
  375. path = path.right(i + 3);
  376. _folder_line_edit->set_text(path);
  377. }
  378. void FindInFilesDialog::_bind_methods() {
  379. ClassDB::bind_method("_on_folder_button_pressed", &FindInFilesDialog::_on_folder_button_pressed);
  380. ClassDB::bind_method("_on_folder_selected", &FindInFilesDialog::_on_folder_selected);
  381. ClassDB::bind_method("_on_search_text_modified", &FindInFilesDialog::_on_search_text_modified);
  382. ClassDB::bind_method("_on_search_text_entered", &FindInFilesDialog::_on_search_text_entered);
  383. ADD_SIGNAL(MethodInfo(SIGNAL_FIND_REQUESTED));
  384. ADD_SIGNAL(MethodInfo(SIGNAL_REPLACE_REQUESTED));
  385. }
  386. //-----------------------------------------------------------------------------
  387. const char *FindInFilesPanel::SIGNAL_RESULT_SELECTED = "result_selected";
  388. const char *FindInFilesPanel::SIGNAL_FILES_MODIFIED = "files_modified";
  389. FindInFilesPanel::FindInFilesPanel() {
  390. _finder = memnew(FindInFiles);
  391. _finder->connect(FindInFiles::SIGNAL_RESULT_FOUND, this, "_on_result_found");
  392. _finder->connect(FindInFiles::SIGNAL_FINISHED, this, "_on_finished");
  393. add_child(_finder);
  394. VBoxContainer *vbc = memnew(VBoxContainer);
  395. vbc->set_anchor_and_margin(MARGIN_LEFT, ANCHOR_BEGIN, 0);
  396. vbc->set_anchor_and_margin(MARGIN_TOP, ANCHOR_BEGIN, 0);
  397. vbc->set_anchor_and_margin(MARGIN_RIGHT, ANCHOR_END, 0);
  398. vbc->set_anchor_and_margin(MARGIN_BOTTOM, ANCHOR_END, 0);
  399. add_child(vbc);
  400. {
  401. HBoxContainer *hbc = memnew(HBoxContainer);
  402. Label *find_label = memnew(Label);
  403. find_label->set_text(TTR("Find: "));
  404. hbc->add_child(find_label);
  405. _search_text_label = memnew(Label);
  406. _search_text_label->add_font_override("font", EditorNode::get_singleton()->get_gui_base()->get_font("source", "EditorFonts"));
  407. hbc->add_child(_search_text_label);
  408. _progress_bar = memnew(ProgressBar);
  409. _progress_bar->set_h_size_flags(SIZE_EXPAND_FILL);
  410. hbc->add_child(_progress_bar);
  411. set_progress_visible(false);
  412. _status_label = memnew(Label);
  413. hbc->add_child(_status_label);
  414. _cancel_button = memnew(Button);
  415. _cancel_button->set_text(TTR("Cancel"));
  416. _cancel_button->connect("pressed", this, "_on_cancel_button_clicked");
  417. _cancel_button->hide();
  418. hbc->add_child(_cancel_button);
  419. vbc->add_child(hbc);
  420. }
  421. _results_display = memnew(Tree);
  422. _results_display->add_font_override("font", EditorNode::get_singleton()->get_gui_base()->get_font("source", "EditorFonts"));
  423. _results_display->set_v_size_flags(SIZE_EXPAND_FILL);
  424. _results_display->connect("item_selected", this, "_on_result_selected");
  425. _results_display->connect("item_edited", this, "_on_item_edited");
  426. _results_display->set_hide_root(true);
  427. _results_display->set_select_mode(Tree::SELECT_ROW);
  428. _results_display->create_item(); // Root
  429. vbc->add_child(_results_display);
  430. _with_replace = false;
  431. {
  432. _replace_container = memnew(HBoxContainer);
  433. Label *replace_label = memnew(Label);
  434. replace_label->set_text(TTR("Replace: "));
  435. _replace_container->add_child(replace_label);
  436. _replace_line_edit = memnew(LineEdit);
  437. _replace_line_edit->set_h_size_flags(SIZE_EXPAND_FILL);
  438. _replace_line_edit->connect("text_changed", this, "_on_replace_text_changed");
  439. _replace_container->add_child(_replace_line_edit);
  440. _replace_all_button = memnew(Button);
  441. _replace_all_button->set_text(TTR("Replace all (no undo)"));
  442. _replace_all_button->connect("pressed", this, "_on_replace_all_clicked");
  443. _replace_container->add_child(_replace_all_button);
  444. _replace_container->hide();
  445. vbc->add_child(_replace_container);
  446. }
  447. }
  448. void FindInFilesPanel::set_with_replace(bool with_replace) {
  449. _with_replace = with_replace;
  450. _replace_container->set_visible(with_replace);
  451. if (with_replace) {
  452. // Results show checkboxes on their left so they can be opted out
  453. _results_display->set_columns(2);
  454. _results_display->set_column_expand(0, false);
  455. _results_display->set_column_min_width(0, 48 * EDSCALE);
  456. } else {
  457. // Results are single-cell items
  458. _results_display->set_column_expand(0, true);
  459. _results_display->set_columns(1);
  460. }
  461. }
  462. void FindInFilesPanel::clear() {
  463. _file_items.clear();
  464. _result_items.clear();
  465. _results_display->clear();
  466. _results_display->create_item(); // Root
  467. }
  468. void FindInFilesPanel::start_search() {
  469. clear();
  470. _status_label->set_text(TTR("Searching..."));
  471. _search_text_label->set_text(_finder->get_search_text());
  472. set_process(true);
  473. set_progress_visible(true);
  474. _finder->start();
  475. update_replace_buttons();
  476. _cancel_button->show();
  477. }
  478. void FindInFilesPanel::stop_search() {
  479. _finder->stop();
  480. _status_label->set_text("");
  481. update_replace_buttons();
  482. set_progress_visible(false);
  483. _cancel_button->hide();
  484. }
  485. void FindInFilesPanel::_notification(int p_what) {
  486. if (p_what == NOTIFICATION_PROCESS) {
  487. _progress_bar->set_as_ratio(_finder->get_progress());
  488. }
  489. }
  490. void FindInFilesPanel::_on_result_found(String fpath, int line_number, int begin, int end, String text) {
  491. TreeItem *file_item;
  492. Map<String, TreeItem *>::Element *E = _file_items.find(fpath);
  493. if (E == NULL) {
  494. file_item = _results_display->create_item();
  495. file_item->set_text(0, fpath);
  496. file_item->set_metadata(0, fpath);
  497. // The width of this column is restrained to checkboxes, but that doesn't make sense for the parent items,
  498. // so we override their width so they can expand to full width
  499. file_item->set_expand_right(0, true);
  500. _file_items[fpath] = file_item;
  501. } else {
  502. file_item = E->value();
  503. }
  504. int text_index = _with_replace ? 1 : 0;
  505. TreeItem *item = _results_display->create_item(file_item);
  506. // Do this first because it resets properties of the cell...
  507. item->set_cell_mode(text_index, TreeItem::CELL_MODE_CUSTOM);
  508. String item_text = vformat("%3s: %s", line_number, text.replace("\t", " "));
  509. item->set_text(text_index, item_text);
  510. item->set_custom_draw(text_index, this, "_draw_result_text");
  511. Ref<Font> font = _results_display->get_font("font");
  512. float raw_text_width = font->get_string_size(text).x;
  513. float item_text_width = font->get_string_size(item_text).x;
  514. Result r;
  515. r.line_number = line_number;
  516. r.begin = begin;
  517. r.end = end;
  518. r.draw_begin = (item_text_width - raw_text_width) + font->get_string_size(text.left(r.begin)).x;
  519. r.draw_width = font->get_string_size(text.substr(r.begin, r.end - r.begin)).x;
  520. _result_items[item] = r;
  521. if (_with_replace) {
  522. item->set_cell_mode(0, TreeItem::CELL_MODE_CHECK);
  523. item->set_checked(0, true);
  524. item->set_editable(0, true);
  525. }
  526. }
  527. void FindInFilesPanel::draw_result_text(Object *item_obj, Rect2 rect) {
  528. TreeItem *item = Object::cast_to<TreeItem>(item_obj);
  529. if (!item)
  530. return;
  531. Map<TreeItem *, Result>::Element *E = _result_items.find(item);
  532. if (!E)
  533. return;
  534. Result r = E->value();
  535. Rect2 match_rect = rect;
  536. match_rect.position.x += r.draw_begin;
  537. match_rect.size.x = r.draw_width;
  538. match_rect.position.y += 1 * EDSCALE;
  539. match_rect.size.y -= 2 * EDSCALE;
  540. _results_display->draw_rect(match_rect, Color(0, 0, 0, 0.5));
  541. // Text is drawn by Tree already
  542. }
  543. void FindInFilesPanel::_on_item_edited() {
  544. TreeItem *item = _results_display->get_selected();
  545. if (item->is_checked(0)) {
  546. item->set_custom_color(1, _results_display->get_color("font_color"));
  547. } else {
  548. // Grey out
  549. Color color = _results_display->get_color("font_color");
  550. color.a /= 2.0;
  551. item->set_custom_color(1, color);
  552. }
  553. }
  554. void FindInFilesPanel::_on_finished() {
  555. _status_label->set_text(TTR("Search complete"));
  556. update_replace_buttons();
  557. set_progress_visible(false);
  558. _cancel_button->hide();
  559. }
  560. void FindInFilesPanel::_on_cancel_button_clicked() {
  561. stop_search();
  562. }
  563. void FindInFilesPanel::_on_result_selected() {
  564. TreeItem *item = _results_display->get_selected();
  565. Map<TreeItem *, Result>::Element *E = _result_items.find(item);
  566. if (E == NULL)
  567. return;
  568. Result r = E->value();
  569. TreeItem *file_item = item->get_parent();
  570. String fpath = file_item->get_metadata(0);
  571. emit_signal(SIGNAL_RESULT_SELECTED, fpath, r.line_number, r.begin, r.end);
  572. }
  573. void FindInFilesPanel::_on_replace_text_changed(String text) {
  574. update_replace_buttons();
  575. }
  576. void FindInFilesPanel::_on_replace_all_clicked() {
  577. String replace_text = get_replace_text();
  578. ERR_FAIL_COND(replace_text.empty());
  579. PoolStringArray modified_files;
  580. for (Map<String, TreeItem *>::Element *E = _file_items.front(); E; E = E->next()) {
  581. TreeItem *file_item = E->value();
  582. String fpath = file_item->get_metadata(0);
  583. Vector<Result> locations;
  584. for (TreeItem *item = file_item->get_children(); item; item = item->get_next()) {
  585. if (!item->is_checked(0))
  586. continue;
  587. Map<TreeItem *, Result>::Element *E = _result_items.find(item);
  588. ERR_FAIL_COND(E == NULL);
  589. locations.push_back(E->value());
  590. }
  591. if (locations.size() != 0) {
  592. // Results are sorted by file, so we can batch replaces
  593. apply_replaces_in_file(fpath, locations, replace_text);
  594. modified_files.append(fpath);
  595. }
  596. }
  597. // Hide replace bar so we can't trigger the action twice without doing a new search
  598. _replace_container->hide();
  599. emit_signal(SIGNAL_FILES_MODIFIED, modified_files);
  600. }
  601. // Same as get_line, but preserves line ending characters
  602. class ConservativeGetLine {
  603. public:
  604. String get_line(FileAccess *f) {
  605. _line_buffer.clear();
  606. CharType c = f->get_8();
  607. while (!f->eof_reached()) {
  608. if (c == '\n') {
  609. _line_buffer.push_back(c);
  610. _line_buffer.push_back(0);
  611. return String::utf8(_line_buffer.ptr());
  612. } else if (c == '\0') {
  613. _line_buffer.push_back(c);
  614. return String::utf8(_line_buffer.ptr());
  615. } else if (c != '\r') {
  616. _line_buffer.push_back(c);
  617. }
  618. c = f->get_8();
  619. }
  620. _line_buffer.push_back(0);
  621. return String::utf8(_line_buffer.ptr());
  622. }
  623. private:
  624. Vector<char> _line_buffer;
  625. };
  626. void FindInFilesPanel::apply_replaces_in_file(String fpath, const Vector<Result> &locations, String new_text) {
  627. // If the file is already open, I assume the editor will reload it.
  628. // If there are unsaved changes, the user will be asked on focus,
  629. // however that means either losing changes or losing replaces.
  630. FileAccess *f = FileAccess::open(fpath, FileAccess::READ);
  631. ERR_FAIL_COND(f == NULL);
  632. String buffer;
  633. int current_line = 1;
  634. ConservativeGetLine conservative;
  635. String line = conservative.get_line(f);
  636. String search_text = _finder->get_search_text();
  637. int offset = 0;
  638. for (int i = 0; i < locations.size(); ++i) {
  639. int repl_line_number = locations[i].line_number;
  640. while (current_line < repl_line_number) {
  641. buffer += line;
  642. line = conservative.get_line(f);
  643. ++current_line;
  644. offset = 0;
  645. }
  646. int repl_begin = locations[i].begin + offset;
  647. int repl_end = locations[i].end + offset;
  648. int _;
  649. if (!find_next(line, search_text, repl_begin, _finder->is_match_case(), _finder->is_whole_words(), _, _)) {
  650. // Make sure the replace is still valid in case the file was tampered with.
  651. print_verbose(String("Occurrence no longer matches, replace will be ignored in {0}: line {1}, col {2}").format(varray(fpath, repl_line_number, repl_begin)));
  652. continue;
  653. }
  654. line = line.left(repl_begin) + new_text + line.right(repl_end);
  655. // keep an offset in case there are successive replaces in the same line
  656. offset += new_text.length() - (repl_end - repl_begin);
  657. }
  658. buffer += line;
  659. while (!f->eof_reached()) {
  660. buffer += conservative.get_line(f);
  661. }
  662. // Now the modified contents are in the buffer, rewrite the file with our changes
  663. Error err = f->reopen(fpath, FileAccess::WRITE);
  664. ERR_FAIL_COND(err != OK);
  665. f->store_string(buffer);
  666. f->close();
  667. }
  668. String FindInFilesPanel::get_replace_text() {
  669. return _replace_line_edit->get_text().strip_edges();
  670. }
  671. void FindInFilesPanel::update_replace_buttons() {
  672. String text = get_replace_text();
  673. bool disabled = text.empty() || _finder->is_searching();
  674. _replace_all_button->set_disabled(disabled);
  675. }
  676. void FindInFilesPanel::set_progress_visible(bool visible) {
  677. _progress_bar->set_self_modulate(Color(1, 1, 1, visible ? 1 : 0));
  678. }
  679. void FindInFilesPanel::_bind_methods() {
  680. ClassDB::bind_method("_on_result_found", &FindInFilesPanel::_on_result_found);
  681. ClassDB::bind_method("_on_item_edited", &FindInFilesPanel::_on_item_edited);
  682. ClassDB::bind_method("_on_finished", &FindInFilesPanel::_on_finished);
  683. ClassDB::bind_method("_on_cancel_button_clicked", &FindInFilesPanel::_on_cancel_button_clicked);
  684. ClassDB::bind_method("_on_result_selected", &FindInFilesPanel::_on_result_selected);
  685. ClassDB::bind_method("_on_replace_text_changed", &FindInFilesPanel::_on_replace_text_changed);
  686. ClassDB::bind_method("_on_replace_all_clicked", &FindInFilesPanel::_on_replace_all_clicked);
  687. ClassDB::bind_method("_draw_result_text", &FindInFilesPanel::draw_result_text);
  688. ADD_SIGNAL(MethodInfo(SIGNAL_RESULT_SELECTED,
  689. PropertyInfo(Variant::STRING, "path"),
  690. PropertyInfo(Variant::INT, "line_number"),
  691. PropertyInfo(Variant::INT, "begin"),
  692. PropertyInfo(Variant::INT, "end")));
  693. ADD_SIGNAL(MethodInfo(SIGNAL_FILES_MODIFIED, PropertyInfo(Variant::STRING, "paths")));
  694. }