From 6a0d923ef406bdf2d825c999f6814ba0259df834 Mon Sep 17 00:00:00 2001
From: NejcZdovc <nejc.zdovc@3zsistemi.si>
Date: Wed, 14 Mar 2018 15:41:00 -0700
Subject: [PATCH] Fixes formatting for YT on shields page

Resolves #12271

Auditors:

Test Plan:
---
 .../preferences/payment/deletedSites.js       | 12 +++++++---
 .../preferences/payment/ledgerTable.js        |  5 ++--
 .../components/preferences/paymentsTab.js     | 10 +++++++-
 app/sessionStore.js                           | 23 ++++++++++++++++++-
 docs/state.md                                 |  7 +++---
 5 files changed, 47 insertions(+), 10 deletions(-)

diff --git a/app/renderer/components/preferences/payment/deletedSites.js b/app/renderer/components/preferences/payment/deletedSites.js
index 6342c30e1de..54a36cd9c40 100644
--- a/app/renderer/components/preferences/payment/deletedSites.js
+++ b/app/renderer/components/preferences/payment/deletedSites.js
@@ -33,16 +33,22 @@ class DeletedSitesContent extends ImmutableComponent {
     return <div id='sitePermissionsPage'>
       <ul className={css(styles.sitePermissions__list)}>
         {
-          this.props.sites.map(hostPattern => {
+          this.props.sites.map(data => {
+            let siteName = (data.hostPattern || '').replace('?://', '://')
+
+            if (data.siteName) {
+              siteName = data.siteName
+            }
+
             return <div className={css(styles.sitePermissions__list__item)}>
               <BrowserButton
                 iconOnly
                 iconClass={globalStyles.appIcons.remove}
                 size='1rem'
                 custom={styles.sitePermissions__list__item__button}
-                onClick={this.deletePermission.bind(this, permissionName, hostPattern)}
+                onClick={this.deletePermission.bind(this, permissionName, data.hostPattern)}
               />
-              <span>{hostPattern}</span>
+              <span>{siteName}</span>
             </div>
           })
         }
diff --git a/app/renderer/components/preferences/payment/ledgerTable.js b/app/renderer/components/preferences/payment/ledgerTable.js
index 00dceb3cac3..eadbc261024 100644
--- a/app/renderer/components/preferences/payment/ledgerTable.js
+++ b/app/renderer/components/preferences/payment/ledgerTable.js
@@ -101,10 +101,11 @@ class LedgerTable extends ImmutableComponent {
     return synopsis.get('pinPercentage')
   }
 
-  banSite (hostPattern) {
+  banSite (hostPattern, siteName) {
     const confMsg = locale.translation('banSiteConfirmation')
     if (window.confirm(confMsg)) {
       aboutActions.changeSiteSetting(hostPattern, 'ledgerPaymentsShown', false)
+      aboutActions.changeSiteSetting(hostPattern, 'siteName', siteName)
     }
   }
 
@@ -258,7 +259,7 @@ class LedgerTable extends ImmutableComponent {
             styles.actionIcons__icon,
             styles.actionIcons__icon_remove
           )}
-            onClick={this.banSite.bind(this, this.getHostPattern(synopsis))}
+            onClick={this.banSite.bind(this, this.getHostPattern(synopsis), siteName)}
           />
         </div>,
         value: ''
diff --git a/app/renderer/components/preferences/paymentsTab.js b/app/renderer/components/preferences/paymentsTab.js
index 02c847b0a09..187f35d16c2 100644
--- a/app/renderer/components/preferences/paymentsTab.js
+++ b/app/renderer/components/preferences/paymentsTab.js
@@ -144,7 +144,15 @@ class PaymentsTab extends ImmutableComponent {
       }
 
       if (['boolean', 'number'].includes(typeof granted)) {
-        sites.push(hostPattern)
+        let siteName = null
+        if (value.has('siteName')) {
+          siteName = value.get('siteName')
+        }
+
+        sites.push({
+          siteName,
+          hostPattern
+        })
       }
     })
 
diff --git a/app/sessionStore.js b/app/sessionStore.js
index b3a4c9e48af..7121ebf50c8 100644
--- a/app/sessionStore.js
+++ b/app/sessionStore.js
@@ -825,8 +825,12 @@ module.exports.runPreMigrations = (data) => {
   if (data.lastAppVersion) {
     // Force WidevineCdm to be upgraded when last app version <= 0.18.25
     let runWidevineCleanup = false
+    let formatPublishers = false
 
-    try { runWidevineCleanup = compareVersions(data.lastAppVersion, '0.18.25') < 1 } catch (e) {}
+    try {
+      runWidevineCleanup = compareVersions(data.lastAppVersion, '0.18.25') < 1
+      formatPublishers = compareVersions(data.lastAppVersion, '0.21.9') < 1
+    } catch (e) {}
 
     if (runWidevineCleanup) {
       const fs = require('fs-extra')
@@ -838,6 +842,23 @@ module.exports.runPreMigrations = (data) => {
       })
     }
 
+    if (formatPublishers) {
+      const publishers = data.ledger.synopsis.publishers
+
+      if (publishers && Object.keys(publishers).length > 0) {
+        Object.entries(publishers).forEach((item) => {
+          const publisherKey = item[0]
+          const publisher = item[1]
+          const siteKey = `https?://${publisherKey}`
+          if (data.siteSettings[siteKey] == null || publisher.faviconName == null) {
+            return
+          }
+
+          data.siteSettings[siteKey].siteName = publisher.faviconName
+        })
+      }
+    }
+
     // Bookmark cache was generated wrongly on and before 0.20.25 from 0.19.x upgrades
     let runCacheClean = false
     try { runCacheClean = compareVersions(data.lastAppVersion, '0.20.25') < 1 } catch (e) {}
diff --git a/docs/state.md b/docs/state.md
index aa5f8bbc5b9..b5574d111f8 100644
--- a/docs/state.md
+++ b/docs/state.md
@@ -510,6 +510,7 @@ AppStore
   siteSettings: {
     [hostPattern]: {
       adControl: string, // (showBraveAds | blockAds | allowAdsAndTracking)
+      autoplay: boolean,
       cookieControl: string, // (block3rdPartyCookie | allowAllCookies | blockAllCookies)
       fingerprintingProtection: string, // (block3rdPartyFingerprinting | allowAllFingerprinting | blockAllFingerprinting)
       flash: (number|boolean), // approval expiration time if allowed, false if never allow
@@ -528,14 +529,14 @@ AppStore
       openExternalPermission: boolean,
       pointerLockPermission: boolean,
       protocolRegistrationPermission: boolean,
-      skipSync: boolean, // Set for objects FETCHed by sync
       runInsecureContent: boolean, // allow active mixed content
       safeBrowsing: boolean,
+      siteName: string, // display name of the publisher
+      skipSync: boolean, // Set for objects FETCHed by sync
       savePasswords: boolean, // only false or undefined/null
       shieldsUp: boolean,
       widevine: (number|boolean), // false = block widevine, 0 = allow once, 1 = allow always
-      zoomLevel: number,
-      autoplay: boolean,
+      zoomLevel: number
     }
   },
   defaultSiteSettingsListImported: boolean,