ConsoleChannel.cpp 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274
  1. //
  2. // ConsoleChannel.cpp
  3. //
  4. // $Id: //poco/1.4/Foundation/src/ConsoleChannel.cpp#2 $
  5. //
  6. // Library: Foundation
  7. // Package: Logging
  8. // Module: ConsoleChannel
  9. //
  10. // Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH.
  11. // and Contributors.
  12. //
  13. // SPDX-License-Identifier: BSL-1.0
  14. //
  15. #include "Poco/ConsoleChannel.h"
  16. #include "Poco/Message.h"
  17. #include "Poco/String.h"
  18. #include "Poco/Exception.h"
  19. #include <iostream>
  20. namespace Poco {
  21. FastMutex ConsoleChannel::_mutex;
  22. ConsoleChannel::ConsoleChannel(): _str(std::clog)
  23. {
  24. }
  25. ConsoleChannel::ConsoleChannel(std::ostream& str): _str(str)
  26. {
  27. }
  28. ConsoleChannel::~ConsoleChannel()
  29. {
  30. }
  31. void ConsoleChannel::log(const Message& msg)
  32. {
  33. FastMutex::ScopedLock lock(_mutex);
  34. _str << msg.getText() << std::endl;
  35. }
  36. FastMutex ColorConsoleChannel::_mutex;
  37. const std::string ColorConsoleChannel::CSI("\033[");
  38. ColorConsoleChannel::ColorConsoleChannel():
  39. _str(std::clog),
  40. _enableColors(true)
  41. {
  42. initColors();
  43. }
  44. ColorConsoleChannel::ColorConsoleChannel(std::ostream& str):
  45. _str(str),
  46. _enableColors(true)
  47. {
  48. initColors();
  49. }
  50. ColorConsoleChannel::~ColorConsoleChannel()
  51. {
  52. }
  53. void ColorConsoleChannel::log(const Message& msg)
  54. {
  55. FastMutex::ScopedLock lock(_mutex);
  56. if (_enableColors)
  57. {
  58. int color = _colors[msg.getPriority()];
  59. if (color & 0x100)
  60. {
  61. _str << CSI << "1m";
  62. }
  63. color &= 0xff;
  64. _str << CSI << color << "m";
  65. }
  66. _str << msg.getText();
  67. if (_enableColors)
  68. {
  69. _str << CSI << "0m";
  70. }
  71. _str << std::endl;
  72. }
  73. void ColorConsoleChannel::setProperty(const std::string& name, const std::string& value)
  74. {
  75. if (name == "enableColors")
  76. {
  77. _enableColors = icompare(value, "true") == 0;
  78. }
  79. else if (name == "traceColor")
  80. {
  81. _colors[Message::PRIO_TRACE] = parseColor(value);
  82. }
  83. else if (name == "debugColor")
  84. {
  85. _colors[Message::PRIO_DEBUG] = parseColor(value);
  86. }
  87. else if (name == "informationColor")
  88. {
  89. _colors[Message::PRIO_INFORMATION] = parseColor(value);
  90. }
  91. else if (name == "noticeColor")
  92. {
  93. _colors[Message::PRIO_NOTICE] = parseColor(value);
  94. }
  95. else if (name == "warningColor")
  96. {
  97. _colors[Message::PRIO_WARNING] = parseColor(value);
  98. }
  99. else if (name == "errorColor")
  100. {
  101. _colors[Message::PRIO_ERROR] = parseColor(value);
  102. }
  103. else if (name == "criticalColor")
  104. {
  105. _colors[Message::PRIO_CRITICAL] = parseColor(value);
  106. }
  107. else if (name == "fatalColor")
  108. {
  109. _colors[Message::PRIO_FATAL] = parseColor(value);
  110. }
  111. else
  112. {
  113. Channel::setProperty(name, value);
  114. }
  115. }
  116. std::string ColorConsoleChannel::getProperty(const std::string& name) const
  117. {
  118. if (name == "enableColors")
  119. {
  120. return _enableColors ? "true" : "false";
  121. }
  122. else if (name == "traceColor")
  123. {
  124. return formatColor(_colors[Message::PRIO_TRACE]);
  125. }
  126. else if (name == "debugColor")
  127. {
  128. return formatColor(_colors[Message::PRIO_DEBUG]);
  129. }
  130. else if (name == "informationColor")
  131. {
  132. return formatColor(_colors[Message::PRIO_INFORMATION]);
  133. }
  134. else if (name == "noticeColor")
  135. {
  136. return formatColor(_colors[Message::PRIO_NOTICE]);
  137. }
  138. else if (name == "warningColor")
  139. {
  140. return formatColor(_colors[Message::PRIO_WARNING]);
  141. }
  142. else if (name == "errorColor")
  143. {
  144. return formatColor(_colors[Message::PRIO_ERROR]);
  145. }
  146. else if (name == "criticalColor")
  147. {
  148. return formatColor(_colors[Message::PRIO_CRITICAL]);
  149. }
  150. else if (name == "fatalColor")
  151. {
  152. return formatColor(_colors[Message::PRIO_FATAL]);
  153. }
  154. else
  155. {
  156. return Channel::getProperty(name);
  157. }
  158. }
  159. ColorConsoleChannel::Color ColorConsoleChannel::parseColor(const std::string& color) const
  160. {
  161. if (icompare(color, "default") == 0)
  162. return CC_DEFAULT;
  163. else if (icompare(color, "black") == 0)
  164. return CC_BLACK;
  165. else if (icompare(color, "red") == 0)
  166. return CC_RED;
  167. else if (icompare(color, "green") == 0)
  168. return CC_GREEN;
  169. else if (icompare(color, "brown") == 0)
  170. return CC_BROWN;
  171. else if (icompare(color, "blue") == 0)
  172. return CC_BLUE;
  173. else if (icompare(color, "magenta") == 0)
  174. return CC_MAGENTA;
  175. else if (icompare(color, "cyan") == 0)
  176. return CC_CYAN;
  177. else if (icompare(color, "gray") == 0)
  178. return CC_GRAY;
  179. else if (icompare(color, "darkGray") == 0)
  180. return CC_DARKGRAY;
  181. else if (icompare(color, "lightRed") == 0)
  182. return CC_LIGHTRED;
  183. else if (icompare(color, "lightGreen") == 0)
  184. return CC_LIGHTGREEN;
  185. else if (icompare(color, "yellow") == 0)
  186. return CC_YELLOW;
  187. else if (icompare(color, "lightBlue") == 0)
  188. return CC_LIGHTBLUE;
  189. else if (icompare(color, "lightMagenta") == 0)
  190. return CC_LIGHTMAGENTA;
  191. else if (icompare(color, "lightCyan") == 0)
  192. return CC_LIGHTCYAN;
  193. else if (icompare(color, "white") == 0)
  194. return CC_WHITE;
  195. else throw InvalidArgumentException("Invalid color value", color);
  196. }
  197. std::string ColorConsoleChannel::formatColor(Color color) const
  198. {
  199. switch (color)
  200. {
  201. case CC_DEFAULT: return "default";
  202. case CC_BLACK: return "black";
  203. case CC_RED: return "red";
  204. case CC_GREEN: return "green";
  205. case CC_BROWN: return "brown";
  206. case CC_BLUE: return "blue";
  207. case CC_MAGENTA: return "magenta";
  208. case CC_CYAN: return "cyan";
  209. case CC_GRAY: return "gray";
  210. case CC_DARKGRAY: return "darkGray";
  211. case CC_LIGHTRED: return "lightRed";
  212. case CC_LIGHTGREEN: return "lightGreen";
  213. case CC_YELLOW: return "yellow";
  214. case CC_LIGHTBLUE: return "lightBlue";
  215. case CC_LIGHTMAGENTA: return "lightMagenta";
  216. case CC_LIGHTCYAN: return "lightCyan";
  217. case CC_WHITE: return "white";
  218. default: return "invalid";
  219. }
  220. }
  221. void ColorConsoleChannel::initColors()
  222. {
  223. _colors[0] = CC_DEFAULT; // unused
  224. _colors[Message::PRIO_FATAL] = CC_LIGHTRED;
  225. _colors[Message::PRIO_CRITICAL] = CC_LIGHTRED;
  226. _colors[Message::PRIO_ERROR] = CC_LIGHTRED;
  227. _colors[Message::PRIO_WARNING] = CC_YELLOW;
  228. _colors[Message::PRIO_NOTICE] = CC_DEFAULT;
  229. _colors[Message::PRIO_INFORMATION] = CC_DEFAULT;
  230. _colors[Message::PRIO_DEBUG] = CC_GRAY;
  231. _colors[Message::PRIO_TRACE] = CC_GRAY;
  232. }
  233. } // namespace Poco