1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
/*
* Copyright 2010 Google Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.google.android.testing.mocking;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;
import java.net.URL;
import java.net.URLClassLoader;
import java.text.SimpleDateFormat;
import java.util.Date;
import javax.annotation.processing.ProcessingEnvironment;
import javax.tools.Diagnostic.Kind;
/**
* @author swoodward@google.com (Stephen Woodward)
*
*/
class ProcessorLogger {
private final OutputStream logFile;
private final ProcessingEnvironment processingEnv;
ProcessorLogger(OutputStream logFile, ProcessingEnvironment processingEnv) {
this.logFile = logFile;
this.processingEnv = processingEnv;
}
ProcessorLogger(String logFileName, ProcessingEnvironment processingEnv) {
this.logFile = openLogFile(logFileName);
this.processingEnv = processingEnv;
}
void reportClasspathError(String clazz, Throwable e) {
printMessage(Kind.ERROR, "Could not find " + clazz);
printMessage(Kind.ERROR, e);
printMessage(Kind.NOTE, "Known Classpath: ");
URL[] allUrls = ((URLClassLoader) this.getClass().getClassLoader()).getURLs();
for (URL url : allUrls) {
printMessage(Kind.NOTE, url.toString());
}
}
void printMessage(Kind kind, String message) {
processingEnv.getMessager().printMessage(kind, message);
if (logFile != null) {
try {
logFile.write((SimpleDateFormat.getDateTimeInstance().format(new Date()) + " - "
+ kind.toString() + " : " + message + "\n").getBytes());
} catch (IOException e) {
// That's unfortunate, but not much to do about it.
processingEnv.getMessager().printMessage(Kind.WARNING,
"IOException logging to file" + e.toString());
}
}
}
void printMessage(Kind kind, Throwable e) {
ByteArrayOutputStream stackTraceByteStream = new ByteArrayOutputStream();
PrintStream stackTraceStream = new PrintStream(stackTraceByteStream);
e.printStackTrace(stackTraceStream);
printMessage(kind, stackTraceByteStream.toString());
}
FileOutputStream openLogFile(String logFileName) {
try {
if (logFileName != null) {
File log = new File(logFileName);
if (!log.exists() && log.getParentFile() != null) {
log.getParentFile().mkdirs();
}
return new FileOutputStream(log, true);
}
} catch (FileNotFoundException e) {
printMessage(Kind.WARNING, e);
}
return null;
}
void close() {
if (logFile != null) {
try {
logFile.close();
} catch (IOException e) {
// That's ok
}
}
}
}
|