widget/nsBaseFilePicker.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
   2  *
   3  * This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "nsCOMPtr.h"
   8 #include "nsPIDOMWindow.h"
   9 #include "nsIInterfaceRequestorUtils.h"
  10 #include "nsIWidget.h"
  11
  12 #include "nsIStringBundle.h"
  13 #include "nsString.h"
  14 #include "nsCOMArray.h"
  15 #include "nsIFile.h"
  16 #include "nsEnumeratorUtils.h"
  17 #include "mozilla/dom/Directory.h"
  18 #include "mozilla/dom/File.h"
  19 #include "mozilla/dom/Promise.h"
  20 #include "mozilla/dom/Element.h"
  21 #include "mozilla/dom/CanonicalBrowsingContext.h"
  22 #include "mozilla/Components.h"
  23 #include "mozilla/StaticPrefs_widget.h"
  24 #include "WidgetUtils.h"
  25 #include "nsSimpleEnumerator.h"
  26 #include "nsThreadUtils.h"
  27 #include "nsContentUtils.h"
  28
  29 #include "nsBaseFilePicker.h"
  30
  31 using namespace mozilla::widget;
  32 using namespace mozilla::dom;
  33 using mozilla::ErrorResult;
  34
  35 #define FILEPICKER_TITLES "chrome://global/locale/filepicker.properties"
  36 #define FILEPICKER_FILTERS "chrome://global/content/filepicker.properties"
  37
  38 namespace {
  39
  40 nsresult LocalFileToDirectoryOrBlob(nsPIDOMWindowInner* aWindow,
  41                                     bool aIsDirectory, nsIFile* aFile,
  42                                     nsISupports** aResult) {
  43   MOZ_ASSERT(aWindow);
  44
  45   if (aIsDirectory) {
  46 #ifdef DEBUG
  47     bool isDir;
  48     aFile->IsDirectory(&isDir);
  49     MOZ_ASSERT(isDir);
  50 #endif
  51
  52     RefPtr<Directory> directory = Directory::Create(aWindow->AsGlobal(), aFile);
  53     MOZ_ASSERT(directory);
  54
  55     directory.forget(aResult);
  56     return NS_OK;
  57   }
  58
  59   RefPtr<File> file = File::CreateFromFile(aWindow->AsGlobal(), aFile);
  60   if (NS_WARN_IF(!file)) {
  61     return NS_ERROR_FAILURE;
  62   }
  63
  64   file.forget(aResult);
  65   return NS_OK;
  66 }
  67
  68 }  // anonymous namespace
  69
  70 #ifndef XP_WIN
  71 /**
  72  * A runnable to dispatch from the main thread to the main thread to display
  73  * the file picker while letting the showAsync method return right away.
  74  *
  75  * Not needed on Windows, where nsFilePicker::Open() is fully async.
  76  */
  77 class nsBaseFilePicker::AsyncShowFilePicker : public mozilla::Runnable {
  78  public:
  79   AsyncShowFilePicker(nsBaseFilePicker* aFilePicker,
  80                       nsIFilePickerShownCallback* aCallback)
  81       : mozilla::Runnable("AsyncShowFilePicker"),
  82         mFilePicker(aFilePicker),
  83         mCallback(aCallback) {}
  84
  85   NS_IMETHOD Run() override {
  86     NS_ASSERTION(NS_IsMainThread(),
  87                  "AsyncShowFilePicker should be on the main thread!");
  88
  89     // It's possible that some widget implementations require GUI operations
  90     // to be on the main thread, so that's why we're not dispatching to another
  91     // thread and calling back to the main after it's done.
  92     nsIFilePicker::ResultCode result = nsIFilePicker::returnCancel;
  93     nsresult rv = mFilePicker->Show(&result);
  94     if (NS_FAILED(rv)) {
  95       NS_ERROR("FilePicker's Show() implementation failed!");
  96     }
  97
  98     if (mCallback) {
  99       mCallback->Done(result);
 100     }
 101     return NS_OK;
 102   }
 103
 104  private:
 105   RefPtr<nsBaseFilePicker> mFilePicker;
 106   RefPtr<nsIFilePickerShownCallback> mCallback;
 107 };
 108 #endif
 109
 110 class nsBaseFilePickerEnumerator : public nsSimpleEnumerator {
 111  public:
 112   nsBaseFilePickerEnumerator(nsPIDOMWindowOuter* aParent,
 113                              nsISimpleEnumerator* iterator,
 114                              nsIFilePicker::Mode aMode)
 115       : mIterator(iterator),
 116         mParent(aParent->GetCurrentInnerWindow()),
 117         mMode(aMode) {}
 118
 119   const nsID& DefaultInterface() override { return NS_GET_IID(nsIFile); }
 120
 121   NS_IMETHOD
 122   GetNext(nsISupports** aResult) override {
 123     nsCOMPtr<nsISupports> tmp;
 124     nsresult rv = mIterator->GetNext(getter_AddRefs(tmp));
 125     NS_ENSURE_SUCCESS(rv, rv);
 126
 127     if (!tmp) {
 128       return NS_OK;
 129     }
 130
 131     nsCOMPtr<nsIFile> localFile = do_QueryInterface(tmp);
 132     if (!localFile) {
 133       return NS_ERROR_FAILURE;
 134     }
 135
 136     if (!mParent) {
 137       return NS_ERROR_FAILURE;
 138     }
 139
 140     return LocalFileToDirectoryOrBlob(
 141         mParent, mMode == nsIFilePicker::modeGetFolder, localFile, aResult);
 142   }
 143
 144   NS_IMETHOD
 145   HasMoreElements(bool* aResult) override {
 146     return mIterator->HasMoreElements(aResult);
 147   }
 148
 149  private:
 150   nsCOMPtr<nsISimpleEnumerator> mIterator;
 151   nsCOMPtr<nsPIDOMWindowInner> mParent;
 152   nsIFilePicker::Mode mMode;
 153 };
 154
 155 nsBaseFilePicker::nsBaseFilePicker()
 156     : mAddToRecentDocs(true), mMode(nsIFilePicker::modeOpen) {}
 157
 158 nsBaseFilePicker::~nsBaseFilePicker() = default;
 159
 160 NS_IMETHODIMP nsBaseFilePicker::Init(BrowsingContext* aBrowsingContext,
 161                                      const nsAString& aTitle,
 162                                      nsIFilePicker::Mode aMode) {
 163   MOZ_ASSERT(XRE_IsParentProcess());
 164   MOZ_ASSERT(aBrowsingContext,
 165              "Null bc passed to filepicker, no file "
 166              "picker for you!");
 167
 168   nsCOMPtr<nsIWidget> widget =
 169       aBrowsingContext->Canonical()->GetParentProcessWidgetContaining();
 170   NS_ENSURE_TRUE(widget, NS_ERROR_FAILURE);
 171
 172   mBrowsingContext = aBrowsingContext;
 173   mMode = aMode;
 174   InitNative(widget, aTitle);
 175
 176   return NS_OK;
 177 }
 178
 179 NS_IMETHODIMP
 180 nsBaseFilePicker::IsModeSupported(nsIFilePicker::Mode aMode, JSContext* aCx,
 181                                   Promise** aPromise) {
 182   MOZ_ASSERT(aCx);
 183   MOZ_ASSERT(aPromise);
 184
 185   nsIGlobalObject* globalObject = xpc::CurrentNativeGlobal(aCx);
 186   if (NS_WARN_IF(!globalObject)) {
 187     return NS_ERROR_FAILURE;
 188   }
 189
 190   ErrorResult result;
 191   RefPtr<Promise> promise = Promise::Create(globalObject, result);
 192   if (NS_WARN_IF(result.Failed())) {
 193     return result.StealNSResult();
 194   }
 195
 196   promise->MaybeResolve(true);
 197   promise.forget(aPromise);
 198
 199   return NS_OK;
 200 }
 201
 202 #ifndef XP_WIN
 203 NS_IMETHODIMP
 204 nsBaseFilePicker::Open(nsIFilePickerShownCallback* aCallback) {
 205   if (MaybeBlockFilePicker(aCallback)) {
 206     return NS_OK;
 207   }
 208
 209   nsCOMPtr<nsIRunnable> filePickerEvent =
 210       new AsyncShowFilePicker(this, aCallback);
 211   return NS_DispatchToMainThread(filePickerEvent);
 212 }
 213 #endif
 214
 215 NS_IMETHODIMP
 216 nsBaseFilePicker::Close() {
 217   NS_WARNING("Unimplemented nsFilePicker::Close");
 218   return NS_OK;
 219 }
 220
 221 NS_IMETHODIMP
 222 nsBaseFilePicker::AppendFilters(int32_t aFilterMask) {
 223   nsCOMPtr<nsIStringBundleService> stringService =
 224       mozilla::components::StringBundle::Service();
 225   if (!stringService) return NS_ERROR_FAILURE;
 226
 227   nsCOMPtr<nsIStringBundle> titleBundle, filterBundle;
 228
 229   nsresult rv = stringService->CreateBundle(FILEPICKER_TITLES,
 230                                             getter_AddRefs(titleBundle));
 231   if (NS_FAILED(rv)) return NS_ERROR_FAILURE;
 232
 233   rv = stringService->CreateBundle(FILEPICKER_FILTERS,
 234                                    getter_AddRefs(filterBundle));
 235   if (NS_FAILED(rv)) return NS_ERROR_FAILURE;
 236
 237   nsAutoString title;
 238   nsAutoString filter;
 239
 240   if (aFilterMask & filterAll) {
 241     titleBundle->GetStringFromName("allTitle", title);
 242     filterBundle->GetStringFromName("allFilter", filter);
 243     AppendFilter(title, filter);
 244   }
 245   if (aFilterMask & filterHTML) {
 246     titleBundle->GetStringFromName("htmlTitle", title);
 247     filterBundle->GetStringFromName("htmlFilter", filter);
 248     AppendFilter(title, filter);
 249   }
 250   if (aFilterMask & filterText) {
 251     titleBundle->GetStringFromName("textTitle", title);
 252     filterBundle->GetStringFromName("textFilter", filter);
 253     AppendFilter(title, filter);
 254   }
 255   if (aFilterMask & filterImages) {
 256     titleBundle->GetStringFromName("imageTitle", title);
 257     filterBundle->GetStringFromName("imageFilter", filter);
 258     AppendFilter(title, filter);
 259   }
 260   if (aFilterMask & filterAudio) {
 261     titleBundle->GetStringFromName("audioTitle", title);
 262     filterBundle->GetStringFromName("audioFilter", filter);
 263     AppendFilter(title, filter);
 264   }
 265   if (aFilterMask & filterVideo) {
 266     titleBundle->GetStringFromName("videoTitle", title);
 267     filterBundle->GetStringFromName("videoFilter", filter);
 268     AppendFilter(title, filter);
 269   }
 270   if (aFilterMask & filterXML) {
 271     titleBundle->GetStringFromName("xmlTitle", title);
 272     filterBundle->GetStringFromName("xmlFilter", filter);
 273     AppendFilter(title, filter);
 274   }
 275   if (aFilterMask & filterXUL) {
 276     titleBundle->GetStringFromName("xulTitle", title);
 277     filterBundle->GetStringFromName("xulFilter", filter);
 278     AppendFilter(title, filter);
 279   }
 280   if (aFilterMask & filterApps) {
 281     titleBundle->GetStringFromName("appsTitle", title);
 282     // Pass the magic string "..apps" to the platform filepicker, which it
 283     // should recognize and do the correct platform behavior for.
 284     AppendFilter(title, u"..apps"_ns);
 285   }
 286   if (aFilterMask & filterPDF) {
 287     titleBundle->GetStringFromName("pdfTitle", title);
 288     filterBundle->GetStringFromName("pdfFilter", filter);
 289     AppendFilter(title, filter);
 290   }
 291   return NS_OK;
 292 }
 293
 294 NS_IMETHODIMP nsBaseFilePicker::AppendRawFilter(const nsAString& aFilter) {
 295   mRawFilters.AppendElement(aFilter);
 296   return NS_OK;
 297 }
 298
 299 NS_IMETHODIMP nsBaseFilePicker::GetCapture(
 300     nsIFilePicker::CaptureTarget* aCapture) {
 301   *aCapture = nsIFilePicker::CaptureTarget::captureNone;
 302   return NS_OK;
 303 }
 304
 305 NS_IMETHODIMP nsBaseFilePicker::SetCapture(
 306     nsIFilePicker::CaptureTarget aCapture) {
 307   return NS_OK;
 308 }
 309
 310 // Set the filter index
 311 NS_IMETHODIMP nsBaseFilePicker::GetFilterIndex(int32_t* aFilterIndex) {
 312   *aFilterIndex = 0;
 313   return NS_OK;
 314 }
 315
 316 NS_IMETHODIMP nsBaseFilePicker::SetFilterIndex(int32_t aFilterIndex) {
 317   return NS_OK;
 318 }
 319
 320 NS_IMETHODIMP nsBaseFilePicker::GetFiles(nsISimpleEnumerator** aFiles) {
 321   NS_ENSURE_ARG_POINTER(aFiles);
 322   nsCOMArray<nsIFile> files;
 323   nsresult rv;
 324
 325   // if we get into the base class, the platform
 326   // doesn't implement GetFiles() yet.
 327   // so we fake it.
 328   nsCOMPtr<nsIFile> file;
 329   rv = GetFile(getter_AddRefs(file));
 330   NS_ENSURE_SUCCESS(rv, rv);
 331
 332   files.AppendObject(file);
 333
 334   return NS_NewArrayEnumerator(aFiles, files, NS_GET_IID(nsIFile));
 335 }
 336
 337 // Set the display directory
 338 NS_IMETHODIMP nsBaseFilePicker::SetDisplayDirectory(nsIFile* aDirectory) {
 339   // if displaySpecialDirectory has been previously called, let's abort this
 340   // operation.
 341   if (!mDisplaySpecialDirectory.IsEmpty()) {
 342     return NS_OK;
 343   }
 344
 345   if (!aDirectory) {
 346     mDisplayDirectory = nullptr;
 347     return NS_OK;
 348   }
 349   nsCOMPtr<nsIFile> directory;
 350   nsresult rv = aDirectory->Clone(getter_AddRefs(directory));
 351   if (NS_FAILED(rv)) return rv;
 352   mDisplayDirectory = directory;
 353   return NS_OK;
 354 }
 355
 356 // Get the display directory
 357 NS_IMETHODIMP nsBaseFilePicker::GetDisplayDirectory(nsIFile** aDirectory) {
 358   *aDirectory = nullptr;
 359
 360   // if displaySpecialDirectory has been previously called, let's abort this
 361   // operation.
 362   if (!mDisplaySpecialDirectory.IsEmpty()) {
 363     return NS_OK;
 364   }
 365
 366   if (!mDisplayDirectory) return NS_OK;
 367   nsCOMPtr<nsIFile> directory;
 368   nsresult rv = mDisplayDirectory->Clone(getter_AddRefs(directory));
 369   if (NS_FAILED(rv)) {
 370     return rv;
 371   }
 372   directory.forget(aDirectory);
 373   return NS_OK;
 374 }
 375
 376 // Set the display special directory
 377 NS_IMETHODIMP nsBaseFilePicker::SetDisplaySpecialDirectory(
 378     const nsAString& aDirectory) {
 379   // if displayDirectory has been previously called, let's abort this operation.
 380   if (mDisplayDirectory && mDisplaySpecialDirectory.IsEmpty()) {
 381     return NS_OK;
 382   }
 383
 384   mDisplaySpecialDirectory = aDirectory;
 385   if (mDisplaySpecialDirectory.IsEmpty()) {
 386     mDisplayDirectory = nullptr;
 387     return NS_OK;
 388   }
 389
 390   return ResolveSpecialDirectory(aDirectory);
 391 }
 392
 393 bool nsBaseFilePicker::MaybeBlockFilePicker(
 394     nsIFilePickerShownCallback* aCallback) {
 395   if (!mozilla::StaticPrefs::widget_disable_file_pickers()) {
 396     return false;
 397   }
 398
 399   if (aCallback) {
 400     // File pickers are disabled, so we answer the callback with returnCancel.
 401     aCallback->Done(nsIFilePicker::returnCancel);
 402   }
 403   if (mBrowsingContext) {
 404     RefPtr<Element> topFrameElement = mBrowsingContext->GetTopFrameElement();
 405     if (topFrameElement) {
 406       // Dispatch an event that the frontend may use.
 407       nsContentUtils::DispatchEventOnlyToChrome(
 408           topFrameElement->OwnerDoc(), topFrameElement, u"FilePickerBlocked"_ns,
 409           mozilla::CanBubble::eYes, mozilla::Cancelable::eNo);
 410     }
 411   }
 412
 413   return true;
 414 }
 415
 416 nsresult nsBaseFilePicker::ResolveSpecialDirectory(
 417     const nsAString& aSpecialDirectory) {
 418   // Only perform special-directory name resolution in the parent process.
 419   // (Subclasses of `nsBaseFilePicker` used in other processes must override
 420   // this function.)
 421   MOZ_ASSERT(XRE_IsParentProcess());
 422   return NS_GetSpecialDirectory(NS_ConvertUTF16toUTF8(aSpecialDirectory).get(),
 423                                 getter_AddRefs(mDisplayDirectory));
 424 }
 425
 426 // Get the display special directory
 427 NS_IMETHODIMP nsBaseFilePicker::GetDisplaySpecialDirectory(
 428     nsAString& aDirectory) {
 429   aDirectory = mDisplaySpecialDirectory;
 430   return NS_OK;
 431 }
 432
 433 NS_IMETHODIMP
 434 nsBaseFilePicker::GetAddToRecentDocs(bool* aFlag) {
 435   *aFlag = mAddToRecentDocs;
 436   return NS_OK;
 437 }
 438
 439 NS_IMETHODIMP
 440 nsBaseFilePicker::SetAddToRecentDocs(bool aFlag) {
 441   mAddToRecentDocs = aFlag;
 442   return NS_OK;
 443 }
 444
 445 NS_IMETHODIMP
 446 nsBaseFilePicker::GetMode(nsIFilePicker::Mode* aMode) {
 447   *aMode = mMode;
 448   return NS_OK;
 449 }
 450
 451 NS_IMETHODIMP
 452 nsBaseFilePicker::SetOkButtonLabel(const nsAString& aLabel) {
 453   mOkButtonLabel = aLabel;
 454   return NS_OK;
 455 }
 456
 457 NS_IMETHODIMP
 458 nsBaseFilePicker::GetOkButtonLabel(nsAString& aLabel) {
 459   aLabel = mOkButtonLabel;
 460   return NS_OK;
 461 }
 462
 463 NS_IMETHODIMP
 464 nsBaseFilePicker::GetDomFileOrDirectory(nsISupports** aValue) {
 465   MOZ_ASSERT(XRE_IsParentProcess());
 466   NS_ENSURE_ARG_POINTER(mBrowsingContext);
 467   nsCOMPtr<nsIFile> localFile;
 468   nsresult rv = GetFile(getter_AddRefs(localFile));
 469   NS_ENSURE_SUCCESS(rv, rv);
 470
 471   if (!localFile) {
 472     *aValue = nullptr;
 473     return NS_OK;
 474   }
 475
 476   auto* innerParent =
 477       mBrowsingContext->GetDOMWindow()
 478           ? mBrowsingContext->GetDOMWindow()->GetCurrentInnerWindow()
 479           : nullptr;
 480
 481   if (!innerParent) {
 482     return NS_ERROR_FAILURE;
 483   }
 484
 485   return LocalFileToDirectoryOrBlob(
 486       innerParent, mMode == nsIFilePicker::modeGetFolder, localFile, aValue);
 487 }
 488
 489 NS_IMETHODIMP
 490 nsBaseFilePicker::GetDomFileOrDirectoryEnumerator(
 491     nsISimpleEnumerator** aValue) {
 492   nsCOMPtr<nsISimpleEnumerator> iter;
 493   MOZ_ASSERT(XRE_IsParentProcess());
 494   NS_ENSURE_ARG_POINTER(mBrowsingContext);
 495   nsresult rv = GetFiles(getter_AddRefs(iter));
 496   NS_ENSURE_SUCCESS(rv, rv);
 497
 498   auto* parent = mBrowsingContext->GetDOMWindow();
 499
 500   if (!parent) {
 501     return NS_ERROR_FAILURE;
 502   }
 503
 504   RefPtr<nsBaseFilePickerEnumerator> retIter =
 505       new nsBaseFilePickerEnumerator(parent, iter, mMode);
 506
 507   retIter.forget(aValue);
 508   return NS_OK;
 509 }