toolkit/xre/nsConsoleWriter.cpp

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3  *
   4  * The contents of this file are subject to the Mozilla Public License Version
   5  * 1.1 (the "License"); you may not use this file except in compliance with
   6  * the License. You may obtain a copy of the License at
   7  * http://www.mozilla.org/MPL/
   8  *
   9  * Software distributed under the License is distributed on an "AS IS" basis,
  10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11  * for the specific language governing rights and limitations under the
  12  * License.
  13  *
  14  * The Original Code is the Mozilla Toolkit.
  15  *
  16  * The Initial Developer of the Original Code is
  17  * Benjamin Smedberg <benjamin@smedbergs.us>
  18  * Portions created by the Initial Developer are Copyright (C) 2005
  19  * the Initial Developer. All Rights Reserved.
  20  *
  21  * Contributor(s):
  22  *
  23  * Alternatively, the contents of this file may be used under the terms of
  24  * either the GNU General Public License Version 2 or later (the "GPL"), or
  25  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  26  * in which case the provisions of the GPL or the LGPL are applicable instead
  27  * of those above. If you wish to allow use of your version of this file only
  28  * under the terms of either the GPL or the LGPL, and not to allow others to
  29  * use your version of this file under the terms of the MPL, indicate your
  30  * decision by deleting the provisions above and replace them with the notice
  31  * and other provisions required by the GPL or the LGPL. If you do not delete
  32  * the provisions above, a recipient may use your version of this file under
  33  * the terms of any one of the MPL, the GPL or the LGPL.
  34  *
  35  * ***** END LICENSE BLOCK ***** */
  36
  37 #ifdef NO_NSPR_10_SUPPORT
  38 #undef NO_NSPR_10_SUPPORT
  39 #endif
  40
  41 #include "nsAppRunner.h"
  42
  43 #include "prio.h"
  44 #include "prprf.h"
  45 #include "prtime.h"
  46 #include "prenv.h"
  47
  48 #include "nsCRT.h"
  49 #include "nsNativeCharsetUtils.h"
  50 #include "nsString.h"
  51 #include "nsXREDirProvider.h"
  52 #include "nsXULAppAPI.h"
  53
  54 #include "nsIConsoleService.h"
  55 #include "nsIConsoleMessage.h"
  56
  57 void
  58 WriteConsoleLog()
  59 {
  60   nsresult rv;
  61
  62   nsCOMPtr<nsILocalFile> lfile;
  63
  64   char* logFileEnv = PR_GetEnv("XRE_CONSOLE_LOG");
  65   if (logFileEnv && *logFileEnv) {
  66     rv = XRE_GetFileFromPath(logFileEnv, getter_AddRefs(lfile));
  67     if (NS_FAILED(rv))
  68       return;
  69   }
  70   else {
  71     if (!gLogConsoleErrors)
  72       return;
  73
  74     rv = gDirServiceProvider->GetUserAppDataDirectory(getter_AddRefs(lfile));
  75     if (NS_FAILED(rv))
  76       return;
  77
  78     lfile->AppendNative(NS_LITERAL_CSTRING("console.log"));
  79   }
  80
  81   PRFileDesc *file;
  82   rv = lfile->OpenNSPRFileDesc(PR_WRONLY | PR_APPEND | PR_CREATE_FILE,
  83                                0660, &file);
  84   if (NS_FAILED(rv))
  85     return;
  86
  87   nsCOMPtr<nsIConsoleService> csrv
  88     (do_GetService(NS_CONSOLESERVICE_CONTRACTID));
  89   if (!csrv) {
  90     PR_Close(file);
  91     return;
  92   }
  93
  94   nsIConsoleMessage** messages;
  95   PRUint32 mcount;
  96
  97   rv = csrv->GetMessageArray(&messages, &mcount);
  98   if (NS_FAILED(rv)) {
  99     PR_Close(file);
 100     return;
 101   }
 102
 103   if (mcount) {
 104     PRExplodedTime etime;
 105     PR_ExplodeTime(PR_Now(), PR_LocalTimeParameters, &etime);
 106     char datetime[512];
 107     PR_FormatTimeUSEnglish(datetime, sizeof(datetime),
 108                            "%Y-%m-%d %H:%M:%S", &etime);
 109
 110     PR_fprintf(file, NS_LINEBREAK
 111                      "*** Console log: %s ***" NS_LINEBREAK,
 112                datetime);
 113   }
 114
 115   // From this point on, we have to release all the messages, and free
 116   // the memory allocated for the messages array. XPCOM arrays suck.
 117
 118   nsXPIDLString msg;
 119   nsCAutoString nativemsg;
 120
 121   for (PRUint32 i = 0; i < mcount; ++i) {
 122     rv = messages[i]->GetMessageMoz(getter_Copies(msg));
 123     if (NS_SUCCEEDED(rv)) {
 124       NS_CopyUnicodeToNative(msg, nativemsg);
 125       PR_fprintf(file, "%s" NS_LINEBREAK, nativemsg.get());
 126     }
 127     NS_IF_RELEASE(messages[i]);
 128   }
 129
 130   PR_Close(file);
 131   NS_Free(messages);
 132 }