/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- * * This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ #include "nsFilePickerProxy.h" #include "nsComponentManagerUtils.h" #include "nsIFile.h" #include "mozilla/dom/Directory.h" #include "mozilla/dom/File.h" #include "mozilla/dom/TabChild.h" #include "mozilla/dom/ipc/BlobChild.h" using namespace mozilla::dom; NS_IMPL_ISUPPORTS(nsFilePickerProxy, nsIFilePicker) nsFilePickerProxy::nsFilePickerProxy() : mSelectedType(0) { } nsFilePickerProxy::~nsFilePickerProxy() { } NS_IMETHODIMP nsFilePickerProxy::Init(mozIDOMWindowProxy* aParent, const nsAString& aTitle, int16_t aMode) { TabChild* tabChild = TabChild::GetFrom(aParent); if (!tabChild) { return NS_ERROR_FAILURE; } mParent = nsPIDOMWindowOuter::From(aParent); mMode = aMode; NS_ADDREF_THIS(); tabChild->SendPFilePickerConstructor(this, nsString(aTitle), aMode); return NS_OK; } void nsFilePickerProxy::InitNative(nsIWidget* aParent, const nsAString& aTitle) { } NS_IMETHODIMP nsFilePickerProxy::AppendFilter(const nsAString& aTitle, const nsAString& aFilter) { mFilterNames.AppendElement(aTitle); mFilters.AppendElement(aFilter); return NS_OK; } NS_IMETHODIMP nsFilePickerProxy::GetDefaultString(nsAString& aDefaultString) { aDefaultString = mDefault; return NS_OK; } NS_IMETHODIMP nsFilePickerProxy::SetDefaultString(const nsAString& aDefaultString) { mDefault = aDefaultString; return NS_OK; } NS_IMETHODIMP nsFilePickerProxy::GetDefaultExtension(nsAString& aDefaultExtension) { aDefaultExtension = mDefaultExtension; return NS_OK; } NS_IMETHODIMP nsFilePickerProxy::SetDefaultExtension(const nsAString& aDefaultExtension) { mDefaultExtension = aDefaultExtension; return NS_OK; } NS_IMETHODIMP nsFilePickerProxy::GetFilterIndex(int32_t* aFilterIndex) { *aFilterIndex = mSelectedType; return NS_OK; } NS_IMETHODIMP nsFilePickerProxy::SetFilterIndex(int32_t aFilterIndex) { mSelectedType = aFilterIndex; return NS_OK; } NS_IMETHODIMP nsFilePickerProxy::GetFile(nsIFile** aFile) { MOZ_ASSERT(false, "GetFile is unimplemented; use GetDomFileOrDirectory"); return NS_ERROR_FAILURE; } NS_IMETHODIMP nsFilePickerProxy::GetFileURL(nsIURI** aFileURL) { MOZ_ASSERT(false, "GetFileURL is unimplemented; use GetDomFileOrDirectory"); return NS_ERROR_FAILURE; } NS_IMETHODIMP nsFilePickerProxy::GetFiles(nsISimpleEnumerator** aFiles) { MOZ_ASSERT(false, "GetFiles is unimplemented; use GetDomFileOrDirectoryEnumerator"); return NS_ERROR_FAILURE; } NS_IMETHODIMP nsFilePickerProxy::Show(int16_t* aReturn) { MOZ_ASSERT(false, "Show is unimplemented; use Open"); return NS_ERROR_NOT_IMPLEMENTED; } NS_IMETHODIMP nsFilePickerProxy::Open(nsIFilePickerShownCallback* aCallback) { mCallback = aCallback; nsString displayDirectory; if (mDisplayDirectory) { mDisplayDirectory->GetPath(displayDirectory); } SendOpen(mSelectedType, mAddToRecentDocs, mDefault, mDefaultExtension, mFilters, mFilterNames, displayDirectory, mOkButtonLabel); return NS_OK; } bool nsFilePickerProxy::Recv__delete__(const MaybeInputData& aData, const int16_t& aResult) { if (aData.type() == MaybeInputData::TInputBlobs) { const InfallibleTArray<PBlobChild*>& blobs = aData.get_InputBlobs().blobsChild(); for (uint32_t i = 0; i < blobs.Length(); ++i) { BlobChild* actor = static_cast<BlobChild*>(blobs[i]); RefPtr<BlobImpl> blobImpl = actor->GetBlobImpl(); NS_ENSURE_TRUE(blobImpl, true); if (!blobImpl->IsFile()) { return true; } nsPIDOMWindowInner* inner = mParent ? mParent->GetCurrentInnerWindow() : nullptr; RefPtr<File> file = File::Create(inner, blobImpl); MOZ_ASSERT(file); OwningFileOrDirectory* element = mFilesOrDirectories.AppendElement(); element->SetAsFile() = file; } } else if (aData.type() == MaybeInputData::TInputDirectory) { nsCOMPtr<nsIFile> file; NS_ConvertUTF16toUTF8 path(aData.get_InputDirectory().directoryPath()); nsresult rv = NS_NewNativeLocalFile(path, true, getter_AddRefs(file)); if (NS_WARN_IF(NS_FAILED(rv))) { return true; } RefPtr<Directory> directory = Directory::Create(mParent->GetCurrentInnerWindow(), file); MOZ_ASSERT(directory); OwningFileOrDirectory* element = mFilesOrDirectories.AppendElement(); element->SetAsDirectory() = directory; } if (mCallback) { mCallback->Done(aResult); mCallback = nullptr; } return true; } NS_IMETHODIMP nsFilePickerProxy::GetDomFileOrDirectory(nsISupports** aValue) { *aValue = nullptr; if (mFilesOrDirectories.IsEmpty()) { return NS_OK; } MOZ_ASSERT(mFilesOrDirectories.Length() == 1); if (mFilesOrDirectories[0].IsFile()) { nsCOMPtr<nsIDOMBlob> blob = mFilesOrDirectories[0].GetAsFile().get(); blob.forget(aValue); return NS_OK; } MOZ_ASSERT(mFilesOrDirectories[0].IsDirectory()); RefPtr<Directory> directory = mFilesOrDirectories[0].GetAsDirectory(); directory.forget(aValue); return NS_OK; } namespace { class SimpleEnumerator final : public nsISimpleEnumerator { public: NS_DECL_ISUPPORTS explicit SimpleEnumerator(const nsTArray<OwningFileOrDirectory>& aFilesOrDirectories) : mFilesOrDirectories(aFilesOrDirectories) , mIndex(0) {} NS_IMETHOD HasMoreElements(bool* aRetvalue) override { MOZ_ASSERT(aRetvalue); *aRetvalue = mIndex < mFilesOrDirectories.Length(); return NS_OK; } NS_IMETHOD GetNext(nsISupports** aValue) override { NS_ENSURE_TRUE(mIndex < mFilesOrDirectories.Length(), NS_ERROR_FAILURE); uint32_t index = mIndex++; if (mFilesOrDirectories[index].IsFile()) { nsCOMPtr<nsIDOMBlob> blob = mFilesOrDirectories[index].GetAsFile().get(); blob.forget(aValue); return NS_OK; } MOZ_ASSERT(mFilesOrDirectories[index].IsDirectory()); RefPtr<Directory> directory = mFilesOrDirectories[index].GetAsDirectory(); directory.forget(aValue); return NS_OK; } private: ~SimpleEnumerator() {} nsTArray<mozilla::dom::OwningFileOrDirectory> mFilesOrDirectories; uint32_t mIndex; }; NS_IMPL_ISUPPORTS(SimpleEnumerator, nsISimpleEnumerator) } // namespace NS_IMETHODIMP nsFilePickerProxy::GetDomFileOrDirectoryEnumerator(nsISimpleEnumerator** aDomfiles) { RefPtr<SimpleEnumerator> enumerator = new SimpleEnumerator(mFilesOrDirectories); enumerator.forget(aDomfiles); return NS_OK; }