dom/file/ArchiveEvent.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=2 et sw=2 tw=80: */
   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 file,
   5  * You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "ArchiveEvent.h"
   8
   9 #include "nsContentUtils.h"
  10 #include "nsCExternalHandlerService.h"
  11 #include "nsProxyRelease.h"
  12
  13 USING_FILE_NAMESPACE
  14
  15 NS_IMPL_THREADSAFE_ISUPPORTS0(ArchiveItem)
  16
  17 ArchiveItem::ArchiveItem()
  18 {
  19   MOZ_COUNT_CTOR(ArchiveItem);
  20 }
  21
  22 ArchiveItem::~ArchiveItem()
  23 {
  24   MOZ_COUNT_DTOR(ArchiveItem);
  25 }
  26
  27
  28 nsCString
  29 ArchiveItem::GetType()
  30 {
  31   if (mType.IsEmpty()) {
  32     return NS_LITERAL_CSTRING("binary/octet-stream");
  33   }
  34
  35   return mType;
  36 }
  37
  38 void
  39 ArchiveItem::SetType(const nsCString& aType)
  40 {
  41   mType = aType;
  42 }
  43
  44 ArchiveReaderEvent::ArchiveReaderEvent(ArchiveReader* aArchiveReader)
  45 : mArchiveReader(aArchiveReader)
  46 {
  47   MOZ_COUNT_CTOR(ArchiveReaderEvent);
  48 }
  49
  50 ArchiveReaderEvent::~ArchiveReaderEvent()
  51 {
  52   if (!NS_IsMainThread()) {
  53     nsIMIMEService* mimeService;
  54     mMimeService.forget(&mimeService);
  55
  56     if (mimeService) {
  57       nsCOMPtr<nsIThread> mainThread = do_GetMainThread();
  58       NS_WARN_IF_FALSE(mainThread, "Couldn't get the main thread! Leaking!");
  59
  60       if (mainThread) {
  61         NS_ProxyRelease(mainThread, mimeService);
  62       }
  63     }
  64   }
  65
  66   MOZ_COUNT_DTOR(ArchiveReaderEvent);
  67 }
  68
  69 // From the filename to the mimetype:
  70 nsresult
  71 ArchiveReaderEvent::GetType(nsCString& aExt,
  72                             nsCString& aMimeType)
  73 {
  74   MOZ_ASSERT(NS_IsMainThread());
  75
  76   nsresult rv;
  77
  78   if (mMimeService.get() == nullptr) {
  79     mMimeService = do_GetService(NS_MIMESERVICE_CONTRACTID, &rv);
  80     NS_ENSURE_SUCCESS(rv, rv);
  81   }
  82
  83   rv = mMimeService->GetTypeFromExtension(aExt, aMimeType);
  84   NS_ENSURE_SUCCESS(rv, rv);
  85
  86   return NS_OK;
  87 }
  88
  89 NS_IMETHODIMP
  90 ArchiveReaderEvent::Run()
  91 {
  92   return Exec();
  93 }
  94
  95 nsresult
  96 ArchiveReaderEvent::RunShare(nsresult aStatus)
  97 {
  98   mStatus = aStatus;
  99
 100   nsCOMPtr<nsIRunnable> event = NS_NewRunnableMethod(this, &ArchiveReaderEvent::ShareMainThread);
 101   NS_DispatchToMainThread(event, NS_DISPATCH_NORMAL);
 102
 103   return NS_OK;
 104 }
 105
 106 void
 107 ArchiveReaderEvent::ShareMainThread()
 108 {
 109   nsTArray<nsCOMPtr<nsIDOMFile> > fileList;
 110
 111   if (!NS_FAILED(mStatus)) {
 112     // This extra step must run in the main thread:
 113     for (uint32_t index = 0; index < mFileList.Length(); ++index) {
 114       nsRefPtr<ArchiveItem> item = mFileList[index];
 115
 116       nsString tmp;
 117       nsresult rv = item->GetFilename(tmp);
 118       nsCString filename = NS_ConvertUTF16toUTF8(tmp);
 119       if (NS_FAILED(rv)) {
 120         continue;
 121       }
 122
 123       int32_t offset = filename.RFindChar('.');
 124       if (offset != kNotFound) {
 125         filename.Cut(0, offset + 1);
 126
 127         // Just to be sure, if something goes wrong, the mimetype is an empty string:
 128         nsCString type;
 129         if (NS_SUCCEEDED(GetType(filename, type))) {
 130           item->SetType(type);
 131         }
 132       }
 133
 134       // This is a nsDOMFile:
 135       nsRefPtr<nsIDOMFile> file = item->File(mArchiveReader);
 136       if (file) {
 137         fileList.AppendElement(file);
 138       }
 139     }
 140   }
 141
 142   mArchiveReader->Ready(fileList, mStatus);
 143 }