DevTools: Show product badge in console

third_party/WebKit/Source/devtools/front_end/console/ConsoleViewMessage.js

 /*
  * Copyright (C) 2011 Google Inc.  All rights reserved.
  * Copyright (C) 2007, 2008 Apple Inc.  All rights reserved.
  * Copyright (C) 2009 Joseph Pecoraro
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  * 1.  Redistributions of source code must retain the above copyright
@@ skipping 17 matching lines @@
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 /**
  * @implements {Console.ConsoleViewportElement}
  * @unrestricted
  */
 Console.ConsoleViewMessage = class {
   /**
    * @param {!ConsoleModel.ConsoleMessage} consoleMessage
    * @param {!Components.Linkifier} linkifier
+   * @param {!ProductRegistry.BadgePool} badgePool
    * @param {number} nestingLevel
    */
-  constructor(consoleMessage, linkifier, nestingLevel) {
+  constructor(consoleMessage, linkifier, badgePool, nestingLevel) {
     this._message = consoleMessage;
     this._linkifier = linkifier;
+    this._badgePool = badgePool;
     this._repeatCount = 1;
     this._closeGroupDecorationCount = 0;
     this._nestingLevel = nestingLevel;
 
     /** @type {?DataGrid.DataGrid} */
     this._dataGrid = null;
     this._previewFormatter = new ObjectUI.RemoteObjectPreviewFormatter();
     this._searchRegex = null;
     /** @type {?UI.Icon} */
     this._messageLevelIcon = null;
@@ skipping 58 matching lines @@
   /**
    * @return {!Element}
    */
   _buildTableMessage() {
     var formattedMessage = createElement('span');
     UI.appendStyle(formattedMessage, 'object_ui/objectValue.css');
     formattedMessage.className = 'source-code';
     var anchorElement = this._buildMessageAnchor();
     if (anchorElement)
       formattedMessage.appendChild(anchorElement);
+    var badgeElement = this._buildMessageBadge();
+    if (badgeElement)
+      formattedMessage.appendChild(badgeElement);
 
     var table = this._message.parameters && this._message.parameters.length ? this._message.parameters[0] : null;
     if (table)
       table = this._parameterToRemoteObject(table);
     if (!table || !table.preview)
       return formattedMessage;
 
     var columnNames = [];
     var preview = table.preview;
     var rows = [];
@@ skipping 119 matching lines @@
     }
     messageElement.classList.add('console-message-text');
 
     var formattedMessage = createElement('span');
     UI.appendStyle(formattedMessage, 'object_ui/objectValue.css');
     formattedMessage.className = 'source-code';
 
     var anchorElement = this._buildMessageAnchor();
     if (anchorElement)
       formattedMessage.appendChild(anchorElement);
+    var badgeElement = this._buildMessageBadge();
+    if (badgeElement)
+      formattedMessage.appendChild(badgeElement);
     formattedMessage.appendChild(messageElement);
     return formattedMessage;
   }
 
   /**
    * @return {?Element}
    */
   _buildMessageAnchor() {
     var anchorElement = null;
     if (this._message.source !== ConsoleModel.ConsoleMessage.MessageSource.Network || this._message.request) {
@@ skipping 13 matching lines @@
     if (anchorElement) {
       var anchorWrapperElement = createElementWithClass('span', 'console-message-anchor');
       anchorWrapperElement.appendChild(anchorElement);
       anchorWrapperElement.createTextChild(' ');
       return anchorWrapperElement;
     }
     return null;
   }
 
   /**
+   * @return {?Element}
+   */
+  _buildMessageBadge() {
+    if (!this._message.executionContextId)
+      return null;
+    var runtimeModel = this._message.runtimeModel();
+    if (!runtimeModel)
+      return null;
+    var executionContext = runtimeModel.executionContext(this._message.executionContextId);
+    if (!executionContext || !executionContext.frameId)
+      return null;
+    var resourceTreeModel = executionContext.target().model(SDK.ResourceTreeModel);
+    if (!resourceTreeModel)
+      return null;
+    var frame = resourceTreeModel.frameForId(executionContext.frameId);
+    if (!frame || !frame.parentFrame)
+      return null;
+    var badgeElement = this._badgePool.badgeForFrame(frame);
+    badgeElement.classList.add('console-message-badge');
+    return badgeElement;
+  }
+
+  /**
    * @return {!Element}
    */
   _buildMessageWithStackTrace() {
     var toggleElement = createElementWithClass('div', 'console-message-stack-trace-toggle');
     var contentElement = toggleElement.createChild('div', 'console-message-stack-trace-wrapper');
 
     var messageElement = this._buildMessage();
     var icon = UI.Icon.create('smallicon-triangle-right', 'console-message-expand-icon');
     var clickableElement = contentElement.createChild('div');
     clickableElement.appendChild(icon);
@@ skipping 873 matching lines @@
   }
 };
 
 /**
  * @unrestricted
  */
 Console.ConsoleGroupViewMessage = class extends Console.ConsoleViewMessage {
   /**
    * @param {!ConsoleModel.ConsoleMessage} consoleMessage
    * @param {!Components.Linkifier} linkifier
+   * @param {!ProductRegistry.BadgePool} badgePool
    * @param {number} nestingLevel
    */
-  constructor(consoleMessage, linkifier, nestingLevel) {
+  constructor(consoleMessage, linkifier, badgePool, nestingLevel) {
     console.assert(consoleMessage.isGroupStartMessage());
-    super(consoleMessage, linkifier, nestingLevel);
+    super(consoleMessage, linkifier, badgePool, nestingLevel);
     this._collapsed = consoleMessage.type === ConsoleModel.ConsoleMessage.MessageType.StartGroupCollapsed;
     /** @type {?UI.Icon} */
     this._expandGroupIcon = null;
   }
 
   /**
    * @param {boolean} collapsed
    */
   setCollapsed(collapsed) {
     this._collapsed = collapsed;
@@ skipping 15 matching lines @@
   toMessageElement() {
     if (!this._element) {
       super.toMessageElement();
       this._expandGroupIcon = UI.Icon.create('', 'expand-group-icon');
       this._contentElement.insertBefore(this._expandGroupIcon, this._contentElement.firstChild);
       this.setCollapsed(this._collapsed);
     }
     return this._element;
   }
 };