Files
eepp/src/tools/ecode/globalsearchcontroller.cpp
Martín Lucas Golini d6f60354c1 Improve UIWindow default focusing behavior.
Adjust `window-flags` behavior when used from CSS.
Display a download progress window when download-file-web is used. Also support loading images and sounds from downloads.
Minor safety checks added for engine termination and window closing.
Minor changes in Http.
2026-01-10 01:49:20 -03:00

961 lines
40 KiB
C++

#include "globalsearchcontroller.hpp"
#include "ecode.hpp"
#include "uitreeviewglobalsearch.hpp"
namespace ecode {
static int LOCATEBAR_MAX_VISIBLE_ITEMS = 18;
GlobalSearchController::GlobalSearchController( UICodeEditorSplitter* editorSplitter,
UISceneNode* sceneNode, App* app ) :
mSplitter( editorSplitter ), mUISceneNode( sceneNode ), mApp( app ) {
mApp->getPluginManager()->subscribeMessages(
"GlobalSearchController", [this]( const PluginMessage& msg ) -> PluginRequestHandle {
return processMessage( msg );
} );
}
static bool replaceInFile( const std::string& path, const std::string& replaceText,
const std::vector<std::pair<Int64, Int64>>& replacements ) {
std::string data;
if ( !FileSystem::fileGet( path, data ) )
return false;
Int64 diff = 0;
for ( const auto& range : replacements ) {
data.replace( range.first + diff, range.second - range.first, replaceText );
diff += replaceText.size() - ( range.second - range.first );
}
if ( !FileSystem::fileWrite( path, (const Uint8*)data.c_str(), data.size() ) )
return false;
return true;
}
static bool replaceInFile( const std::string& path, const std::vector<std::string>& replaceTexts,
const std::vector<std::pair<Int64, Int64>>& replacements ) {
std::string data;
if ( !FileSystem::fileGet( path, data ) )
return false;
Int64 diff = 0;
int pos = 0;
for ( const auto& range : replacements ) {
data.replace( range.first + diff, range.second - range.first, replaceTexts[pos] );
diff += replaceTexts[pos].size() - ( range.second - range.first );
pos++;
}
if ( !FileSystem::fileWrite( path, (const Uint8*)data.c_str(), data.size() ) )
return false;
return true;
}
static std::string getReplaceText( const ProjectSearch::ResultData::Result& result,
const std::string& replaceText ) {
std::string newText( replaceText );
LuaPattern ptrn( "$(%d+)" );
for ( auto& match : ptrn.gmatch( replaceText ) ) {
std::string matchSubStr( match.group( 0 ) ); // $1, $2, etc.
std::string matchNum( match.group( 1 ) ); // 1, 2, etc.
int num;
if ( String::fromString( num, matchNum ) && num > 0 &&
num - 1 < static_cast<int>( result.captures.size() ) ) {
String::replaceAll( newText, matchSubStr, result.captures[num - 1] );
}
}
return newText;
}
void GlobalSearchController::processNextReplaceInBuffer(
std::string&& replaceText, ProjectSearch::SearchConfig&& searchConfig,
std::function<void( int )>&& onDoneCb, int count,
std::vector<ProjectSearch::ResultData>&& pendingRes, bool hasCaptures ) {
if ( pendingRes.empty() ) {
onDoneCb( count );
return;
}
auto current = std::move( pendingRes.back() );
pendingRes.pop_back();
auto doc = mSplitter->findDocFromPath( current.file );
if ( doc ) {
auto results =
ProjectSearch::fileResFromDoc( searchConfig.searchString, searchConfig.caseSensitive,
searchConfig.wholeWord, searchConfig.type, doc );
for ( const auto& result : results ) {
if ( result.selected ) {
auto oldSel = doc->getSelections();
doc->setSelection( result.position );
doc->replaceSelection( hasCaptures ? getReplaceText( result, replaceText )
: replaceText );
count++;
}
}
processNextReplaceInBuffer( std::move( replaceText ), std::move( searchConfig ),
std::move( onDoneCb ), count, std::move( pendingRes ),
hasCaptures );
} else {
mSplitter->loadAsyncFileFromPathInNewTab(
current.file,
[this, current, count, pendingRes = std::move( pendingRes ),
replaceText = std::move( replaceText ), searchConfig = std::move( searchConfig ),
onDoneCb = std::move( onDoneCb ),
hasCaptures]( UICodeEditor* editor, const std::string& ) mutable {
auto doc = editor->getDocumentRef();
auto results = ProjectSearch::fileResFromDoc(
searchConfig.searchString, searchConfig.caseSensitive, searchConfig.wholeWord,
searchConfig.type, doc );
for ( const auto& result : results ) {
if ( result.selected ) {
auto oldSel = doc->getSelections();
doc->setSelection( result.position );
doc->replaceSelection( hasCaptures ? getReplaceText( result, replaceText )
: replaceText );
count++;
}
}
editor->scrollToCursor( true );
processNextReplaceInBuffer( std::move( replaceText ), std::move( searchConfig ),
std::move( onDoneCb ), count, std::move( pendingRes ),
hasCaptures );
} );
}
}
void GlobalSearchController::replaceInFiles( std::string replaceText,
std::shared_ptr<ProjectSearch::ResultModel> model,
std::function<void( int )> onDoneCb,
bool bufferOnlyMode,
ProjectSearch::SearchConfig searchConfig ) {
int count = 0;
if ( model->isResultFromSymbolReference() ) {
// TODO Implement replacement from result from symbol reference
return onDoneCb( count );
}
const ProjectSearch::Result& res = model->getResult();
bool hasCaptures =
model->isResultFromPatternMatch() && LuaPattern::hasMatches( replaceText, "$%d+" );
if ( bufferOnlyMode ) {
std::vector<ProjectSearch::ResultData> pendingResults;
for ( const auto& fileResult : res ) {
if ( !fileResult.selected )
continue;
if ( fileResult.openDoc ) {
for ( const auto& result : fileResult.results ) {
if ( !result.selected )
continue;
auto oldSel = fileResult.openDoc->getSelections();
fileResult.openDoc->setSelection( result.position );
fileResult.openDoc->replaceSelection(
hasCaptures ? getReplaceText( result, replaceText ) : replaceText );
fileResult.openDoc->setSelection( oldSel );
count++;
}
} else {
pendingResults.push_back( fileResult );
}
}
if ( pendingResults.empty() ) {
onDoneCb( count );
} else {
processNextReplaceInBuffer( std::move( replaceText ), std::move( searchConfig ),
std::move( onDoneCb ), count, std::move( pendingResults ),
hasCaptures );
}
return;
}
if ( hasCaptures ) {
for ( const auto& fileResult : res ) {
std::vector<std::pair<Int64, Int64>> replacements;
std::vector<std::string> replaceTexts;
for ( const auto& result : fileResult.results ) {
if ( !result.selected )
continue;
replaceTexts.emplace_back( getReplaceText( result, replaceText ) );
if ( fileResult.openDoc ) {
auto oldSel = fileResult.openDoc->getSelections();
fileResult.openDoc->setSelection( result.position );
fileResult.openDoc->replaceSelection( replaceTexts.back() );
fileResult.openDoc->setSelection( oldSel );
count++;
} else {
replacements.push_back( { result.start, result.end } );
}
}
if ( replacements.size() == replaceTexts.size() &&
replaceInFile( fileResult.file, replaceTexts, replacements ) )
count += static_cast<int>( replacements.size() );
}
return onDoneCb( count );
}
for ( const auto& fileResult : res ) {
std::vector<std::pair<Int64, Int64>> replacements;
for ( const auto& result : fileResult.results ) {
if ( result.selected ) {
if ( fileResult.openDoc ) {
auto oldSel = fileResult.openDoc->getSelections();
fileResult.openDoc->setSelection( result.position );
fileResult.openDoc->replaceSelection( replaceText );
fileResult.openDoc->setSelection( oldSel );
count++;
} else {
replacements.push_back( { result.start, result.end } );
}
}
}
if ( !replacements.empty() && replaceInFile( fileResult.file, replaceText, replacements ) )
count += static_cast<int>( replacements.size() );
}
onDoneCb( count );
}
void GlobalSearchController::showGlobalSearch() {
showGlobalSearch( isUsingSearchReplaceTree() );
}
void GlobalSearchController::initGlobalSearchBar(
UIGlobalSearchBar* globalSearchBar, const GlobalSearchBarConfig& globalSearchBarConfig,
std::unordered_map<std::string, std::string> keybindings ) {
mGlobalSearchBarLayout = globalSearchBar;
mGlobalSearchBarLayout->setVisible( false )->setEnabled( false );
auto addClickListener = [this]( UIWidget* widget, std::string cmd ) {
widget->on( Event::MouseClick, [this, cmd]( const Event* event ) {
const MouseEvent* mouseEvent = static_cast<const MouseEvent*>( event );
if ( mouseEvent->getFlags() & EE_BUTTON_LMASK )
mGlobalSearchBarLayout->execute( cmd );
} );
};
auto& kbind = mGlobalSearchBarLayout->getKeyBindings();
kbind.addKeybindsString( { { mApp->getKeybind( "find-replace" ), "find-replace" } } );
kbind.addKeybindsStringUnordered( keybindings );
UIPushButton* searchButton = mGlobalSearchBarLayout->find<UIPushButton>( "global_search" );
UIPushButton* searchReplaceButton =
mGlobalSearchBarLayout->find<UIPushButton>( "global_search_replace" );
searchReplaceButton->setTooltipText(
kbind.getCommandKeybindString( "search-replace-in-files" ) );
UIPushButton* searchClearHistory =
mGlobalSearchBarLayout->find<UIPushButton>( "global_search_clear_history" );
searchClearHistory->setTooltipText(
kbind.getCommandKeybindString( "global-search-clear-history" ) );
UICheckBox* caseSensitiveChk = mGlobalSearchBarLayout->find<UICheckBox>( "case_sensitive" );
caseSensitiveChk->setTooltipText( kbind.getCommandKeybindString( "change-case" ) );
UICheckBox* wholeWordChk = mGlobalSearchBarLayout->find<UICheckBox>( "whole_word" );
wholeWordChk->setTooltipText( kbind.getCommandKeybindString( "change-whole-word" ) );
UICheckBox* regexChk = mGlobalSearchBarLayout->find<UICheckBox>( "regex" );
regexChk->setTooltipText( kbind.getCommandKeybindString( "toggle-regex" ) );
UICheckBox* luaPatternChk = mGlobalSearchBarLayout->find<UICheckBox>( "lua_pattern" );
luaPatternChk->setTooltipText( kbind.getCommandKeybindString( "toggle-lua-pattern" ) );
UICheckBox* escapeSequenceChk = mGlobalSearchBarLayout->find<UICheckBox>( "escape_sequence" );
std::string kbindEscape = kbind.getCommandKeybindString( "change-escape-sequence" );
if ( !kbindEscape.empty() )
escapeSequenceChk->setTooltipText( escapeSequenceChk->getTooltipText() + " (" +
kbindEscape + ")" );
UICheckBox* bufferOnlyModeChk = mGlobalSearchBarLayout->find<UICheckBox>( "buffer_only_mode" );
std::string kbindBufferOnlyMode = kbind.getCommandKeybindString( "buffer-only-mode" );
if ( !kbindBufferOnlyMode.empty() )
bufferOnlyModeChk->setTooltipText( bufferOnlyModeChk->getTooltipText() + " (" +
kbindBufferOnlyMode + ")" );
UIWidget* searchBarClose = mGlobalSearchBarLayout->find<UIWidget>( "global_searchbar_close" );
caseSensitiveChk->setChecked( globalSearchBarConfig.caseSensitive );
regexChk->setChecked( globalSearchBarConfig.regex );
luaPatternChk->setChecked( globalSearchBarConfig.luaPattern );
wholeWordChk->setChecked( globalSearchBarConfig.wholeWord );
escapeSequenceChk->setChecked( globalSearchBarConfig.escapeSequence );
bufferOnlyModeChk->setChecked( globalSearchBarConfig.bufferOnlyMode );
mGlobalSearchInput = mGlobalSearchBarLayout->find<UITextInput>( "global_search_find" );
mGlobalSearchWhereInput = mGlobalSearchBarLayout->find<UITextInput>( "global_search_where" );
mGlobalSearchHistoryList =
mGlobalSearchBarLayout->find<UIDropDownList>( "global_search_history" );
mGlobalSearchBarLayout->setCommand( "global-search-clear-history", [this] { clearHistory(); } );
mGlobalSearchBarLayout->setCommand(
"search-in-files",
[this, caseSensitiveChk, wholeWordChk, luaPatternChk, escapeSequenceChk, regexChk] {
doGlobalSearch( mGlobalSearchInput->getText(), mGlobalSearchWhereInput->getText(),
caseSensitiveChk->isChecked(), wholeWordChk->isChecked(),
luaPatternChk->isChecked()
? TextDocument::FindReplaceType::LuaPattern
: ( regexChk->isChecked() ? TextDocument::FindReplaceType::RegEx
: TextDocument::FindReplaceType::Normal ),
escapeSequenceChk->isChecked(), false );
} );
mGlobalSearchBarLayout->setCommand( "search-again", [this, caseSensitiveChk, wholeWordChk,
luaPatternChk, escapeSequenceChk,
regexChk] {
auto listBox = mGlobalSearchHistoryList->getListBox();
if ( listBox->getItemSelectedIndex() < mGlobalSearchHistory.size() ) {
const auto& item = mGlobalSearchHistory[mGlobalSearchHistory.size() - 1 -
listBox->getItemSelectedIndex()];
doGlobalSearch( item.search, item.filter, caseSensitiveChk->isChecked(),
wholeWordChk->isChecked(),
luaPatternChk->isChecked()
? TextDocument::FindReplaceType::LuaPattern
: ( regexChk->isChecked() ? TextDocument::FindReplaceType::RegEx
: TextDocument::FindReplaceType::Normal ),
escapeSequenceChk->isChecked(),
mGlobalSearchTreeReplace == mGlobalSearchTree, true );
}
} );
mGlobalSearchBarLayout->setCommand( "search-set-string", [this] {
auto listBox = mGlobalSearchHistoryList->getListBox();
const auto& item =
mGlobalSearchHistory[mGlobalSearchHistory.size() - 1 - listBox->getItemSelectedIndex()];
mGlobalSearchInput->setText( item.search );
} );
mGlobalSearchBarLayout->setCommand( "close-global-searchbar", [this] {
hideGlobalSearchBar();
if ( mSplitter->getCurWidget() )
mSplitter->getCurWidget()->setFocus();
} );
mGlobalSearchBarLayout->setCommand( "expand-all", [this] {
mGlobalSearchTree->expandAll();
mGlobalSearchTree->setFocus();
} );
mGlobalSearchBarLayout->setCommand( "collapse-all", [this] {
mGlobalSearchTree->collapseAll();
mGlobalSearchTree->setFocus();
} );
mGlobalSearchBarLayout->setCommand( "change-case", [caseSensitiveChk] {
caseSensitiveChk->setChecked( !caseSensitiveChk->isChecked() );
} );
mGlobalSearchBarLayout->setCommand( "change-whole-word", [wholeWordChk] {
wholeWordChk->setChecked( !wholeWordChk->isChecked() );
} );
mGlobalSearchBarLayout->setCommand(
"toggle-regex", [regexChk] { regexChk->setChecked( !regexChk->isChecked() ); } );
mGlobalSearchBarLayout->setCommand( "toggle-lua-pattern", [luaPatternChk] {
luaPatternChk->setChecked( !luaPatternChk->isChecked() );
} );
mGlobalSearchBarLayout->setCommand( "change-escape-sequence", [escapeSequenceChk] {
escapeSequenceChk->setChecked( !escapeSequenceChk->isChecked() );
} );
mGlobalSearchBarLayout->setCommand( "buffer-only-mode", [bufferOnlyModeChk] {
bufferOnlyModeChk->setChecked( !bufferOnlyModeChk->isChecked() );
} );
mGlobalSearchBarLayout->setCommand( "find-replace", [this] { mApp->showFindView(); } );
const auto pressEnterCb = [this]( const Event* event ) {
if ( event->getNode()->hasFocus() ) {
mGlobalSearchBarLayout->execute( "search-in-files" );
} else {
KeyEvent keyEvent( mGlobalSearchTree, Event::KeyDown, KEY_RETURN, SCANCODE_UNKNOWN, 0,
0 );
mGlobalSearchTree->forceKeyDown( keyEvent );
}
};
luaPatternChk->on( Event::OnValueChange, [this, luaPatternChk, regexChk]( const Event* ) {
if ( mValueChanging )
return;
BoolScopedOp op( mValueChanging, true );
if ( luaPatternChk->isChecked() && regexChk->isChecked() )
regexChk->setChecked( false );
} );
regexChk->on( Event::OnValueChange, [this, luaPatternChk, regexChk]( const Event* ) {
if ( mValueChanging )
return;
BoolScopedOp op( mValueChanging, true );
if ( regexChk->isChecked() && luaPatternChk->isChecked() )
luaPatternChk->setChecked( false );
} );
mGlobalSearchInput->setSelectAllDocOnTabNavigate( false );
mGlobalSearchWhereInput->setSelectAllDocOnTabNavigate( false );
mGlobalSearchInput->on( Event::OnPressEnter, pressEnterCb );
mGlobalSearchWhereInput->on( Event::OnPressEnter, pressEnterCb );
mGlobalSearchWhereInput->on( Event::OnTabNavigate,
[this]( auto ) { mGlobalSearchInput->setFocus(); } );
auto switchInputToTree = [this]( const Event* event ) {
const KeyEvent* keyEvent = static_cast<const KeyEvent*>( event );
Uint32 keyCode = keyEvent->getKeyCode();
if ( ( keyCode == KEY_UP || keyCode == KEY_DOWN || keyCode == KEY_PAGEUP ||
keyCode == KEY_PAGEDOWN || keyCode == KEY_HOME || keyCode == KEY_END ) &&
mGlobalSearchTree->forceKeyDown( *keyEvent ) && !mGlobalSearchTree->hasFocus() ) {
mGlobalSearchTree->setFocus();
}
};
mGlobalSearchInput->on( Event::KeyDown, switchInputToTree );
mGlobalSearchWhereInput->on( Event::KeyDown, switchInputToTree );
mGlobalSearchInput->on( Event::OnSizeChange, [this]( const Event* ) {
if ( mGlobalSearchBarLayout->isVisible() )
updateGlobalSearchBar();
} );
addClickListener( searchButton, "search-in-files" );
addClickListener( searchReplaceButton, "search-replace-in-files" );
addClickListener( searchBarClose, "close-global-searchbar" );
addClickListener( searchClearHistory, "global-search-clear-history" );
mGlobalSearchLayout = mUISceneNode
->loadLayoutFromString( R"xml(
<vbox id="global_search_layout" layout_width="fixed" layout_height="wrap_content" visible="false">
<hbox class="status_box" layout_width="match_parent" layout_height="wrap_content" visible="false">
<TextView layout_width="wrap_content" layout_height="match_parent" text="Searched for:" margin-right="4dp" />
<TextView class="search_str" layout_width="wrap_content" layout_height="match_parent" />
<PushButton id="global_search_again" layout_width="wrap_content" layout_height="18dp" text="Search Again" margin-left="8dp" />
<PushButton id="global_search_set_search" layout_width="wrap_content" layout_height="18dp" text="Set Search String" margin-left="8dp" />
<PushButton id="global_search_collapse" layout_width="wrap_content" layout_height="18dp" tooltip='@string(collapse_all, "Collapse All")' margin-left="8dp" icon="menu-fold" />
<PushButton id="global_search_expand" layout_width="wrap_content" layout_height="18dp" tooltip='@string(expand_all, "Expand All")' margin-left="8dp" icon="menu-unfold" />
<Widget layout_width="0" layout_weight="1" layout_height="match_parent" />
<TextView class="search_total" layout_width="wrap_content" layout_height="match_parent" margin-right="8dp" />
</hbox>
<hbox class="replace_box" layout_width="match_parent" layout_height="wrap_content" visible="false">
<TextView layout_width="wrap_content" layout_height="wrap_content" text="Replace with:" margin-right="4dp" />
<TextInput id="global_search_replace_input" class="small_input" layout_width="200dp" layout_height="18dp" padding="0" margin-right="4dp" />
<PushButton id="global_search_replace_button" layout_width="wrap_content" layout_height="18dp" text="Replace" />
</hbox>
</vbox>
)xml",
mUISceneNode->getRoot() )
->asType<UILayout>();
UIPushButton* searchAgainBtn = mGlobalSearchLayout->find<UIPushButton>( "global_search_again" );
searchAgainBtn->setTooltipText( kbind.getCommandKeybindString( "search-again" ) );
UIPushButton* searchSetSearch =
mGlobalSearchLayout->find<UIPushButton>( "global_search_set_search" );
UITextInput* replaceInput =
mGlobalSearchLayout->find<UITextInput>( "global_search_replace_input" );
UIPushButton* replaceButton =
mGlobalSearchLayout->find<UIPushButton>( "global_search_replace_button" );
UIPushButton* searchExpandButton =
mGlobalSearchLayout->find<UIPushButton>( "global_search_expand" );
searchExpandButton->setTooltipText( kbind.getCommandKeybindString( "expand-all" ) );
UIPushButton* searchCollapseButton =
mGlobalSearchLayout->find<UIPushButton>( "global_search_collapse" );
searchCollapseButton->setTooltipText( kbind.getCommandKeybindString( "collapse-all" ) );
addClickListener( searchAgainBtn, "search-again" );
addClickListener( searchSetSearch, "search-set-string" );
addClickListener( replaceButton, "replace-in-files" );
addClickListener( searchExpandButton, "expand-all" );
addClickListener( searchCollapseButton, "collapse-all" );
replaceInput->on( Event::OnPressEnter, [this, replaceInput]( const Event* ) {
if ( replaceInput->hasFocus() )
mGlobalSearchBarLayout->execute( "replace-in-files" );
} );
replaceInput->on( Event::KeyDown, switchInputToTree );
mGlobalSearchLayout->on( Event::KeyDown, [this]( const Event* event ) {
const KeyEvent* keyEvent = static_cast<const KeyEvent*>( event );
if ( keyEvent->getKeyCode() == KEY_ESCAPE ) {
mGlobalSearchBarLayout->execute( "close-global-searchbar" );
return;
}
mGlobalSearchBarLayout->forceKeyDown( *keyEvent );
} );
mGlobalSearchBarLayout->setCommand( "search-replace-in-files", [this, caseSensitiveChk,
wholeWordChk, luaPatternChk,
escapeSequenceChk, replaceInput,
regexChk] {
if ( mGlobalSearchTreeReplace == mGlobalSearchTree ) {
replaceInput->setFocus();
replaceInput->getDocument().selectAll();
return;
}
// TODO Implement replacement from result from symbol reference
/*if ( mGlobalSearchHistory.back().second->isResultFromSymbolReference() ) {
mGlobalSearchTreeReplace->setModel( mGlobalSearchHistory.back().second );
showGlobalSearch( true );
updateGlobalSearchBarResults( mGlobalSearchHistory.back().first,
mGlobalSearchHistory.back().second, true, false );
} else*/
{
doGlobalSearch( mGlobalSearchInput->getText(), mGlobalSearchWhereInput->getText(),
caseSensitiveChk->isChecked(), wholeWordChk->isChecked(),
luaPatternChk->isChecked()
? TextDocument::FindReplaceType::LuaPattern
: ( regexChk->isChecked() ? TextDocument::FindReplaceType::RegEx
: TextDocument::FindReplaceType::Normal ),
escapeSequenceChk->isChecked(), true );
}
} );
mGlobalSearchBarLayout->setCommand(
"replace-in-files", [this, replaceInput, escapeSequenceChk, bufferOnlyModeChk] {
auto listBox = mGlobalSearchHistoryList->getListBox();
if ( listBox->getItemSelectedIndex() < mGlobalSearchHistory.size() ) {
const auto& replaceData = mGlobalSearchHistory[mGlobalSearchHistory.size() - 1 -
listBox->getItemSelectedIndex()];
String text( replaceInput->getText() );
if ( escapeSequenceChk->isChecked() )
text.unescape();
replaceInFiles(
text.toUtf8(), replaceData.result,
[this]( int count ) {
mGlobalSearchBarLayout->ensureMainThread( [this, count] {
mGlobalSearchBarLayout->execute( "search-again" );
mGlobalSearchBarLayout->execute( "close-global-searchbar" );
mApp->getNotificationCenter()->addNotification(
String::format( "Replaced %zu occurrences.", count ) );
} );
},
bufferOnlyModeChk->isChecked(), mLastSearchConfig );
}
} );
mGlobalSearchTreeSearch =
UITreeViewGlobalSearch::New( mSplitter->getCurrentColorScheme(), false );
mGlobalSearchTreeReplace =
UITreeViewGlobalSearch::New( mSplitter->getCurrentColorScheme(), true );
initGlobalSearchTree( mGlobalSearchTreeSearch );
initGlobalSearchTree( mGlobalSearchTreeReplace );
mGlobalSearchTreeReplace->on( Event::KeyDown, [this]( const Event* event ) {
const KeyEvent* keyEvent = static_cast<const KeyEvent*>( event );
if ( keyEvent->getSanitizedMod() == KeyMod::getDefaultModifier() &&
keyEvent->getKeyCode() == KEY_RETURN )
mGlobalSearchBarLayout->execute( "replace-in-files" );
} );
UIWidget* menuBtn =
mGlobalSearchBarLayout->find<UIWidget>( "global_search_filters_menu_button" );
menuBtn->onClick( [this, menuBtn]( auto ) {
UIPopUpMenu* menu = UIPopUpMenu::New();
menu->add( mApp->i18n( "add_include_filter", "Add Include Filter" ) )
->setId( "add-include-filter" );
menu->add( mApp->i18n( "add_exclude_filter", "Add Exclude Filter" ) )
->setId( "add-exclude-filter" );
menu->on( Event::OnItemClicked, [this]( const Event* event ) {
const auto& id = event->getNode()->getId();
String appendTxt = "";
bool addComma = !mGlobalSearchWhereInput->getText().empty() &&
mGlobalSearchWhereInput->getText().back() != ',';
if ( "add-include-filter" == id ) {
appendTxt = ( addComma ? String{ "," } : String{ "" } ) + "*.txt";
} else if ( "add-exclude-filter" == id ) {
appendTxt = ( addComma ? String{ "," } : String{ "" } ) + "-*.txt";
}
if ( !appendTxt.empty() )
mGlobalSearchWhereInput->setText( mGlobalSearchWhereInput->getText() + appendTxt );
} );
menu->showAtScreenPosition( menuBtn->convertToWorldSpace( { 0, 0 } ) );
menu->setFocus();
} );
mGlobalSearchTree = mGlobalSearchTreeSearch;
}
void GlobalSearchController::showGlobalSearch( bool searchReplace,
std::optional<std::string> pathFilters ) {
mApp->hideLocateBar();
mApp->hideSearchBar();
mApp->hideStatusTerminal();
mApp->hideStatusBuildOutput();
mApp->hideStatusAppOutput();
mApp->getStatusBar()->hideAllElements();
bool wasReplaceTree = mGlobalSearchTreeReplace == mGlobalSearchTree;
mGlobalSearchTree = searchReplace ? mGlobalSearchTreeReplace : mGlobalSearchTreeSearch;
mGlobalSearchTreeSearch->setVisible( !searchReplace );
mGlobalSearchTreeReplace->setVisible( searchReplace );
mGlobalSearchBarLayout->setVisible( true )->setEnabled( true );
mGlobalSearchInput->setFocus();
mGlobalSearchLayout->setVisible( true );
UICheckBox* escapeSequenceChk = mGlobalSearchBarLayout->find<UICheckBox>( "escape_sequence" );
if ( mSplitter->curEditorExistsAndFocused() &&
mSplitter->getCurEditor()->getDocument().hasSelection() ) {
auto& doc = mSplitter->getCurEditor()->getDocument();
String text = doc.getSelectedText();
if ( !doc.getSelection().inSameLine() ) {
text.escape();
if ( !escapeSequenceChk->isChecked() )
escapeSequenceChk->setChecked( true );
}
mGlobalSearchInput->setText( text );
}
mGlobalSearchInput->getDocument().selectAll();
auto* loader = mGlobalSearchLayout->getParent()->find( "loader" );
if ( loader )
loader->setVisible( true );
if ( !searchReplace ) {
mGlobalSearchLayout->findByClass( "replace_box" )->setVisible( searchReplace );
mGlobalSearchBarLayout->find( "buffer_only_mode" )->setVisible( searchReplace );
if ( wasReplaceTree ) {
updateGlobalSearchBarResults( mGlobalSearchTreeReplace->getSearchStr(),
std::static_pointer_cast<ProjectSearch::ResultModel>(
mGlobalSearchTreeReplace->getModelShared() ),
searchReplace, escapeSequenceChk->isChecked() );
}
}
if ( mGlobalSearchWhereInput && pathFilters )
mGlobalSearchWhereInput->setText( *pathFilters );
updateGlobalSearchBar();
mApp->getStatusBar()->updateState();
}
void GlobalSearchController::updateColorScheme( const SyntaxColorScheme& colorScheme ) {
mGlobalSearchTreeSearch->updateColorScheme( colorScheme );
mGlobalSearchTreeReplace->updateColorScheme( colorScheme );
}
bool GlobalSearchController::isUsingSearchReplaceTree() {
return mGlobalSearchTreeReplace == mGlobalSearchTree;
}
void GlobalSearchController::clearHistory() {
auto listBox = mGlobalSearchHistoryList->getListBox();
listBox->clear();
mGlobalSearchHistory.clear();
mGlobalSearchHistoryOnItemSelectedCb = 0;
mGlobalSearchTree->setSearchStr( "" );
mGlobalSearchTree->setModel( nullptr );
mGlobalSearchInput->setText( "" );
mGlobalSearchWhereInput->setText( "" );
mGlobalSearchLayout->findByClass( "status_box" )->setVisible( false );
mGlobalSearchLayout->findByClass<UITextView>( "search_str" )->setText( "" );
updateGlobalSearchBar();
}
GlobalSearchBarConfig GlobalSearchController::getGlobalSearchBarConfig() const {
UICheckBox* caseSensitiveChk = mGlobalSearchBarLayout->find<UICheckBox>( "case_sensitive" );
UICheckBox* wholeWordChk = mGlobalSearchBarLayout->find<UICheckBox>( "whole_word" );
UICheckBox* regexChk = mGlobalSearchBarLayout->find<UICheckBox>( "regex" );
UICheckBox* luaPatternChk = mGlobalSearchBarLayout->find<UICheckBox>( "lua_pattern" );
UICheckBox* escapeSequenceChk = mGlobalSearchBarLayout->find<UICheckBox>( "escape_sequence" );
UICheckBox* bufferOnlyModeChk = mGlobalSearchBarLayout->find<UICheckBox>( "buffer_only_mode" );
GlobalSearchBarConfig globalSearchBarConfig;
globalSearchBarConfig.caseSensitive = caseSensitiveChk->isChecked();
globalSearchBarConfig.regex = regexChk->isChecked();
globalSearchBarConfig.luaPattern = luaPatternChk->isChecked();
globalSearchBarConfig.wholeWord = wholeWordChk->isChecked();
globalSearchBarConfig.escapeSequence = escapeSequenceChk->isChecked();
globalSearchBarConfig.bufferOnlyMode = bufferOnlyModeChk->isChecked();
return globalSearchBarConfig;
}
void GlobalSearchController::updateGlobalSearchBar() {
mGlobalSearchBarLayout->runOnMainThread( [this] {
Float width = eeceil( mGlobalSearchInput->getPixelsSize().getWidth() );
Float rowHeight = mGlobalSearchTree->getRowHeight() * LOCATEBAR_MAX_VISIBLE_ITEMS;
mGlobalSearchLayout->setPixelsSize( width, 0 );
mGlobalSearchTree->setPixelsSize( width, rowHeight );
width -= mGlobalSearchTree->shouldVerticalScrollBeVisible()
? mGlobalSearchTree->getVerticalScrollBar()->getPixelsSize().getWidth()
: 0;
mGlobalSearchTree->setColumnWidth( 0, eeceil( width ) );
Vector2f pos( mGlobalSearchInput->convertToWorldSpace( { 0, 0 } ) );
pos = PixelDensity::pxToDp( pos );
mGlobalSearchLayout->setLayoutMarginLeft( pos.x );
mGlobalSearchLayout->setLayoutMarginTop( pos.y -
mGlobalSearchLayout->getSize().getHeight() );
} );
}
void GlobalSearchController::hideGlobalSearchBar() {
mGlobalSearchBarLayout->setEnabled( false )->setVisible( false );
mGlobalSearchLayout->setVisible( false );
auto* loader = mGlobalSearchLayout->getParent()->find( "loader" );
if ( loader ) {
loader->setVisible( false );
loader->close();
}
mApp->getStatusBar()->updateState();
if ( mCurSearch ) {
mCurSearch->active = false;
mApp->getThreadPool()->removeWithTag( mCurSearch->taskTag );
}
}
void GlobalSearchController::toggleGlobalSearchBar() {
if ( mGlobalSearchBarLayout->isVisible() ) {
mGlobalSearchBarLayout->execute( "close-global-searchbar" );
} else {
showGlobalSearch();
}
}
void GlobalSearchController::updateGlobalSearchBarResults(
const std::string& search, std::shared_ptr<ProjectSearch::ResultModel> model,
bool searchReplace, bool isEscaped ) {
updateGlobalSearchBar();
mGlobalSearchTree->hExtLanguageType = mApp->getProjectConfig().hExtLanguageType;
mGlobalSearchTree->setSearchStr( search );
mGlobalSearchTree->setModel( model );
if ( mGlobalSearchTree->getModel()->rowCount() < 50 )
mGlobalSearchTree->expandAll();
mGlobalSearchLayout->findByClass<UITextView>( "search_str" )->setText( search );
mGlobalSearchLayout->findByClass<UITextView>( "search_total" )
->setText( String::format( "%zu matches found.", model->resultCount() ) );
mGlobalSearchLayout->findByClass( "status_box" )->setVisible( true );
mGlobalSearchLayout->findByClass( "replace_box" )->setVisible( searchReplace );
mGlobalSearchBarLayout->find( "buffer_only_mode" )->setVisible( searchReplace );
if ( searchReplace && mGlobalSearchBarLayout->isVisible() ) {
auto* replaceInput =
mGlobalSearchLayout->find<UITextInput>( "global_search_replace_input" );
replaceInput->setText( isEscaped ? String( search ).escape().toUtf8() : search );
replaceInput->getDocument().selectAll();
replaceInput->setFocus();
}
}
void GlobalSearchController::updateGlobalSearchHistory(
std::shared_ptr<ProjectSearch::ResultModel> model, const std::string& search,
const std::string& filter, bool searchReplace, bool searchAgain, bool escapeSequence ) {
auto listBox = mGlobalSearchHistoryList->getListBox();
if ( !searchAgain ) {
mGlobalSearchHistory.push_back( { search, filter, model } );
if ( mGlobalSearchHistory.size() > 10 )
mGlobalSearchHistory.pop_front();
std::vector<String> items;
for ( auto item = mGlobalSearchHistory.rbegin(); item != mGlobalSearchHistory.rend();
item++ ) {
items.push_back( item->search );
}
listBox->clear();
listBox->addListBoxItems( items );
if ( mGlobalSearchHistoryOnItemSelectedCb )
mGlobalSearchHistoryList->removeEventListener( mGlobalSearchHistoryOnItemSelectedCb );
listBox->setSelected( 0 );
mGlobalSearchHistoryOnItemSelectedCb = mGlobalSearchHistoryList->on(
Event::OnItemSelected, [this, escapeSequence, searchReplace]( const Event* ) {
auto idx = mGlobalSearchHistoryList->getListBox()->getItemSelectedIndex();
auto idxItem = mGlobalSearchHistory.at( mGlobalSearchHistory.size() - 1 - idx );
updateGlobalSearchBarResults( idxItem.search, idxItem.result, searchReplace,
escapeSequence );
} );
} else if ( listBox->getItemSelectedIndex() < mGlobalSearchHistory.size() ) {
mGlobalSearchHistory[mGlobalSearchHistory.size() - 1 - listBox->getItemSelectedIndex()]
.result = model;
}
}
std::vector<GlobMatch> GlobalSearchController::parseGlobMatches( const String& str ) {
std::vector<GlobMatch> ret;
auto globs = str.split( ',' );
for ( auto& glob : globs ) {
String::trimInPlace( glob );
if ( !glob.empty() && glob[0] == '-' ) {
ret.push_back( { glob.substr( 1 ), true } );
} else {
ret.push_back( { glob, false } );
}
}
return ret;
}
void GlobalSearchController::doGlobalSearch( String text, String filter, bool caseSensitive,
bool wholeWord,
TextDocument::FindReplaceType searchType,
bool escapeSequence, bool searchReplace,
bool searchAgain ) {
if ( nullptr == mApp->getDirTree() || !mApp->getDirTree()->isReady() ) {
mApp->getNotificationCenter()->addNotification( mApp->i18n(
"project_still_indexing", "Project is still being indexed, please wait a moment." ) );
return;
}
if ( mApp->getDirTree()->getFilesCount() == 0 || text.empty() ) {
return;
}
mGlobalSearchTree = searchReplace ? mGlobalSearchTreeReplace : mGlobalSearchTreeSearch;
mGlobalSearchTreeSearch->setVisible( !searchReplace );
mGlobalSearchTreeReplace->setVisible( searchReplace );
mGlobalSearchLayout->findByClass( "status_box" )->setVisible( true );
mGlobalSearchLayout->findByClass( "replace_box" )->setVisible( false );
mGlobalSearchLayout->findByClass<UITextView>( "search_str" )->setText( text );
UILoader* loader = UILoader::New();
loader->setId( "loader" );
loader->setRadius( 48 );
loader->setOutlineThickness( 6 );
loader->setParent( mGlobalSearchLayout->getParent() );
loader->setPosition( mGlobalSearchLayout->getPosition() +
mGlobalSearchLayout->getSize() * 0.5f - loader->getSize() * 0.5f );
Clock clock;
if ( escapeSequence )
text.unescape();
std::string search( text.toUtf8() );
std::vector<std::shared_ptr<TextDocument>> openDocs;
mSplitter->forEachDocSharedPtr(
[&openDocs]( auto doc ) { openDocs.emplace_back( std::move( doc ) ); } );
auto filters = parseGlobMatches( filter );
auto imageExts = Image::getImageExtensionsSupported();
imageExts.erase( std::remove_if( imageExts.begin(), imageExts.end(),
[]( const std::string& ext ) { return ext == "svg"; } ),
imageExts.end() );
filters.reserve( filters.size() + imageExts.size() );
for ( const auto& ext : imageExts ) {
// If user explicitly requested to filter some extension, respect it and do not add the
// ignore for it.
std::string extGlob( "*." + ext );
if ( std::find_if( filters.begin(), filters.end(),
[&extGlob]( const std::pair<std::string, bool>& filter ) {
return filter.first == extGlob;
} ) != filters.end() )
continue;
filters.emplace_back( extGlob, true );
}
mCurSearch = ProjectSearch::find(
mApp->getDirTree()->getFiles(), search, mApp->getThreadPool(),
[this, clock, search, searchReplace, searchAgain, escapeSequence, searchType,
filter]( const ProjectSearch::ConsolidatedResult& res ) {
Log::info( "Global search for \"%s\" took %s", search.c_str(),
clock.getElapsedTime().toString() );
mUISceneNode->runOnMainThread( [this, res = std::move( res ), search, searchReplace,
searchAgain, escapeSequence, searchType, filter] {
mLastSearchConfig = std::move( res.first );
auto model = ProjectSearch::asModel( res.second );
model->setOpType( searchType );
updateGlobalSearchHistory( model, search, filter, searchReplace, searchAgain,
escapeSequence );
updateGlobalSearchBarResults( search, model, searchReplace, escapeSequence );
auto* loader = mGlobalSearchLayout->getParent()->find( "loader" );
if ( loader ) {
loader->setVisible( false );
loader->close();
}
mCurSearch = nullptr;
} );
},
caseSensitive, wholeWord, searchType, filters, mApp->getCurrentProject(), openDocs );
}
void GlobalSearchController::onLoadDone( const Variant& lineNum, const Variant& colNum ) {
if ( mSplitter->curEditorExistsAndFocused() && lineNum.isValid() && colNum.isValid() &&
lineNum.is( Variant::Type::Int64 ) && colNum.is( Variant::Type::Int64 ) ) {
TextPosition pos{ lineNum.asInt64(), colNum.asInt64() };
mSplitter->getCurEditor()->getDocument().setSelection( pos );
mSplitter->getCurEditor()->goToLine( pos );
mSplitter->addCurrentPositionToNavigationHistory();
hideGlobalSearchBar();
}
}
void GlobalSearchController::initGlobalSearchTree( UITreeViewGlobalSearch* searchTree ) {
searchTree->addClass( "search_tree" );
searchTree->setParent( mGlobalSearchLayout );
searchTree->setVisible( false );
searchTree->setLayoutSizePolicy( SizePolicy::MatchParent, SizePolicy::Fixed );
searchTree->setExpanderIconSize( PixelDensity::dpToPx( 20 ) );
searchTree->setHeadersVisible( false );
searchTree->setColumnsHidden(
{ ProjectSearch::ResultModel::Line, ProjectSearch::ResultModel::ColumnStart }, true );
searchTree->on( Event::OnModelEvent, [this]( const Event* event ) {
const ModelEvent* modelEvent = static_cast<const ModelEvent*>( event );
if ( modelEvent->getModelEventType() == ModelEventType::Open ) {
const Model* model = modelEvent->getModel();
if ( !model )
return;
if ( mGlobalSearchTreeReplace == mGlobalSearchTree ) {
if ( modelEvent->getTriggerEvent()->getType() == Event::KeyDown ) {
const KeyEvent* keyEvent =
static_cast<const KeyEvent*>( modelEvent->getTriggerEvent() );
if ( keyEvent->getKeyCode() == KEY_SPACE &&
keyEvent->getNode()->isType( UI_TYPE_TREEVIEW_CELL ) ) {
auto* cell =
static_cast<UITreeViewCellGlobalSearch*>( keyEvent->getNode() );
cell->toggleSelected();
return;
}
}
}
Variant vPath( model->data( model->index( modelEvent->getModelIndex().internalId(),
ProjectSearch::ResultModel::FileOrPosition ),
ModelRole::Custom ) );
if ( vPath.isValid() && vPath.isString() ) {
std::string path( vPath.toString() );
UITab* tab = mSplitter->isDocumentOpen( path );
Variant lineNum(
model->data( model->index( modelEvent->getModelIndex().row(),
ProjectSearch::ResultModel::FileOrPosition,
modelEvent->getModelIndex().parent() ),
ModelRole::Custom ) );
Variant colNum( model->data( model->index( modelEvent->getModelIndex().row(),
ProjectSearch::ResultModel::ColumnStart,
modelEvent->getModelIndex().parent() ),
ModelRole::Custom ) );
if ( !tab ) {
FileInfo fileInfo( path );
if ( fileInfo.exists() && fileInfo.isRegularFile() )
mApp->loadFileFromPath(
path, true, nullptr,
[this, lineNum, colNum]( UICodeEditor*, const std::string& ) {
onLoadDone( lineNum, colNum );
} );
} else {
tab->getTabWidget()->setTabSelected( tab );
onLoadDone( lineNum, colNum );
}
}
}
} );
}
PluginRequestHandle GlobalSearchController::processMessage( const PluginMessage& msg ) {
if ( msg.type != PluginMessageType::SymbolReference || !msg.isBroadcast() )
return {};
const ProjectSearch::Result& res = msg.asProjectSearchResult();
if ( res.empty() )
return {};
auto model = ProjectSearch::asModel( res );
model->removeLastNewLineCharacter();
model->setResultFromSymbolReference( true );
ProjectSearch::ResultData::Result sample;
for ( const auto& r : res ) {
if ( !r.results.empty() ) {
sample = r.results.front();
break;
}
}
if ( sample.line.empty() )
return {};
if ( sample.position.end().column() - sample.position.start().column() <= 0 )
return {};
auto search =
sample.line.substr( sample.position.start().column(),
sample.position.end().column() - sample.position.start().column() );
mUISceneNode->runOnMainThread( [this, search, model] {
showGlobalSearch( false );
mGlobalSearchInput->setText( search );
mGlobalSearchWhereInput->setText( "" );
updateGlobalSearchHistory( model, search, "", false, false, false );
updateGlobalSearchBarResults( search, model, false, false );
} );
return {};
}
} // namespace ecode