/* -*- Mode: C++; tab-width: 2; 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 "nsClipboardHelper.h" // basics #include "nsCOMPtr.h" #include "nsXPCOM.h" #include "nsISupportsPrimitives.h" #include "nsIServiceManager.h" // helpers #include "nsIClipboard.h" #include "nsIDocument.h" #include "nsIDOMDocument.h" #include "nsITransferable.h" #include "nsReadableUtils.h" NS_IMPL_ISUPPORTS(nsClipboardHelper, nsIClipboardHelper) /***************************************************************************** * nsClipboardHelper ctor / dtor *****************************************************************************/ nsClipboardHelper::nsClipboardHelper() { } nsClipboardHelper::~nsClipboardHelper() { // no members, nothing to destroy } /***************************************************************************** * nsIClipboardHelper methods *****************************************************************************/ NS_IMETHODIMP nsClipboardHelper::CopyStringToClipboard(const nsAString& aString, int32_t aClipboardID) { nsresult rv; // get the clipboard nsCOMPtr<nsIClipboard> clipboard(do_GetService("@mozilla.org/widget/clipboard;1", &rv)); NS_ENSURE_SUCCESS(rv, rv); NS_ENSURE_TRUE(clipboard, NS_ERROR_FAILURE); bool clipboardSupported; // don't go any further if they're asking for the selection // clipboard on a platform which doesn't support it (i.e., unix) if (nsIClipboard::kSelectionClipboard == aClipboardID) { rv = clipboard->SupportsSelectionClipboard(&clipboardSupported); NS_ENSURE_SUCCESS(rv, rv); if (!clipboardSupported) return NS_ERROR_FAILURE; } // don't go any further if they're asking for the find clipboard on a platform // which doesn't support it (i.e., non-osx) if (nsIClipboard::kFindClipboard == aClipboardID) { rv = clipboard->SupportsFindClipboard(&clipboardSupported); NS_ENSURE_SUCCESS(rv, rv); if (!clipboardSupported) return NS_ERROR_FAILURE; } // create a transferable for putting data on the clipboard nsCOMPtr<nsITransferable> trans(do_CreateInstance("@mozilla.org/widget/transferable;1", &rv)); NS_ENSURE_SUCCESS(rv, rv); NS_ENSURE_TRUE(trans, NS_ERROR_FAILURE); trans->Init(nullptr); // Add the text data flavor to the transferable rv = trans->AddDataFlavor(kUnicodeMime); NS_ENSURE_SUCCESS(rv, rv); // get wStrings to hold clip data nsCOMPtr<nsISupportsString> data(do_CreateInstance("@mozilla.org/supports-string;1", &rv)); NS_ENSURE_SUCCESS(rv, rv); NS_ENSURE_TRUE(data, NS_ERROR_FAILURE); // populate the string rv = data->SetData(aString); NS_ENSURE_SUCCESS(rv, rv); // qi the data object an |nsISupports| so that when the transferable holds // onto it, it will addref the correct interface. nsCOMPtr<nsISupports> genericData(do_QueryInterface(data, &rv)); NS_ENSURE_SUCCESS(rv, rv); NS_ENSURE_TRUE(genericData, NS_ERROR_FAILURE); // set the transfer data rv = trans->SetTransferData(kUnicodeMime, genericData, aString.Length() * 2); NS_ENSURE_SUCCESS(rv, rv); // put the transferable on the clipboard rv = clipboard->SetData(trans, nullptr, aClipboardID); NS_ENSURE_SUCCESS(rv, rv); return NS_OK; } NS_IMETHODIMP nsClipboardHelper::CopyString(const nsAString& aString) { nsresult rv; // copy to the global clipboard. it's bad if this fails in any way. rv = CopyStringToClipboard(aString, nsIClipboard::kGlobalClipboard); NS_ENSURE_SUCCESS(rv, rv); // unix also needs us to copy to the selection clipboard. this will // fail in CopyStringToClipboard if we're not on a platform that // supports the selection clipboard. (this could have been #ifdef // XP_UNIX, but using the SupportsSelectionClipboard call is the // more correct thing to do. // // if this fails in any way other than "not being unix", we'll get // the assertion we need in CopyStringToClipboard, and we needn't // assert again here. CopyStringToClipboard(aString, nsIClipboard::kSelectionClipboard); return NS_OK; }