/* Open Tracker License Terms and Conditions Copyright (c) 1991-2000, Be Incorporated. All rights reserved. Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice applies to all licensees and shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF TITLE, MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL BE INCORPORATED BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF, OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. Except as contained in this notice, the name of Be Incorporated shall not be used in advertising or otherwise to promote the sale, use or other dealings in this Software without prior written authorization from Be Incorporated. Tracker(TM), Be(R), BeOS(R), and BeIA(TM) are trademarks or registered trademarks of Be Incorporated in the United States and other countries. Other brand product names are registered trademarks or trademarks of their respective holders. All rights reserved. */ #include "FindPanel.h" #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "Attributes.h" #include "AutoLock.h" #include "Commands.h" #include "ContainerWindow.h" #include "FSUtils.h" #include "FunctionObject.h" #include "IconMenuItem.h" #include "MimeTypes.h" #include "Tracker.h" #undef B_TRANSLATION_CONTEXT #define B_TRANSLATION_CONTEXT "FindPanel" const char* kAllMimeTypes = "mime/ALLTYPES"; const uint32 kNameModifiedMessage = 'nmmd'; const uint32 kSwitchToQueryTemplate = 'swqt'; const uint32 kRunSaveAsTemplatePanel = 'svtm'; const uint32 kLatchChanged = 'ltch'; const char* kDragNDropTypes[] = { B_QUERY_MIMETYPE, B_QUERY_TEMPLATE_MIMETYPE }; static const char* kDragNDropActionSpecifiers[] = { B_TRANSLATE_MARK("Create a Query"), B_TRANSLATE_MARK("Create a Query template") }; const uint32 kAttachFile = 'attf'; const int32 operators[] = { B_CONTAINS, B_EQ, B_NE, B_BEGINS_WITH, B_ENDS_WITH, B_GT, B_LT }; static const char* operatorLabels[] = { B_TRANSLATE_MARK("contains"), B_TRANSLATE_MARK("is"), B_TRANSLATE_MARK("is not"), B_TRANSLATE_MARK("starts with"), B_TRANSLATE_MARK("ends with"), B_TRANSLATE_MARK("greater than"), B_TRANSLATE_MARK("less than"), B_TRANSLATE_MARK("before"), B_TRANSLATE_MARK("after") }; namespace BPrivate { class MostUsedNames { public: MostUsedNames(const char* fileName, const char* directory, int32 maxCount = 5); ~MostUsedNames(); bool ObtainList(BList* list); void ReleaseList(); void AddName(const char*); protected: struct list_entry { char* name; int32 count; }; static int CompareNames(const void* a, const void* b); void LoadList(); void UpdateList(); const char* fFileName; const char* fDirectory; bool fLoaded; mutable Benaphore fLock; BList fList; int32 fCount; }; MostUsedNames gMostUsedMimeTypes("MostUsedMimeTypes", "Tracker"); // #pragma mark - MoreOptionsStruct void MoreOptionsStruct::EndianSwap(void*) { // noop for now } void MoreOptionsStruct::SetQueryTemporary(BNode* node, bool on) { MoreOptionsStruct saveMoreOptions; if (ReadAttr(node, kAttrQueryMoreOptions, kAttrQueryMoreOptionsForeign, B_RAW_TYPE, 0, &saveMoreOptions, sizeof(MoreOptionsStruct), &MoreOptionsStruct::EndianSwap) == B_OK) { saveMoreOptions.temporary = on; node->WriteAttr(kAttrQueryMoreOptions, B_RAW_TYPE, 0, &saveMoreOptions, sizeof(saveMoreOptions)); } } bool MoreOptionsStruct::QueryTemporary(const BNode* node) { MoreOptionsStruct saveMoreOptions; if (ReadAttr(node, kAttrQueryMoreOptions, kAttrQueryMoreOptionsForeign, B_RAW_TYPE, 0, &saveMoreOptions, sizeof(MoreOptionsStruct), &MoreOptionsStruct::EndianSwap) == kReadAttrFailed) { return false; } return saveMoreOptions.temporary; } // #pragma mark - FindWindow FindWindow::FindWindow(const entry_ref* newRef, bool editIfTemplateOnly) : BWindow(BRect(), B_TRANSLATE("Find"), B_TITLED_WINDOW, B_NOT_RESIZABLE | B_NOT_ZOOMABLE | B_CLOSE_ON_ESCAPE | B_AUTO_UPDATE_SIZE_LIMITS), fFile(TryOpening(newRef)), fFromTemplate(false), fEditTemplateOnly(false), fSaveAsTemplatePanel(NULL) { if (fFile != NULL) { fRef = *newRef; if (editIfTemplateOnly) { char type[B_MIME_TYPE_LENGTH]; if (BNodeInfo(fFile).GetType(type) == B_OK && strcasecmp(type, B_QUERY_TEMPLATE_MIMETYPE) == 0) { fEditTemplateOnly = true; SetTitle(B_TRANSLATE("Edit Query template")); } } } else { // no initial query, fall back on the default query template BEntry entry; GetDefaultQuery(entry); entry.GetRef(&fRef); if (entry.Exists()) fFile = TryOpening(&fRef); else { // no default query template yet fFile = new BFile(&entry, O_RDWR | O_CREAT); if (fFile->InitCheck() < B_OK) { delete fFile; fFile = NULL; } else SaveQueryAttributes(fFile, true); } } fFromTemplate = IsQueryTemplate(fFile); fBackground = new FindPanel(fFile, this, fFromTemplate, fEditTemplateOnly); SetLayout(new BGroupLayout(B_VERTICAL)); GetLayout()->AddView(fBackground); CenterOnScreen(); } FindWindow::~FindWindow() { delete fFile; delete fSaveAsTemplatePanel; } BFile* FindWindow::TryOpening(const entry_ref* ref) { if (!ref) return NULL; BFile* result = new BFile(ref, O_RDWR); if (result->InitCheck() != B_OK) { delete result; result = NULL; } return result; } void FindWindow::GetDefaultQuery(BEntry& entry) { BPath path; if (find_directory(B_USER_DIRECTORY, &path, true) == B_OK && path.Append("queries") == B_OK && (mkdir(path.Path(), 0777) == 0 || errno == EEXIST)) { BDirectory directory(path.Path()); entry.SetTo(&directory, "default"); } } bool FindWindow::IsQueryTemplate(BNode* file) { char type[B_MIME_TYPE_LENGTH]; if (BNodeInfo(file).GetType(type) != B_OK) return false; return strcasecmp(type, B_QUERY_TEMPLATE_MIMETYPE) == 0; } void FindWindow::SwitchToTemplate(const entry_ref* ref) { try { BEntry entry(ref, true); BFile templateFile(&entry, O_RDONLY); ThrowOnInitCheckError(&templateFile); fBackground->SwitchToTemplate(&templateFile); } catch (...) { ; } } const char* FindWindow::QueryName() const { if (fFromTemplate) { if (!fQueryNameFromTemplate.Length() && fFile != NULL) { fFile->ReadAttrString(kAttrQueryTemplateName, &fQueryNameFromTemplate); } return fQueryNameFromTemplate.String(); } if (fFile == NULL) return ""; return fRef.name; } static const char* MakeValidFilename(BString& string) { // make a file name that is legal under bfs and hfs - possibly could // add code here to accomodate FAT32 etc. too if (string.Length() > B_FILE_NAME_LENGTH - 1) { string.Truncate(B_FILE_NAME_LENGTH - 4); string += B_UTF8_ELLIPSIS; } // replace slashes int32 length = string.Length(); char* buf = string.LockBuffer(length); for (int32 index = length; index-- > 0;) { if (buf[index] == '/' /*|| buf[index] == ':'*/) buf[index] = '_'; } string.UnlockBuffer(length); return string.String(); } void FindWindow::GetPredicateString(BString& predicate, bool& dynamicDate) { BQuery query; switch (fBackground->Mode()) { case kByNameItem: fBackground->GetByNamePredicate(&query); query.GetPredicate(&predicate); break; case kByFormulaItem: { BTextControl* textControl = dynamic_cast(FindView("TextControl")); if (textControl != NULL) predicate.SetTo(textControl->Text(), 1023); break; } case kByAttributeItem: fBackground->GetByAttrPredicate(&query, dynamicDate); query.GetPredicate(&predicate); break; } } void FindWindow::GetDefaultName(BString& name) { fBackground->GetDefaultName(name); time_t timeValue = time(0); char namebuf[B_FILE_NAME_LENGTH]; tm timeData; localtime_r(&timeValue, &timeData); strftime(namebuf, 32, " - %b %d, %I:%M:%S %p", &timeData); name << namebuf; MakeValidFilename(name); } void FindWindow::SaveQueryAttributes(BNode* file, bool queryTemplate) { ThrowOnError(BNodeInfo(file).SetType( queryTemplate ? B_QUERY_TEMPLATE_MIMETYPE : B_QUERY_MIMETYPE)); // save date/time info for recent query support and transient query killer int32 currentTime = (int32)time(0); file->WriteAttr(kAttrQueryLastChange, B_INT32_TYPE, 0, ¤tTime, sizeof(int32)); int32 tmp = 1; file->WriteAttr("_trk/recentQuery", B_INT32_TYPE, 0, &tmp, sizeof(int32)); } status_t FindWindow::SaveQueryAsAttributes(BNode* file, BEntry* entry, bool queryTemplate, const BMessage* oldAttributes, const BPoint* oldLocation) { if (oldAttributes != NULL) { // revive old window settings BContainerWindow::SetLayoutState(file, oldAttributes); } if (oldLocation != NULL) { // and the file's location FSSetPoseLocation(entry, *oldLocation); } BNodeInfo(file).SetType(queryTemplate ? B_QUERY_TEMPLATE_MIMETYPE : B_QUERY_MIMETYPE); BString predicate; bool dynamicDate; GetPredicateString(predicate, dynamicDate); file->WriteAttrString(kAttrQueryString, &predicate); if (dynamicDate) { file->WriteAttr(kAttrDynamicDateQuery, B_BOOL_TYPE, 0, &dynamicDate, sizeof(dynamicDate)); } int32 tmp = 1; file->WriteAttr("_trk/recentQuery", B_INT32_TYPE, 0, &tmp, sizeof(int32)); // write some useful info to help locate the volume to query BMenuItem* item = fBackground->VolMenu()->FindMarked(); if (item != NULL) { dev_t dev; BMessage message; uint32 count = 0; int32 itemCount = fBackground->VolMenu()->CountItems(); for (int32 index = 2; index < itemCount; index++) { BMenuItem* item = fBackground->VolMenu()->ItemAt(index); if (!item->IsMarked()) continue; if (item->Message()->FindInt32("device", &dev) != B_OK) continue; count++; BVolume volume(dev); EmbedUniqueVolumeInfo(&message, &volume); } if (count > 0) { // do we need to embed any volumes ssize_t size = message.FlattenedSize(); BString buffer; status_t result = message.Flatten(buffer.LockBuffer(size), size); if (result == B_OK) { if (file->WriteAttr(kAttrQueryVolume, B_MESSAGE_TYPE, 0, buffer.String(), (size_t)size) != size) { return B_IO_ERROR; } } buffer.UnlockBuffer(); } // default to query for everything } fBackground->SaveWindowState(file, fEditTemplateOnly); // write out all the dialog items as attributes so that the query can // be reopened and edited later BView* focusedItem = CurrentFocus(); if (focusedItem != NULL) { // text controls never get the focus, their internal text views do BView* parent = focusedItem->Parent(); if (dynamic_cast(parent) != NULL) focusedItem = parent; // write out the current focus and, if text control, selection BString name(focusedItem->Name()); file->WriteAttrString("_trk/focusedView", &name); BTextControl* textControl = dynamic_cast(focusedItem); if (textControl != NULL && textControl->TextView() != NULL) { int32 selStart; int32 selEnd; textControl->TextView()->GetSelection(&selStart, &selEnd); file->WriteAttr("_trk/focusedSelStart", B_INT32_TYPE, 0, &selStart, sizeof(selStart)); file->WriteAttr("_trk/focusedSelEnd", B_INT32_TYPE, 0, &selEnd, sizeof(selEnd)); } } return B_OK; } void FindWindow::Save() { FindSaveCommon(false); // close the find panel PostMessage(B_QUIT_REQUESTED); } void FindWindow::Find() { if (!FindSaveCommon(true)) { // have to wait for the node monitor to force old query to close // to avoid a race condition TTracker* tracker = dynamic_cast(be_app); ASSERT(tracker != NULL); for (int32 timeOut = 0; ; timeOut++) { if (tracker != NULL && !tracker->EntryHasWindowOpen(&fRef)) { // window quit, we can post refs received to open a // new copy break; } // PRINT(("waiting for query window to quit, %d\n", timeOut)); if (timeOut == 5000) { // the old query window would not quit for some reason TRESPASS(); PostMessage(B_QUIT_REQUESTED); return; } snooze(1000); } } int32 currentTime = (int32)time(0); fFile->WriteAttr(kAttrQueryLastChange, B_INT32_TYPE, 0, ¤tTime, sizeof(int32)); // tell the tracker about it BMessage message(B_REFS_RECEIVED); message.AddRef("refs", &fRef); be_app->PostMessage(&message); // close the find panel PostMessage(B_QUIT_REQUESTED); } bool FindWindow::FindSaveCommon(bool find) { // figure out what we need to do bool readFromOldFile = fFile != NULL; bool replaceOriginal = fFile && (!fFromTemplate || fEditTemplateOnly); bool keepPoseLocation = replaceOriginal; bool newFile = !fFile || (fFromTemplate && !fEditTemplateOnly); BEntry entry; BMessage oldAttributes; BPoint location; bool hadLocation = false; const char* userSpecifiedName = fBackground->UserSpecifiedName(); if (readFromOldFile) { entry.SetTo(&fRef); BContainerWindow::GetLayoutState(fFile, &oldAttributes); hadLocation = FSGetPoseLocation(fFile, &location); } if (replaceOriginal) { fFile->Unset(); entry.Remove(); // remove the current entry - need to do this to quit the // running query and to close the corresponding window if (userSpecifiedName != NULL && !fEditTemplateOnly) { // change the name of the old query per users request fRef.set_name(userSpecifiedName); entry.SetTo(&fRef); } } if (newFile) { // create query file in the user's directory BPath path; // there might be no queries folder yet, create one if (find_directory(B_USER_DIRECTORY, &path, true) == B_OK && path.Append("queries") == B_OK && (mkdir(path.Path(), 0777) == 0 || errno == EEXIST)) { // either use the user specified name, or go with the name // generated from the predicate, etc. BString name; if (userSpecifiedName == NULL) GetDefaultName(name); else name << userSpecifiedName; if (path.Append(name.String()) == B_OK) { entry.SetTo(path.Path()); entry.Remove(); entry.GetRef(&fRef); } } } fFile = new BFile(&entry, O_RDWR | O_CREAT); ASSERT(fFile->InitCheck() == B_OK); SaveQueryAsAttributes(fFile, &entry, !find, newFile ? 0 : &oldAttributes, (hadLocation && keepPoseLocation) ? &location : 0); return newFile; } void FindWindow::MessageReceived(BMessage* message) { switch (message->what) { case kFindButton: Find(); break; case kSaveButton: Save(); break; case kAttachFile: { entry_ref dir; const char* name; bool queryTemplate; if (message->FindString("name", &name) == B_OK && message->FindRef("directory", &dir) == B_OK && message->FindBool("template", &queryTemplate) == B_OK) { delete fFile; fFile = NULL; BDirectory directory(&dir); BEntry entry(&directory, name); entry_ref tmpRef; entry.GetRef(&tmpRef); fFile = TryOpening(&tmpRef); if (fFile != NULL) { fRef = tmpRef; SaveQueryAsAttributes(fFile, &entry, queryTemplate, 0, 0); // try to save whatever state we aleady have // to the new query so that if the user // opens it before runing it from the find panel, // something reasonable happens } } } break; case kSwitchToQueryTemplate: { entry_ref ref; if (message->FindRef("refs", &ref) == B_OK) SwitchToTemplate(&ref); break; } case kRunSaveAsTemplatePanel: if (fSaveAsTemplatePanel != NULL) fSaveAsTemplatePanel->Show(); else { BMessenger panel(BackgroundView()); fSaveAsTemplatePanel = new BFilePanel(B_SAVE_PANEL, &panel); fSaveAsTemplatePanel->SetSaveText( B_TRANSLATE("Query template")); fSaveAsTemplatePanel->Window()->SetTitle( B_TRANSLATE("Save as Query template:")); fSaveAsTemplatePanel->Show(); } break; default: _inherited::MessageReceived(message); break; } } // #pragma mark - FindPanel FindPanel::FindPanel(BFile* node, FindWindow* parent, bool fromTemplate, bool editTemplateOnly) : BView("MainView", B_WILL_DRAW), fMode(kByNameItem), fAttrGrid(NULL), fDraggableIcon(NULL) { SetViewUIColor(B_PANEL_BACKGROUND_COLOR); SetLowUIColor(ViewUIColor()); uint32 initialMode = InitialMode(node); BMessenger self(this); fRecentQueries = new BPopUpMenu(B_TRANSLATE("Recent queries"), false, false); AddRecentQueries(fRecentQueries, true, &self, kSwitchToQueryTemplate); // add popup for mime types fMimeTypeMenu = new BPopUpMenu("MimeTypeMenu"); fMimeTypeMenu->SetRadioMode(false); AddMimeTypesToMenu(); fMimeTypeField = new BMenuField("MimeTypeMenu", "", fMimeTypeMenu); fMimeTypeField->SetDivider(0.0f); fMimeTypeField->MenuItem()->SetLabel(B_TRANSLATE("All files and folders")); // add popup for search criteria fSearchModeMenu = new BPopUpMenu("searchMode"); fSearchModeMenu->AddItem(new BMenuItem(B_TRANSLATE("by name"), new BMessage(kByNameItem))); fSearchModeMenu->AddItem(new BMenuItem(B_TRANSLATE("by attribute"), new BMessage(kByAttributeItem))); fSearchModeMenu->AddItem(new BMenuItem(B_TRANSLATE("by formula"), new BMessage(kByFormulaItem))); fSearchModeMenu->ItemAt(initialMode == kByNameItem ? 0 : (initialMode == kByAttributeItem ? 1 : 2))->SetMarked(true); // mark the appropriate mode BMenuField* searchModeField = new BMenuField("", "", fSearchModeMenu); searchModeField->SetDivider(0.0f); // add popup for volume list fVolMenu = new BPopUpMenu("", false, false); BMenuField* volumeField = new BMenuField("", B_TRANSLATE("On"), fVolMenu); volumeField->SetDivider(volumeField->StringWidth(volumeField->Label()) + 8); AddVolumes(fVolMenu); if (!editTemplateOnly) { BPoint draggableIconOrigin(0, 0); BMessage dragNDropMessage(B_SIMPLE_DATA); dragNDropMessage.AddInt32("be:actions", B_COPY_TARGET); dragNDropMessage.AddString("be:types", B_FILE_MIME_TYPE); dragNDropMessage.AddString("be:filetypes", kDragNDropTypes[0]); dragNDropMessage.AddString("be:filetypes", kDragNDropTypes[1]); dragNDropMessage.AddString("be:actionspecifier", B_TRANSLATE_NOCOLLECT(kDragNDropActionSpecifiers[0])); dragNDropMessage.AddString("be:actionspecifier", B_TRANSLATE_NOCOLLECT(kDragNDropActionSpecifiers[1])); BMessenger self(this); BRect draggableRect = DraggableIcon::PreferredRect(draggableIconOrigin, B_LARGE_ICON); fDraggableIcon = new DraggableQueryIcon(draggableRect, "saveHere", &dragNDropMessage, self, B_FOLLOW_LEFT | B_FOLLOW_BOTTOM); fDraggableIcon->SetExplicitMaxSize( BSize(draggableRect.right - draggableRect.left, draggableRect.bottom - draggableRect.top)); BCursor grabCursor(B_CURSOR_ID_GRAB); fDraggableIcon->SetViewCursor(&grabCursor); } fQueryName = new BTextControl("query name", B_TRANSLATE("Query name:"), "", NULL, B_WILL_DRAW | B_NAVIGABLE | B_NAVIGABLE_JUMP); FillCurrentQueryName(fQueryName, parent); fSearchTrashCheck = new BCheckBox("searchTrash", B_TRANSLATE("Include trash"), NULL); fTemporaryCheck = new BCheckBox("temporary", B_TRANSLATE("Temporary"), NULL); fTemporaryCheck->SetValue(B_CONTROL_ON); BView* checkboxGroup = BLayoutBuilder::Group<>(B_HORIZONTAL) .Add(fSearchTrashCheck) .Add(fTemporaryCheck) .View(); // add the more options collapsible pane fMoreOptions = new BBox(B_NO_BORDER, BLayoutBuilder::Group<>() .AddGrid(B_USE_SMALL_SPACING, B_USE_SMALL_SPACING) .Add(fQueryName->CreateLabelLayoutItem(), 0, 0) .Add(fQueryName->CreateTextViewLayoutItem(), 1, 0) .Add(BSpaceLayoutItem::CreateHorizontalStrut(0), 0, 1) .Add(checkboxGroup, 1, 1) .End() .View()); // add Search button BButton* button; if (editTemplateOnly) { button = new BButton("save", B_TRANSLATE("Save"), new BMessage(kSaveButton), B_FOLLOW_RIGHT + B_FOLLOW_BOTTOM); } else { button = new BButton("find", B_TRANSLATE("Search"), new BMessage(kFindButton), B_FOLLOW_RIGHT + B_FOLLOW_BOTTOM); } button->MakeDefault(true); BView* icon = fDraggableIcon; if (icon == NULL) { icon = new BBox("no draggable icon", B_WILL_DRAW, B_NO_BORDER); icon->SetExplicitMaxSize(BSize(0, 0)); } BView* mimeTypeFieldSpacer = new BBox("MimeTypeMenuSpacer", B_WILL_DRAW, B_NO_BORDER); mimeTypeFieldSpacer->SetExplicitMaxSize(BSize(0, 0)); BBox* queryControls = new BBox("Box"); queryControls->SetBorder(B_NO_BORDER); BBox* queryBox = new BBox("Outer Controls"); queryBox->SetLabel(new BMenuField("RecentQueries", NULL, fRecentQueries)); BGroupView* queryBoxView = new BGroupView(B_VERTICAL, B_USE_DEFAULT_SPACING); queryBoxView->GroupLayout()->SetInsets(B_USE_DEFAULT_SPACING); queryBox->AddChild(queryBoxView); icon->SetExplicitAlignment(BAlignment(B_ALIGN_LEFT, B_ALIGN_BOTTOM)); button->SetExplicitAlignment(BAlignment(B_ALIGN_RIGHT, B_ALIGN_BOTTOM)); BLayoutBuilder::Group<>(queryBoxView, B_VERTICAL, B_USE_DEFAULT_SPACING) .SetInsets(B_USE_DEFAULT_SPACING) .AddGroup(B_HORIZONTAL, B_USE_SMALL_SPACING) .Add(fMimeTypeField) .Add(mimeTypeFieldSpacer) .Add(searchModeField) .AddStrut(B_USE_DEFAULT_SPACING) .Add(volumeField) .End() .Add(new BSeparatorView(B_HORIZONTAL, B_PLAIN_BORDER)) .Add(queryControls); BLayoutBuilder::Group<>(this, B_VERTICAL, B_USE_DEFAULT_SPACING) .SetInsets(B_USE_WINDOW_SPACING) .Add(queryBox) .AddGroup(B_HORIZONTAL, B_USE_DEFAULT_SPACING) .AddGroup(B_VERTICAL) .Add(icon) .AddGlue() .End() .Add(fMoreOptions) .AddGlue() .AddGroup(B_VERTICAL) .AddGlue() .Add(button) .End(); if (initialMode != kByAttributeItem) AddByNameOrFormulaItems(); else AddByAttributeItems(node); ResizeMenuField(fMimeTypeField); ResizeMenuField(searchModeField); ResizeMenuField(volumeField); } FindPanel::~FindPanel() { } void FindPanel::AttachedToWindow() { FindWindow* findWindow = dynamic_cast(Window()); ASSERT(findWindow != NULL); if (findWindow == NULL) return; BNode* node = findWindow->QueryNode(); fSearchModeMenu->SetTargetForItems(this); fQueryName->SetTarget(this); RestoreMimeTypeMenuSelection(node); // preselect the mime we used the last time have to do it here // because AddByAttributeItems will build different menus based // on which mime type is preselected RestoreWindowState(node); if (!findWindow->CurrentFocus()) { // try to pick a good focus if we restore to one already BTextControl* textControl = dynamic_cast(FindView("TextControl")); if (textControl == NULL) { // pick the last text control in the attribute view BString title("TextEntry"); title << (fAttrGrid->CountRows() - 1); textControl = dynamic_cast(FindView(title.String())); } if (textControl != NULL) textControl->MakeFocus(); } BButton* button = dynamic_cast(FindView("remove button")); if (button != NULL) button->SetTarget(this); button = dynamic_cast(FindView("add button")); if (button != NULL) button->SetTarget(this); fVolMenu->SetTargetForItems(this); // set target for MIME type items for (int32 index = MimeTypeMenu()->CountItems(); index-- > 2;) { BMenu* submenu = MimeTypeMenu()->ItemAt(index)->Submenu(); if (submenu != NULL) submenu->SetTargetForItems(this); } fMimeTypeMenu->SetTargetForItems(this); // set the MIME type to the default value, if no value is already selected if (fMimeTypeMenu->FindMarked() == NULL) { BMenuItem* firstItem = fMimeTypeMenu->ItemAt(0); if (firstItem != NULL) firstItem->SetMarked(true); } if (fDraggableIcon != NULL) fDraggableIcon->SetTarget(BMessenger(this)); fRecentQueries->SetTargetForItems(findWindow); } void FindPanel::ResizeMenuField(BMenuField* menuField) { BSize size; menuField->GetPreferredSize(&size.width, &size.height); BMenu* menu = menuField->Menu(); float padding = 0.0f; float width = 0.0f; BMenuItem* markedItem = menu->FindMarked(); if (markedItem != NULL) { if (markedItem->Submenu() != NULL) { BMenuItem* markedSubItem = markedItem->Submenu()->FindMarked(); if (markedSubItem != NULL && markedSubItem->Label() != NULL) { float labelWidth = menuField->StringWidth(markedSubItem->Label()); padding = size.width - labelWidth; } } else if (markedItem->Label() != NULL) { float labelWidth = menuField->StringWidth(markedItem->Label()); padding = size.width - labelWidth; } } for (int32 index = menu->CountItems(); index-- > 0; ) { BMenuItem* item = menu->ItemAt(index); if (item->Label() != NULL) width = std::max(width, menuField->StringWidth(item->Label())); BMenu* submenu = item->Submenu(); if (submenu != NULL) { for (int32 subIndex = submenu->CountItems(); subIndex-- > 0; ) { BMenuItem* subItem = submenu->ItemAt(subIndex); if (subItem->Label() == NULL) continue; width = std::max(width, menuField->StringWidth(subItem->Label())); } } } float maxWidth = be_control_look->DefaultItemSpacing() * 20; size.width = std::min(width + padding, maxWidth); menuField->SetExplicitSize(size); } static void PopUpMenuSetTitle(BMenu* menu, const char* title) { // This should really be in BMenuField BMenu* bar = menu->Supermenu(); ASSERT(bar); ASSERT(bar->ItemAt(0)); if (bar == NULL || !bar->ItemAt(0)) return; bar->ItemAt(0)->SetLabel(title); } void FindPanel::ShowVolumeMenuLabel() { if (fVolMenu->ItemAt(0)->IsMarked()) { // "all disks" selected PopUpMenuSetTitle(fVolMenu, fVolMenu->ItemAt(0)->Label()); return; } // find out if more than one items are marked int32 count = fVolMenu->CountItems(); int32 countSelected = 0; BMenuItem* tmpItem = NULL; for (int32 index = 2; index < count; index++) { BMenuItem* item = fVolMenu->ItemAt(index); if (item->IsMarked()) { countSelected++; tmpItem = item; } } if (countSelected == 0) { // no disk selected, for now revert to search all disks // ToDo: // show no disks here and add a check that will not let the // query go if the user doesn't pick at least one fVolMenu->ItemAt(0)->SetMarked(true); PopUpMenuSetTitle(fVolMenu, fVolMenu->ItemAt(0)->Label()); } else if (countSelected > 1) // if more than two disks selected, don't use the disk name // as a label PopUpMenuSetTitle(fVolMenu, B_TRANSLATE("multiple disks")); else { ASSERT(tmpItem); PopUpMenuSetTitle(fVolMenu, tmpItem->Label()); } } void FindPanel::Draw(BRect) { if (fAttrGrid == NULL) return; for (int32 index = 0; index < fAttrGrid->CountRows(); index++) { BMenuField* menuField = dynamic_cast(FindAttrView("MenuField", index)); if (menuField == NULL) continue; BLayoutItem* stringViewLayoutItem = fAttrGrid->ItemAt(1, index); if (stringViewLayoutItem == NULL) continue; BMenu* menuFieldMenu = menuField->Menu(); if (menuFieldMenu == NULL) continue; BMenuItem* item = menuFieldMenu->FindMarked(); if (item == NULL || item->Submenu() == NULL || item->Submenu()->FindMarked() == NULL) { continue; } if (stringViewLayoutItem == NULL) { stringViewLayoutItem = fAttrGrid->AddView(new BStringView("", item->Submenu()->FindMarked()->Label()), 1, index); stringViewLayoutItem->SetExplicitAlignment(BAlignment(B_ALIGN_RIGHT, B_ALIGN_VERTICAL_UNSET)); } if (stringViewLayoutItem != NULL) { BStringView* stringView = dynamic_cast(stringViewLayoutItem->View()); if (stringView != NULL) { BMenu* submenu = item->Submenu(); if (submenu != NULL) { BMenuItem* selected = submenu->FindMarked(); if (selected != NULL) stringView->SetText(selected->Label()); } } } } } void FindPanel::MessageReceived(BMessage* message) { entry_ref dir; const char* name; BMenuItem* item; switch (message->what) { case kVolumeItem: { // volume changed BMenuItem* invokedItem; dev_t dev; if (message->FindPointer("source", (void**)&invokedItem) != B_OK) return; if (message->FindInt32("device", &dev) != B_OK) break; BMenu* menu = invokedItem->Menu(); ASSERT(menu); if (dev == -1) { // all disks selected, uncheck everything else int32 count = menu->CountItems(); for (int32 index = 2; index < count; index++) menu->ItemAt(index)->SetMarked(false); // make all disks the title and check it PopUpMenuSetTitle(menu, menu->ItemAt(0)->Label()); menu->ItemAt(0)->SetMarked(true); } else { // a specific volume selected, unmark "all disks" menu->ItemAt(0)->SetMarked(false); // toggle mark on invoked item int32 count = menu->CountItems(); for (int32 index = 2; index < count; index++) { BMenuItem* item = menu->ItemAt(index); if (invokedItem == item) { // we just selected this bool wasMarked = item->IsMarked(); item->SetMarked(!wasMarked); } } } // make sure the right label is showing ShowVolumeMenuLabel(); break; } case kByAttributeItem: case kByNameItem: case kByFormulaItem: SwitchMode(message->what); break; case kAddItem: AddAttrRow(); break; case kRemoveItem: RemoveAttrRow(); break; case kMIMETypeItem: { BMenuItem* item; if (message->FindPointer("source", (void**)&item) == B_OK) { // don't add the "All files and folders" to the list if (fMimeTypeMenu->IndexOf(item) != 0) gMostUsedMimeTypes.AddName(item->Label()); SetCurrentMimeType(item); } if (fMode == kByAttributeItem) { // the attributes for this type may be different RemoveAttrViewItems(false); AddAttrRow(); } break; } case kNameModifiedMessage: // the query name was edited, make the query permanent fTemporaryCheck->SetValue(0); break; case kAttributeItem: if (message->FindPointer("source", (void**)&item) != B_OK) return; item->Menu()->Superitem()->SetMarked(true); Invalidate(); break; case kAttributeItemMain: // in case someone selected just an attribute without the // comparator if (message->FindPointer("source", (void**)&item) != B_OK) return; if (item->Submenu()->ItemAt(0) != NULL) item->Submenu()->ItemAt(0)->SetMarked(true); Invalidate(); break; case kLatchChanged: { int32 value; if (message->FindInt32("be:value", &value) != B_OK) break; if (value == 0 && !fMoreOptions->IsHidden(this)) fMoreOptions->Hide(); else if (value == 1 && fMoreOptions->IsHidden(this)) fMoreOptions->Show(); break; } case B_SAVE_REQUESTED: { // finish saving query template from a SaveAs panel entry_ref ref; status_t error = message->FindRef("refs", &ref); if (error == B_OK) { // direct entry selected, convert to parent dir and name BEntry entry(&ref); error = entry.GetParent(&entry); if (error == B_OK) { entry.GetRef(&dir); name = ref.name; } } else { // parent dir and name selected error = message->FindRef("directory", &dir); if (error == B_OK) error = message->FindString("name", &name); } if (error == B_OK) SaveAsQueryOrTemplate(&dir, name, true); break; } case B_COPY_TARGET: { // finish drag&drop const char* str; const char* mimeType = NULL; const char* actionSpecifier = NULL; if (message->FindString("be:types", &str) == B_OK && strcasecmp(str, B_FILE_MIME_TYPE) == 0 && (message->FindString("be:actionspecifier", &actionSpecifier) == B_OK || message->FindString("be:filetypes", &mimeType) == B_OK) && message->FindString("name", &name) == B_OK && message->FindRef("directory", &dir) == B_OK) { bool query = false; bool queryTemplate = false; if (actionSpecifier && strcasecmp(actionSpecifier, B_TRANSLATE_NOCOLLECT( kDragNDropActionSpecifiers[0])) == 0) { query = true; } else if (actionSpecifier && strcasecmp(actionSpecifier, B_TRANSLATE_NOCOLLECT( kDragNDropActionSpecifiers[1])) == 0) { queryTemplate = true; } else if (mimeType && strcasecmp(mimeType, kDragNDropTypes[0]) == 0) { query = true; } else if (mimeType && strcasecmp(mimeType, kDragNDropTypes[1]) == 0) { queryTemplate = true; } if (query || queryTemplate) SaveAsQueryOrTemplate(&dir, name, queryTemplate); } break; } default: _inherited::MessageReceived(message); break; } } void FindPanel::SaveAsQueryOrTemplate(const entry_ref* dir, const char* name, bool queryTemplate) { BDirectory directory(dir); BFile file(&directory, name, O_RDWR | O_CREAT | O_TRUNC); BNodeInfo(&file).SetType(queryTemplate ? B_QUERY_TEMPLATE_MIMETYPE : B_QUERY_MIMETYPE); BMessage attach(kAttachFile); attach.AddRef("directory", dir); attach.AddString("name", name); attach.AddBool("template", queryTemplate); Window()->PostMessage(&attach, 0); } BView* FindPanel::FindAttrView(const char* name, int row) const { for (int32 index = 0; index < fAttrGrid->CountColumns(); index++) { BLayoutItem* item = fAttrGrid->ItemAt(index, row); if (item == NULL) continue; BView* view = item->View(); if (view == NULL) continue; view = view->FindView(name); if (view != NULL) return view; } return NULL; } void FindPanel::BuildAttrQuery(BQuery* query, bool& dynamicDate) const { dynamicDate = false; // go through each attrview and add the attr and comparison info for (int32 index = 0; index < fAttrGrid->CountRows(); index++) { BString title; title << "TextEntry" << index; BTextControl* textControl = dynamic_cast( FindAttrView(title, index)); if (textControl == NULL) return; BMenuField* menuField = dynamic_cast( FindAttrView("MenuField", index)); if (menuField == NULL) return; BMenuItem* item = menuField->Menu()->FindMarked(); if (item == NULL) continue; BMessage* message = item->Message(); int32 type; if (message->FindInt32("type", &type) == B_OK) { const char* str; if (message->FindString("name", &str) == B_OK) query->PushAttr(str); else query->PushAttr(item->Label()); switch (type) { case B_STRING_TYPE: query->PushString(textControl->Text(), true); break; case B_TIME_TYPE: { int flags = 0; DEBUG_ONLY(time_t result =) parsedate_etc(textControl->Text(), -1, &flags); dynamicDate = (flags & PARSEDATE_RELATIVE_TIME) != 0; PRINT(("parsedate_etc - date is %srelative, %" B_PRIdTIME "\n", dynamicDate ? "" : "not ", result)); query->PushDate(textControl->Text()); break; } case B_BOOL_TYPE: { uint32 value; if (strcasecmp(textControl->Text(), "true") == 0) { value = 1; } else if (strcasecmp(textControl->Text(), "false") == 0) { value = 0; } else value = (uint32)atoi(textControl->Text()); value %= 2; query->PushUInt32(value); break; } case B_UINT8_TYPE: case B_UINT16_TYPE: case B_UINT32_TYPE: query->PushUInt32((uint32)StringToScalar( textControl->Text())); break; case B_INT8_TYPE: case B_INT16_TYPE: case B_INT32_TYPE: query->PushInt32((int32)StringToScalar( textControl->Text())); break; case B_UINT64_TYPE: query->PushUInt64((uint64)StringToScalar( textControl->Text())); break; case B_OFF_T_TYPE: case B_INT64_TYPE: query->PushInt64(StringToScalar( textControl->Text())); break; case B_FLOAT_TYPE: { float floatVal; sscanf(textControl->Text(), "%f", &floatVal); query->PushFloat(floatVal); break; } case B_DOUBLE_TYPE: { double doubleVal; sscanf(textControl->Text(), "%lf", &doubleVal); query->PushDouble(doubleVal); break; } } } query_op theOperator; BMenuItem* operatorItem = item->Submenu()->FindMarked(); if (operatorItem && operatorItem->Message() != NULL) { operatorItem->Message()->FindInt32("operator", (int32*)&theOperator); query->PushOp(theOperator); } else query->PushOp(B_EQ); // add logic based on selection in Logic menufield if (index > 0) { menuField = dynamic_cast( FindAttrView("Logic", index - 1)); if (menuField) { item = menuField->Menu()->FindMarked(); if (item) { message = item->Message(); message->FindInt32("combine", (int32*)&theOperator); query->PushOp(theOperator); } } else query->PushOp(B_AND); } } } void FindPanel::PushMimeType(BQuery* query) const { const char* type; if (CurrentMimeType(&type) == NULL) return; if (strcmp(kAllMimeTypes, type)) { // add an asterisk if we are searching for a supertype char buffer[B_FILE_NAME_LENGTH]; if (strchr(type, '/') == NULL) { strlcpy(buffer, type, sizeof(buffer)); strlcat(buffer, "/*", sizeof(buffer)); type = buffer; } query->PushAttr(kAttrMIMEType); query->PushString(type); query->PushOp(B_EQ); query->PushOp(B_AND); } } void FindPanel::GetByAttrPredicate(BQuery* query, bool& dynamicDate) const { ASSERT(Mode() == (int32)kByAttributeItem); BuildAttrQuery(query, dynamicDate); PushMimeType(query); } void FindPanel::GetDefaultName(BString& name) const { BTextControl* textControl = dynamic_cast( FindView("TextControl")); switch (Mode()) { case kByNameItem: if (textControl != NULL) { name.SetTo(B_TRANSLATE_COMMENT("Name = %name", "FindResultTitle")); name.ReplaceFirst("%name", textControl->Text()); } break; case kByFormulaItem: if (textControl != NULL) { name.SetTo(B_TRANSLATE_COMMENT("Formula %formula", "FindResultTitle")); name.ReplaceFirst("%formula", textControl->Text()); } break; case kByAttributeItem: { BMenuItem* item = fMimeTypeMenu->FindMarked(); if (item != NULL) name << item->Label() << ": "; for (int32 i = 0; i < fAttrGrid->CountRows(); i++) { GetDefaultAttrName(name, i); if (i + 1 < fAttrGrid->CountRows()) name << ", "; } break; } } } const char* FindPanel::UserSpecifiedName() const { if (fQueryName->Text()[0] == '\0') return NULL; return fQueryName->Text(); } void FindPanel::GetByNamePredicate(BQuery* query) const { ASSERT(Mode() == (int32)kByNameItem); BTextControl* textControl = dynamic_cast(FindView("TextControl")); ASSERT(textControl != NULL); if (textControl == NULL) return; query->PushAttr("name"); query->PushString(textControl->Text(), true); if (strstr(textControl->Text(), "*") != NULL) { // assume pattern is a regular expression, try doing an exact match query->PushOp(B_EQ); } else query->PushOp(B_CONTAINS); PushMimeType(query); } void FindPanel::SwitchMode(uint32 mode) { if (fMode == mode) // no work, bail return; uint32 oldMode = fMode; BString buffer; switch (mode) { case kByFormulaItem: { if (oldMode == kByAttributeItem || oldMode == kByNameItem) { BQuery query; if (oldMode == kByAttributeItem) { bool dummy; GetByAttrPredicate(&query, dummy); } else GetByNamePredicate(&query); query.GetPredicate(&buffer); } } // fall-through case kByNameItem: { fMode = mode; RemoveByAttributeItems(); ShowOrHideMimeTypeMenu(); AddByNameOrFormulaItems(); if (buffer.Length() > 0) { ASSERT(mode == kByFormulaItem || oldMode == kByAttributeItem); BTextControl* textControl = dynamic_cast(FindView("TextControl")); if (textControl != NULL) textControl->SetText(buffer.String()); } break; } case kByAttributeItem: { fMode = mode; BTextControl* textControl = dynamic_cast(FindView("TextControl")); if (textControl != NULL) { textControl->RemoveSelf(); delete textControl; } ShowOrHideMimeTypeMenu(); AddAttrRow(); break; } } } BMenuItem* FindPanel::CurrentMimeType(const char** type) const { // search for marked item in the list BMenuItem* item = MimeTypeMenu()->FindMarked(); if (item != NULL && MimeTypeMenu()->IndexOf(item) != 0 && item->Submenu() == NULL) { // if it's one of the most used items, ignore it item = NULL; } if (item == NULL) { for (int32 index = MimeTypeMenu()->CountItems(); index-- > 0;) { BMenu* submenu = MimeTypeMenu()->ItemAt(index)->Submenu(); if (submenu != NULL && (item = submenu->FindMarked()) != NULL) break; } } if (type != NULL && item != NULL) { BMessage* message = item->Message(); if (message == NULL) return NULL; if (message->FindString("mimetype", type) != B_OK) return NULL; } return item; } status_t FindPanel::SetCurrentMimeType(BMenuItem* item) { // unmark old MIME type (in most used list, and the tree) BMenuItem* marked = CurrentMimeType(); if (marked != NULL) { marked->SetMarked(false); if ((marked = MimeTypeMenu()->FindMarked()) != NULL) marked->SetMarked(false); } // mark new MIME type (in most used list, and the tree) if (item != NULL) { item->SetMarked(true); fMimeTypeField->MenuItem()->SetLabel(item->Label()); BMenuItem* search; for (int32 i = 2; (search = MimeTypeMenu()->ItemAt(i)) != NULL; i++) { if (item == search || search->Label() == NULL) continue; if (strcmp(item->Label(), search->Label()) == 0) { search->SetMarked(true); break; } BMenu* submenu = search->Submenu(); if (submenu == NULL) continue; for (int32 j = submenu->CountItems(); j-- > 0;) { BMenuItem* sub = submenu->ItemAt(j); if (strcmp(item->Label(), sub->Label()) == 0) { sub->SetMarked(true); break; } } } } return B_OK; } status_t FindPanel::SetCurrentMimeType(const char* label) { // unmark old MIME type (in most used list, and the tree) BMenuItem* marked = CurrentMimeType(); if (marked != NULL) { marked->SetMarked(false); if ((marked = MimeTypeMenu()->FindMarked()) != NULL) marked->SetMarked(false); } // mark new MIME type (in most used list, and the tree) fMimeTypeField->MenuItem()->SetLabel(label); bool found = false; for (int32 index = MimeTypeMenu()->CountItems(); index-- > 0;) { BMenuItem* item = MimeTypeMenu()->ItemAt(index); BMenu* submenu = item->Submenu(); if (submenu != NULL && !found) { for (int32 subIndex = submenu->CountItems(); subIndex-- > 0;) { BMenuItem* subItem = submenu->ItemAt(subIndex); if (subItem->Label() != NULL && strcmp(label, subItem->Label()) == 0) { subItem->SetMarked(true); found = true; } } } if (item->Label() != NULL && strcmp(label, item->Label()) == 0) { item->SetMarked(true); return B_OK; } } return found ? B_OK : B_ENTRY_NOT_FOUND; } static void AddSubtype(BString& text, const BMimeType& type) { text.Append(" ("); text.Append(strchr(type.Type(), '/') + 1); // omit the slash text.Append(")"); } bool FindPanel::AddOneMimeTypeToMenu(const ShortMimeInfo* info, void* castToMenu) { BPopUpMenu* menu = static_cast(castToMenu); BMimeType type(info->InternalName()); BMimeType super; type.GetSupertype(&super); if (super.InitCheck() < B_OK) return false; BMenuItem* superItem = menu->FindItem(super.Type()); if (superItem != NULL) { BMessage* message = new BMessage(kMIMETypeItem); message->AddString("mimetype", info->InternalName()); // check to ensure previous item's name differs BMenu* menu = superItem->Submenu(); BMenuItem* previous = menu->ItemAt(menu->CountItems() - 1); BString text = info->ShortDescription(); if (previous != NULL && strcasecmp(previous->Label(), info->ShortDescription()) == 0) { AddSubtype(text, type); // update the previous item as well BMimeType type(previous->Message()->GetString("mimetype", NULL)); BString label = ShortMimeInfo(type).ShortDescription(); AddSubtype(label, type); previous->SetLabel(label.String()); } menu->AddItem(new IconMenuItem(text.String(), message, info->InternalName())); } return false; } void FindPanel::AddMimeTypesToMenu() { BMessage* itemMessage = new BMessage(kMIMETypeItem); itemMessage->AddString("mimetype", kAllMimeTypes); IconMenuItem* firstItem = new IconMenuItem( B_TRANSLATE("All files and folders"), itemMessage, static_cast(NULL)); MimeTypeMenu()->AddItem(firstItem); MimeTypeMenu()->AddSeparatorItem(); // add recent MIME types TTracker* tracker = dynamic_cast(be_app); ASSERT(tracker != NULL); BList list; if (tracker != NULL && gMostUsedMimeTypes.ObtainList(&list)) { int32 count = 0; for (int32 index = 0; index < list.CountItems(); index++) { const char* name = (const char*)list.ItemAt(index); MimeTypeList* mimeTypes = tracker->MimeTypes(); if (mimeTypes != NULL) { const ShortMimeInfo* info = mimeTypes->FindMimeType(name); if (info == NULL) continue; BMessage* message = new BMessage(kMIMETypeItem); message->AddString("mimetype", info->InternalName()); MimeTypeMenu()->AddItem(new BMenuItem(name, message)); count++; } } if (count != 0) MimeTypeMenu()->AddSeparatorItem(); gMostUsedMimeTypes.ReleaseList(); } // add MIME type tree list BMessage types; if (BMimeType::GetInstalledSupertypes(&types) == B_OK) { const char* superType; int32 index = 0; while (types.FindString("super_types", index++, &superType) == B_OK) { BMenu* superMenu = new BMenu(superType); BMessage* message = new BMessage(kMIMETypeItem); message->AddString("mimetype", superType); MimeTypeMenu()->AddItem(new IconMenuItem(superMenu, message, superType)); // the MimeTypeMenu's font is not correct at this time superMenu->SetFont(be_plain_font); } } if (tracker != NULL) { tracker->MimeTypes()->EachCommonType( &FindPanel::AddOneMimeTypeToMenu, MimeTypeMenu()); } // remove empty super type menus (and set target) for (int32 index = MimeTypeMenu()->CountItems(); index-- > 2;) { BMenuItem* item = MimeTypeMenu()->ItemAt(index); BMenu* submenu = item->Submenu(); if (submenu == NULL) continue; if (submenu->CountItems() == 0) { MimeTypeMenu()->RemoveItem(item); delete item; } else submenu->SetTargetForItems(this); } } void FindPanel::AddVolumes(BMenu* menu) { // ToDo: add calls to this to rebuild the menu when a volume gets mounted BMessage* message = new BMessage(kVolumeItem); message->AddInt32("device", -1); menu->AddItem(new BMenuItem(B_TRANSLATE("All disks"), message)); menu->AddSeparatorItem(); PopUpMenuSetTitle(menu, B_TRANSLATE("All disks")); BVolumeRoster roster; BVolume volume; roster.Rewind(); while (roster.GetNextVolume(&volume) == B_OK) { if (volume.IsPersistent() && volume.KnowsQuery()) { BDirectory root; if (volume.GetRootDirectory(&root) != B_OK) continue; BEntry entry; root.GetEntry(&entry); Model model(&entry, true); if (model.InitCheck() != B_OK) continue; message = new BMessage(kVolumeItem); message->AddInt32("device", volume.Device()); menu->AddItem(new ModelMenuItem(&model, model.Name(), message)); } } if (menu->ItemAt(0)) menu->ItemAt(0)->SetMarked(true); menu->SetTargetForItems(this); } typedef std::pair EntryWithDate; static int SortByDatePredicate(const EntryWithDate* entry1, const EntryWithDate* entry2) { return entry1->second > entry2->second ? -1 : (entry1->second == entry2->second ? 0 : 1); } struct AddOneRecentParams { BMenu* menu; const BMessenger* target; uint32 what; }; static const entry_ref* AddOneRecentItem(const entry_ref* ref, void* castToParams) { AddOneRecentParams* params = (AddOneRecentParams*)castToParams; BMessage* message = new BMessage(params->what); message->AddRef("refs", ref); char type[B_MIME_TYPE_LENGTH]; BNode node(ref); BNodeInfo(&node).GetType(type); BMenuItem* item = new IconMenuItem(ref->name, message, type); item->SetTarget(*params->target); params->menu->AddItem(item); return NULL; } void FindPanel::AddRecentQueries(BMenu* menu, bool addSaveAsItem, const BMessenger* target, uint32 what) { BObjectList templates(10, true); BObjectList recentQueries(10, true); // find all the queries on all volumes BVolumeRoster roster; BVolume volume; roster.Rewind(); while (roster.GetNextVolume(&volume) == B_OK) { if (volume.IsPersistent() && volume.KnowsQuery() && volume.KnowsAttr()) { BQuery query; query.SetVolume(&volume); query.SetPredicate("_trk/recentQuery == 1"); if (query.Fetch() != B_OK) continue; entry_ref ref; while (query.GetNextRef(&ref) == B_OK) { // ignore queries in the Trash if (FSInTrashDir(&ref)) continue; char type[B_MIME_TYPE_LENGTH]; BNode node(&ref); BNodeInfo(&node).GetType(type); if (strcasecmp(type, B_QUERY_TEMPLATE_MIMETYPE) == 0) templates.AddItem(new entry_ref(ref)); else { uint32 changeTime; if (node.ReadAttr(kAttrQueryLastChange, B_INT32_TYPE, 0, &changeTime, sizeof(uint32)) != sizeof(uint32)) continue; recentQueries.AddItem(new EntryWithDate(ref, changeTime)); } } } } // we are only adding last ten queries recentQueries.SortItems(SortByDatePredicate); // but all templates AddOneRecentParams params; params.menu = menu; params.target = target; params.what = what; templates.EachElement(AddOneRecentItem, ¶ms); int32 count = recentQueries.CountItems(); if (count > 10) { // show only up to 10 recent queries count = 10; } else if (count < 0) count = 0; if (templates.CountItems() > 0 && count > 0) menu->AddSeparatorItem(); for (int32 index = 0; index < count; index++) AddOneRecentItem(&recentQueries.ItemAt(index)->first, ¶ms); if (addSaveAsItem) { // add a Save as template item if (count > 0 || templates.CountItems() > 0) menu->AddSeparatorItem(); BMessage* message = new BMessage(kRunSaveAsTemplatePanel); BMenuItem* item = new BMenuItem( B_TRANSLATE("Save query as template" B_UTF8_ELLIPSIS), message); menu->AddItem(item); } } void FindPanel::SetupAddRemoveButtons() { BBox* box = dynamic_cast(FindView("Box")); ASSERT(box != NULL); if (box == NULL) return; BButton* removeButton = new BButton("remove button", B_TRANSLATE("Remove"), new BMessage(kRemoveItem)); removeButton->SetEnabled(false); removeButton->SetTarget(this); BButton* addButton = new BButton("add button", B_TRANSLATE("Add"), new BMessage(kAddItem)); addButton->SetTarget(this); BGroupLayout* layout = dynamic_cast(box->GetLayout()); ASSERT(layout != NULL); if (layout == NULL) return; BLayoutBuilder::Group<>(layout) .AddGroup(B_HORIZONTAL) .AddGlue() .Add(removeButton) .Add(addButton) .End() .End(); } void FindPanel::FillCurrentQueryName(BTextControl* queryName, FindWindow* window) { ASSERT(window); queryName->SetText(window->QueryName()); } void FindPanel::AddAttrRow() { BBox* box = dynamic_cast(FindView("Box")); ASSERT(box != NULL); if (box == NULL) return; BGridView* grid = dynamic_cast(box->FindView("AttrFields")); if (grid == NULL) { // reset layout BLayoutBuilder::Group<>(box, B_VERTICAL); grid = new BGridView("AttrFields"); box->AddChild(grid); } fAttrGrid = grid->GridLayout(); AddAttributeControls(fAttrGrid->CountRows()); // add logic to previous attrview if (fAttrGrid->CountRows() > 1) AddLogicMenu(fAttrGrid->CountRows() - 2); BButton* removeButton = dynamic_cast( box->FindView("remove button")); if (removeButton != NULL) removeButton->SetEnabled(fAttrGrid->CountRows() > 1); else SetupAddRemoveButtons(); } void FindPanel::RemoveAttrRow() { if (fAttrGrid->CountRows() < 2) return; BView* view; int32 row = fAttrGrid->CountRows() - 1; for (int32 col = fAttrGrid->CountColumns(); col > 0; col--) { BLayoutItem* item = fAttrGrid->ItemAt(col - 1, row); if (item == NULL) continue; view = item->View(); if (view == NULL) continue; view->RemoveSelf(); delete view; } BString string = "TextEntry"; string << (row - 1); view = FindAttrView(string.String(), row - 1); if (view != NULL) view->MakeFocus(); if (fAttrGrid->CountRows() > 1) { // remove the And/Or menu field of the previous row BLayoutItem* item = fAttrGrid->ItemAt(3, row - 1); if (item == NULL) return; view = item->View(); if (view == NULL) return; view->RemoveSelf(); delete view; return; } // only one row remains // disable the remove button BButton* button = dynamic_cast(FindView("remove button")); if (button != NULL) button->SetEnabled(false); // remove the And/Or menu field BLayoutItem* item = fAttrGrid->RemoveItem(3); if (item == NULL) return; view = item->View(); if (view == NULL) return; view->RemoveSelf(); delete view; } uint32 FindPanel::InitialMode(const BNode* node) { if (node == NULL || node->InitCheck() != B_OK) return kByNameItem; uint32 result; if (node->ReadAttr(kAttrQueryInitialMode, B_INT32_TYPE, 0, (int32*)&result, sizeof(int32)) <= 0) return kByNameItem; return result; } int32 FindPanel::InitialAttrCount(const BNode* node) { if (node == NULL || node->InitCheck() != B_OK) return 1; int32 result; if (node->ReadAttr(kAttrQueryInitialNumAttrs, B_INT32_TYPE, 0, &result, sizeof(int32)) <= 0) return 1; return result; } static int32 SelectItemWithLabel(BMenu* menu, const char* label) { for (int32 index = menu->CountItems(); index-- > 0;) { BMenuItem* item = menu->ItemAt(index); if (strcmp(label, item->Label()) == 0) { item->SetMarked(true); return index; } } return -1; } void FindPanel::SaveWindowState(BNode* node, bool editTemplate) { ASSERT(node->InitCheck() == B_OK); BMenuItem* item = CurrentMimeType(); if (item) { BString label(item->Label()); node->WriteAttrString(kAttrQueryInitialMime, &label); } uint32 mode = Mode(); node->WriteAttr(kAttrQueryInitialMode, B_INT32_TYPE, 0, (int32*)&mode, sizeof(int32)); MoreOptionsStruct saveMoreOptions; saveMoreOptions.searchTrash = fSearchTrashCheck->Value() != 0; saveMoreOptions.temporary = fTemporaryCheck->Value() != 0; if (node->WriteAttr(kAttrQueryMoreOptions, B_RAW_TYPE, 0, &saveMoreOptions, sizeof(saveMoreOptions)) == sizeof(saveMoreOptions)) { node->RemoveAttr(kAttrQueryMoreOptionsForeign); } if (editTemplate) { if (UserSpecifiedName()) { BString name(UserSpecifiedName()); node->WriteAttrString(kAttrQueryTemplateName, &name); } } switch (Mode()) { case kByAttributeItem: { BMessage message; int32 count = fAttrGrid->CountRows(); node->WriteAttr(kAttrQueryInitialNumAttrs, B_INT32_TYPE, 0, &count, sizeof(int32)); for (int32 index = 0; index < count; index++) SaveAttrState(&message, index); ssize_t size = message.FlattenedSize(); if (size > 0) { char* buffer = new char[(size_t)size]; status_t result = message.Flatten(buffer, size); if (result == B_OK) { node->WriteAttr(kAttrQueryInitialAttrs, B_MESSAGE_TYPE, 0, buffer, (size_t)size); } delete[] buffer; } break; } case kByNameItem: case kByFormulaItem: { BTextControl* textControl = dynamic_cast( FindView("TextControl")); ASSERT(textControl != NULL); if (textControl != NULL) { BString formula(textControl->Text()); node->WriteAttrString(kAttrQueryInitialString, &formula); } break; } } } void FindPanel::SwitchToTemplate(const BNode* node) { SwitchMode(InitialMode(node)); // update the menu to correspond to the mode MarkNamedMenuItem(fSearchModeMenu, InitialMode(node), true); if (Mode() == (int32)kByAttributeItem) { RemoveByAttributeItems(); AddByAttributeItems(node); } RestoreWindowState(node); } void FindPanel::RestoreMimeTypeMenuSelection(const BNode* node) { if (Mode() == (int32)kByFormulaItem || node == NULL || node->InitCheck() != B_OK) { return; } BString buffer; if (node->ReadAttrString(kAttrQueryInitialMime, &buffer) == B_OK) SetCurrentMimeType(buffer.String()); } void FindPanel::RestoreWindowState(const BNode* node) { fMode = InitialMode(node); if (node == NULL || node->InitCheck() != B_OK) return; ShowOrHideMimeTypeMenu(); RestoreMimeTypeMenuSelection(node); MoreOptionsStruct saveMoreOptions; bool storesMoreOptions = ReadAttr(node, kAttrQueryMoreOptions, kAttrQueryMoreOptionsForeign, B_RAW_TYPE, 0, &saveMoreOptions, sizeof(saveMoreOptions), &MoreOptionsStruct::EndianSwap) != kReadAttrFailed; if (storesMoreOptions) { // need to sanitize to true or false here, could have picked // up garbage from attributes saveMoreOptions.showMoreOptions = true; // Now unused fSearchTrashCheck->SetValue(saveMoreOptions.searchTrash); fTemporaryCheck->SetValue(saveMoreOptions.temporary); fQueryName->SetModificationMessage(NULL); FindWindow* findWindow = dynamic_cast(Window()); if (findWindow != NULL) FillCurrentQueryName(fQueryName, findWindow); // set modification message after checking the temporary check box, // and filling out the text control so that we do not always trigger // clearing of the temporary check box. fQueryName->SetModificationMessage( new BMessage(kNameModifiedMessage)); } // get volumes to perform query on bool searchAllVolumes = true; attr_info info; if (node->GetAttrInfo(kAttrQueryVolume, &info) == B_OK) { char* buffer = new char[info.size]; if (node->ReadAttr(kAttrQueryVolume, B_MESSAGE_TYPE, 0, buffer, (size_t)info.size) == info.size) { BMessage message; if (message.Unflatten(buffer) == B_OK) { for (int32 index = 0; ;index++) { ASSERT(index < 100); BVolume volume; // match a volume with the info embedded in // the message status_t result = MatchArchivedVolume(&volume, &message, index); if (result == B_OK) { char name[256]; volume.GetName(name); SelectItemWithLabel(fVolMenu, name); searchAllVolumes = false; } else if (result != B_DEV_BAD_DRIVE_NUM) // if B_DEV_BAD_DRIVE_NUM, the volume just isn't // mounted this time around, keep looking for more // if other error, bail break; } } } delete[] buffer; } // mark or unmark "All disks" fVolMenu->ItemAt(0)->SetMarked(searchAllVolumes); ShowVolumeMenuLabel(); switch (Mode()) { case kByAttributeItem: { int32 count = InitialAttrCount(node); attr_info info; if (node->GetAttrInfo(kAttrQueryInitialAttrs, &info) != B_OK) break; char* buffer = new char[info.size]; if (node->ReadAttr(kAttrQueryInitialAttrs, B_MESSAGE_TYPE, 0, buffer, (size_t)info.size) == info.size) { BMessage message; if (message.Unflatten(buffer) == B_OK) { for (int32 index = 0; index < count; index++) RestoreAttrState(message, index); } } delete[] buffer; break; } case kByNameItem: case kByFormulaItem: { BString buffer; if (node->ReadAttrString(kAttrQueryInitialString, &buffer) == B_OK) { BTextControl* textControl = dynamic_cast( FindView("TextControl")); ASSERT(textControl != NULL); if (textControl != NULL) textControl->SetText(buffer.String()); } break; } } // try to restore focus and possibly text selection BString focusedView; if (node->ReadAttrString("_trk/focusedView", &focusedView) == B_OK) { BView* view = FindView(focusedView.String()); if (view != NULL) { view->MakeFocus(); BTextControl* textControl = dynamic_cast(view); if (textControl != NULL && Mode() == kByFormulaItem) { int32 selStart = 0; int32 selEnd = INT32_MAX; node->ReadAttr("_trk/focusedSelStart", B_INT32_TYPE, 0, &selStart, sizeof(selStart)); node->ReadAttr("_trk/focusedSelEnd", B_INT32_TYPE, 0, &selEnd, sizeof(selEnd)); textControl->TextView()->Select(selStart, selEnd); } } } } void FindPanel::AddByAttributeItems(const BNode* node) { int32 numAttributes = InitialAttrCount(node); if (numAttributes < 1) numAttributes = 1; for (int32 index = 0; index < numAttributes; index ++) AddAttrRow(); } void FindPanel::AddByNameOrFormulaItems() { BBox* box = dynamic_cast(FindView("Box")); ASSERT(box != NULL); if (box == NULL) return; // reset layout BLayoutBuilder::Group<>(box, B_VERTICAL); BTextControl* textControl = new BTextControl("TextControl", "", "", NULL); textControl->SetDivider(0.0f); box->SetBorder(B_NO_BORDER); box->AddChild(textControl); textControl->MakeFocus(); } void FindPanel::RemoveAttrViewItems(bool removeGrid) { if (fAttrGrid == NULL) return; BView* view = fAttrGrid->View(); for (int32 index = view->CountChildren(); index > 0; index--) { BView* child = view->ChildAt(index - 1); child->RemoveSelf(); delete child; } if (removeGrid) { view->RemoveSelf(); delete view; fAttrGrid = NULL; } } void FindPanel::RemoveByAttributeItems() { RemoveAttrViewItems(); BView* view = FindView("add button"); if (view) { view->RemoveSelf(); delete view; } view = FindView("remove button"); if (view) { view->RemoveSelf(); delete view; } view = FindView("TextControl"); if (view) { view->RemoveSelf(); delete view; } } void FindPanel::ShowOrHideMimeTypeMenu() { BView* menuFieldSpacer = FindView("MimeTypeMenuSpacer"); BMenuField* menuField = dynamic_cast(FindView("MimeTypeMenu")); if (menuFieldSpacer == NULL || menuField == NULL) return; if (Mode() == (int32)kByFormulaItem && !menuField->IsHidden(this)) { BSize size = menuField->ExplicitMinSize(); menuField->Hide(); menuFieldSpacer->SetExplicitMinSize(size); menuFieldSpacer->SetExplicitMaxSize(size); if (menuFieldSpacer->IsHidden(this)) menuFieldSpacer->Show(); } else if (menuField->IsHidden(this)) { menuFieldSpacer->Hide(); menuField->Show(); } } void FindPanel::AddAttributeControls(int32 gridRow) { BPopUpMenu* menu = new BPopUpMenu("PopUp"); // add NAME attribute to popup BMenu* submenu = new BMenu(B_TRANSLATE("Name")); submenu->SetRadioMode(true); submenu->SetFont(be_plain_font); BMessage* message = new BMessage(kAttributeItemMain); message->AddString("name", "name"); message->AddInt32("type", B_STRING_TYPE); BMenuItem* item = new BMenuItem(submenu, message); menu->AddItem(item); for (int32 i = 0; i < 5; i++) { message = new BMessage(kAttributeItem); message->AddInt32("operator", operators[i]); submenu->AddItem(new BMenuItem( B_TRANSLATE_NOCOLLECT(operatorLabels[i]), message)); } // mark first items initially menu->ItemAt(0)->SetMarked(true); submenu->ItemAt(0)->SetMarked(true); // add SIZE attribute submenu = new BMenu(B_TRANSLATE("Size")); submenu->SetRadioMode(true); submenu->SetFont(be_plain_font); message = new BMessage(kAttributeItemMain); message->AddString("name", "size"); message->AddInt32("type", B_OFF_T_TYPE); item = new BMenuItem(submenu, message); menu->AddItem(item); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_GT); submenu->AddItem(new BMenuItem(B_TRANSLATE_NOCOLLECT(operatorLabels[5]), message)); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_LT); submenu->AddItem(new BMenuItem(B_TRANSLATE_NOCOLLECT(operatorLabels[6]), message)); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_EQ); submenu->AddItem(new BMenuItem(B_TRANSLATE_NOCOLLECT(operatorLabels[1]), message)); // add "modified" field submenu = new BMenu(B_TRANSLATE("Modified")); submenu->SetRadioMode(true); submenu->SetFont(be_plain_font); message = new BMessage(kAttributeItemMain); message->AddString("name", "last_modified"); message->AddInt32("type", B_TIME_TYPE); item = new BMenuItem(submenu, message); menu->AddItem(item); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_LT); submenu->AddItem(new BMenuItem(B_TRANSLATE_NOCOLLECT(operatorLabels[7]), message)); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_GT); submenu->AddItem(new BMenuItem(B_TRANSLATE_NOCOLLECT(operatorLabels[8]), message)); BMenuField* menuField = new BMenuField("MenuField", "", menu); menuField->SetDivider(0.0f); fAttrGrid->AddView(menuField, 0, gridRow); BStringView* stringView = new BStringView("", menu->FindMarked()->Submenu()->FindMarked()->Label()); BLayoutItem* layoutItem = fAttrGrid->AddView(stringView, 1, gridRow); layoutItem->SetExplicitAlignment(BAlignment(B_ALIGN_RIGHT, B_ALIGN_VERTICAL_UNSET)); BString title("TextEntry"); title << gridRow; BTextControl* textControl = new BTextControl(title.String(), "", "", NULL); textControl->SetDivider(0.0f); fAttrGrid->AddView(textControl, 2, gridRow); textControl->MakeFocus(); // target everything menu->SetTargetForItems(this); for (int32 index = menu->CountItems() - 1; index >= 0; index--) { BMenu* submenuAtIndex = menu->SubmenuAt(index); if (submenuAtIndex != NULL) submenuAtIndex->SetTargetForItems(this); } // populate mime popup AddMimeTypeAttrs(menu); } void FindPanel::RestoreAttrState(const BMessage& message, int32 index) { BMenuField* menuField = dynamic_cast(FindAttrView("MenuField", index)); if (menuField != NULL) { // decode menu selections BMenu* menu = menuField->Menu(); ASSERT(menu != NULL); AddMimeTypeAttrs(menu); const char* label; if (message.FindString("menuSelection", index, &label) == B_OK) { int32 itemIndex = SelectItemWithLabel(menu, label); if (itemIndex >= 0) { menu = menu->SubmenuAt(itemIndex); if (menu != NULL && message.FindString("subMenuSelection", index, &label) == B_OK) { SelectItemWithLabel(menu, label); } } } } // decode attribute text BString textEntryString = "TextEntry"; textEntryString << index; BTextControl* textControl = dynamic_cast( FindAttrView(textEntryString.String(), index)); ASSERT(textControl != NULL); const char* string; if (textControl != NULL && message.FindString("attrViewText", index, &string) == B_OK) { textControl->SetText(string); } int32 logicMenuSelectedIndex; if (message.FindInt32("logicalRelation", index, &logicMenuSelectedIndex) == B_OK) { BMenuField* field = dynamic_cast( FindAttrView("Logic", index)); if (field != NULL) { BMenu* fieldMenu = field->Menu(); if (fieldMenu != NULL) { BMenuItem* logicItem = fieldMenu->ItemAt(logicMenuSelectedIndex); if (logicItem != NULL) { logicItem->SetMarked(true); return; } } } AddLogicMenu(index, logicMenuSelectedIndex == 0); } } void FindPanel::SaveAttrState(BMessage* message, int32 index) { BMenu* menu = dynamic_cast(FindAttrView("MenuField", index)) ->Menu(); // encode main attribute menu selection BMenuItem* item = menu->FindMarked(); message->AddString("menuSelection", item ? item->Label() : ""); // encode submenu selection const char* label = ""; if (item) { BMenu* submenu = menu->SubmenuAt(menu->IndexOf(item)); if (submenu) { item = submenu->FindMarked(); if (item) label = item->Label(); } } message->AddString("subMenuSelection", label); // encode attribute text BString textEntryString = "TextEntry"; textEntryString << index; BTextControl* textControl = dynamic_cast(FindAttrView( textEntryString.String(), index)); ASSERT(textControl != NULL); if (textControl != NULL) message->AddString("attrViewText", textControl->Text()); BMenuField* field = dynamic_cast(FindAttrView("Logic", index)); if (field != NULL) { BMenu* fieldMenu = field->Menu(); if (fieldMenu != NULL) { BMenuItem* item = fieldMenu->FindMarked(); ASSERT(item != NULL); message->AddInt32("logicalRelation", item != NULL ? field->Menu()->IndexOf(item) : 0); } } } void FindPanel::AddLogicMenu(int32 index, bool selectAnd) { // add "AND/OR" menu BPopUpMenu* menu = new BPopUpMenu(""); BMessage* message = new BMessage(); message->AddInt32("combine", B_AND); BMenuItem* item = new BMenuItem(B_TRANSLATE("And"), message); menu->AddItem(item); if (selectAnd) item->SetMarked(true); message = new BMessage(); message->AddInt32("combine", B_OR); item = new BMenuItem(B_TRANSLATE("Or"), message); menu->AddItem(item); if (!selectAnd) item->SetMarked(true); menu->SetTargetForItems(this); BMenuField* menufield = new BMenuField("Logic", "", menu, B_WILL_DRAW); menufield->SetDivider(0.0f); ResizeMenuField(menufield); fAttrGrid->AddView(menufield, 3, index); } void FindPanel::RemoveLogicMenu(int32 index) { BMenuField* menufield = dynamic_cast(FindAttrView("Logic", index)); if (menufield) { menufield->RemoveSelf(); delete menufield; } } void FindPanel::AddAttributes(BMenu* menu, const BMimeType& mimeType) { // only add things to menu which have "user-visible" data BMessage attributeMessage; if (mimeType.GetAttrInfo(&attributeMessage) != B_OK) return; char desc[B_MIME_TYPE_LENGTH]; mimeType.GetShortDescription(desc); // go through each field in meta mime and add it to a menu for (int32 index = 0; ; index++) { const char* publicName; if (attributeMessage.FindString("attr:public_name", index, &publicName) != B_OK) { break; } if (!attributeMessage.FindBool("attr:viewable")) continue; const char* attributeName; if (attributeMessage.FindString("attr:name", index, &attributeName) != B_OK) { continue; } int32 type; if (attributeMessage.FindInt32("attr:type", index, &type) != B_OK) continue; BMenu* submenu = new BMenu(publicName); submenu->SetRadioMode(true); submenu->SetFont(be_plain_font); BMessage* message = new BMessage(kAttributeItemMain); message->AddString("name", attributeName); message->AddInt32("type", type); BMenuItem* item = new BMenuItem(submenu, message); menu->AddItem(item); menu->SetTargetForItems(this); switch (type) { case B_STRING_TYPE: message = new BMessage(kAttributeItem); message->AddInt32("operator", B_CONTAINS); submenu->AddItem(new BMenuItem(operatorLabels[0], message)); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_EQ); submenu->AddItem(new BMenuItem(operatorLabels[1], message)); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_NE); submenu->AddItem(new BMenuItem(operatorLabels[2], message)); submenu->SetTargetForItems(this); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_BEGINS_WITH); submenu->AddItem(new BMenuItem(operatorLabels[3], message)); submenu->SetTargetForItems(this); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_ENDS_WITH); submenu->AddItem(new BMenuItem(operatorLabels[4], message)); break; case B_BOOL_TYPE: case B_INT16_TYPE: case B_UINT8_TYPE: case B_INT8_TYPE: case B_UINT16_TYPE: case B_INT32_TYPE: case B_UINT32_TYPE: case B_INT64_TYPE: case B_UINT64_TYPE: case B_OFF_T_TYPE: case B_FLOAT_TYPE: case B_DOUBLE_TYPE: message = new BMessage(kAttributeItem); message->AddInt32("operator", B_EQ); submenu->AddItem(new BMenuItem(operatorLabels[1], message)); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_GT); submenu->AddItem(new BMenuItem(operatorLabels[5], message)); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_LT); submenu->AddItem(new BMenuItem(operatorLabels[6], message)); break; case B_TIME_TYPE: message = new BMessage(kAttributeItem); message->AddInt32("operator", B_LT); submenu->AddItem(new BMenuItem(operatorLabels[7], message)); message = new BMessage(kAttributeItem); message->AddInt32("operator", B_GT); submenu->AddItem(new BMenuItem(operatorLabels[8], message)); break; } submenu->SetTargetForItems(this); } } void FindPanel::AddMimeTypeAttrs(BMenu* menu) { const char* typeName; if (CurrentMimeType(&typeName) == NULL) return; BMimeType mimeType(typeName); if (!mimeType.IsInstalled()) return; if (!mimeType.IsSupertypeOnly()) { // add supertype attributes BMimeType supertype; mimeType.GetSupertype(&supertype); AddAttributes(menu, supertype); } AddAttributes(menu, mimeType); } void FindPanel::GetDefaultAttrName(BString& attrName, int32 row) const { BMenuItem* item = NULL; BMenuField* menuField = dynamic_cast(fAttrGrid->ItemAt(0, row)->View()); if (menuField != NULL && menuField->Menu() != NULL) item = menuField->Menu()->FindMarked(); if (item != NULL) attrName << item->Label(); else attrName << B_TRANSLATE("Name"); if (item != NULL && item->Submenu() != NULL) item = item->Submenu()->FindMarked(); else item = NULL; if (item != NULL) attrName << " " << item->Label() << " "; else attrName << " = "; BTextControl* textControl = dynamic_cast(fAttrGrid->ItemAt(2, row)->View()); if (textControl != NULL) attrName << textControl->Text(); } // #pragma mark - DeleteTransientQueriesTask::DeleteTransientQueriesTask() : state(kInitial), fWalker(NULL) { } DeleteTransientQueriesTask::~DeleteTransientQueriesTask() { delete fWalker; } bool DeleteTransientQueriesTask::DoSomeWork() { switch (state) { case kInitial: Initialize(); break; case kAllocatedWalker: case kTraversing: if (GetSome()) { PRINT(("transient query killer done\n")); return true; } break; case kError: return true; } return false; } void DeleteTransientQueriesTask::Initialize() { PRINT(("starting up transient query killer\n")); BPath path; status_t result = find_directory(B_USER_DIRECTORY, &path, false); if (result != B_OK) { state = kError; return; } fWalker = new BTrackerPrivate::TNodeWalker(path.Path()); state = kAllocatedWalker; } const int32 kBatchCount = 100; bool DeleteTransientQueriesTask::GetSome() { state = kTraversing; for (int32 count = kBatchCount; count > 0; count--) { entry_ref ref; if (fWalker->GetNextRef(&ref) != B_OK) { state = kError; return true; } Model model(&ref); if (model.IsQuery()) ProcessOneRef(&model); #if xDEBUG else PRINT(("transient query killer: %s not a query\n", model.Name())); #endif } return false; } const int32 kDaysToExpire = 7; static bool QueryOldEnough(Model* model) { // check if it is old and ready to be deleted time_t now = time(0); tm nowTimeData; tm fileModData; localtime_r(&now, &nowTimeData); localtime_r(&model->StatBuf()->st_ctime, &fileModData); if ((nowTimeData.tm_mday - fileModData.tm_mday) < kDaysToExpire && (nowTimeData.tm_mday - fileModData.tm_mday) > -kDaysToExpire) { PRINT(("query %s, not old enough\n", model->Name())); return false; } return true; } bool DeleteTransientQueriesTask::ProcessOneRef(Model* model) { BModelOpener opener(model); // is this a temporary query if (!MoreOptionsStruct::QueryTemporary(model->Node())) { PRINT(("query %s, not temporary\n", model->Name())); return false; } if (!QueryOldEnough(model)) return false; TTracker* tracker = dynamic_cast(be_app); ASSERT(tracker != NULL); // check that it is not showing if (tracker != NULL && tracker->EntryHasWindowOpen(model->EntryRef())) { PRINT(("query %s, showing, can't delete\n", model->Name())); return false; } PRINT(("query %s, old, temporary, not shownig - deleting\n", model->Name())); BEntry entry(model->EntryRef()); entry.Remove(); return true; } class DeleteTransientQueriesFunctor : public FunctionObjectWithResult { public: DeleteTransientQueriesFunctor(DeleteTransientQueriesTask* task) : task(task) {} virtual ~DeleteTransientQueriesFunctor() { delete task; } virtual void operator()() { result = task->DoSomeWork(); } private: DeleteTransientQueriesTask* task; }; void DeleteTransientQueriesTask::StartUpTransientQueryCleaner() { TTracker* tracker = dynamic_cast(be_app); ASSERT(tracker != NULL); if (tracker == NULL) return; // set up a task that wakes up when the machine is idle and starts // killing off old transient queries DeleteTransientQueriesFunctor* worker = new DeleteTransientQueriesFunctor(new DeleteTransientQueriesTask()); tracker->MainTaskLoop()->RunWhenIdle(worker, 30 * 60 * 1000000, // half an hour initial delay 5 * 60 * 1000000, // idle for five minutes 10 * 1000000); } // #pragma mark - RecentFindItemsMenu::RecentFindItemsMenu(const char* title, const BMessenger* target, uint32 what) : BMenu(title, B_ITEMS_IN_COLUMN), fTarget(*target), fWhat(what) { } void RecentFindItemsMenu::AttachedToWindow() { // re-populate the menu with fresh items for (int32 index = CountItems() - 1; index >= 0; index--) delete RemoveItem(index); FindPanel::AddRecentQueries(this, false, &fTarget, fWhat); BMenu::AttachedToWindow(); } #if !B_BEOS_VERSION_DANO _IMPEXP_TRACKER #endif BMenu* TrackerBuildRecentFindItemsMenu(const char* title) { BMessenger trackerMessenger(kTrackerSignature); return new RecentFindItemsMenu(title, &trackerMessenger, B_REFS_RECEIVED); } // #pragma mark - DraggableQueryIcon::DraggableQueryIcon(BRect frame, const char* name, const BMessage* message, BMessenger messenger, uint32 resizeFlags, uint32 flags) : DraggableIcon(frame, name, B_QUERY_MIMETYPE, B_LARGE_ICON, message, messenger, resizeFlags, flags) { } bool DraggableQueryIcon::DragStarted(BMessage* dragMessage) { // override to substitute the user-specified query name dragMessage->RemoveData("be:clip_name"); FindWindow* window = dynamic_cast(Window()); ASSERT(window != NULL); return window != NULL && dragMessage->AddString("be:clip_name", window->BackgroundView()->UserSpecifiedName() != NULL ? window->BackgroundView()->UserSpecifiedName() : B_TRANSLATE("New Query")) == B_OK; } // #pragma mark - MostUsedNames::MostUsedNames(const char* fileName, const char* directory, int32 maxCount) : fFileName(fileName), fDirectory(directory), fLoaded(false), fCount(maxCount) { } MostUsedNames::~MostUsedNames() { // only write back settings when we've been used if (!fLoaded) return; // write most used list to file BPath path; if (find_directory(B_USER_SETTINGS_DIRECTORY, &path, true) != B_OK || path.Append(fDirectory) != B_OK || path.Append(fFileName) != B_OK) { return; } BFile file(path.Path(), B_WRITE_ONLY | B_CREATE_FILE | B_ERASE_FILE); if (file.InitCheck() == B_OK) { for (int32 i = 0; i < fList.CountItems(); i++) { list_entry* entry = static_cast(fList.ItemAt(i)); char line[B_FILE_NAME_LENGTH + 5]; // limit upper bound to react more dynamically to changes if (--entry->count > 20) entry->count = 20; // if the item hasn't been chosen in a while, remove it // (but leave at least one item in the list) if (entry->count < -10 && i > 0) continue; sprintf(line, "%" B_PRId32 " %s\n", entry->count, entry->name); if (file.Write(line, strlen(line)) < B_OK) break; } } file.Unset(); // free data for (int32 i = fList.CountItems(); i-- > 0;) { list_entry* entry = static_cast(fList.ItemAt(i)); free(entry->name); delete entry; } } bool MostUsedNames::ObtainList(BList* list) { if (list == NULL) return false; if (!fLoaded) UpdateList(); fLock.Lock(); list->MakeEmpty(); for (int32 i = 0; i < fCount; i++) { list_entry* entry = static_cast(fList.ItemAt(i)); if (entry == NULL) return true; list->AddItem(entry->name); } return true; } void MostUsedNames::ReleaseList() { fLock.Unlock(); } void MostUsedNames::AddName(const char* name) { fLock.Lock(); if (!fLoaded) LoadList(); // remove last entry if there are more than // 2*fCount entries in the list list_entry* entry = NULL; if (fList.CountItems() > fCount * 2) { entry = static_cast( fList.RemoveItem(fList.CountItems() - 1)); // is this the name we want to add here? if (strcmp(name, entry->name)) { free(entry->name); delete entry; entry = NULL; } else fList.AddItem(entry); } if (entry == NULL) { for (int32 i = 0; (entry = static_cast(fList.ItemAt(i))) != NULL; i++) { if (strcmp(entry->name, name) == 0) break; } } if (entry == NULL) { entry = new list_entry; entry->name = strdup(name); entry->count = 1; fList.AddItem(entry); } else if (entry->count < 0) entry->count = 1; else entry->count++; fLock.Unlock(); UpdateList(); } int MostUsedNames::CompareNames(const void* a,const void* b) { list_entry* entryA = *(list_entry**)a; list_entry* entryB = *(list_entry**)b; if (entryA->count == entryB->count) return strcasecmp(entryA->name,entryB->name); return entryB->count - entryA->count; } void MostUsedNames::LoadList() { if (fLoaded) return; fLoaded = true; // load the most used names list BPath path; if (find_directory(B_USER_SETTINGS_DIRECTORY, &path, true) != B_OK || path.Append(fDirectory) != B_OK || path.Append(fFileName) != B_OK) { return; } FILE* file = fopen(path.Path(), "r"); if (file == NULL) return; char line[B_FILE_NAME_LENGTH + 5]; while (fgets(line, sizeof(line), file) != NULL) { int32 length = (int32)strlen(line) - 1; if (length >= 0 && line[length] == '\n') line[length] = '\0'; int32 count = atoi(line); char* name = strchr(line, ' '); if (name == NULL || *(++name) == '\0') continue; list_entry* entry = new list_entry; entry->name = strdup(name); entry->count = count; fList.AddItem(entry); } fclose(file); } void MostUsedNames::UpdateList() { AutoLock locker(fLock); if (!fLoaded) LoadList(); // sort list items fList.SortItems(MostUsedNames::CompareNames); } } // namespace BPrivate