put private tabs in different sessions

partial fix for #6684
and #8963

TODO: Either the private tab sessions should be cleared and revert back to private session 1 when the last private tab is closed, or we need a private tab session manager menu. Currently the session will keep incrementing until the browser restarts.

Test Plan:
1. open private tab
2. in the private tab, go to a site that requires login
3. login, then open a link from that site in a new private tab
4. you should also be logged in in the second private tab
5. open a new private tab
6. go to the same site. you should not be logged in.

Author: diracdeltas

app/browser/tabs.js

@@ -13,7 +13,7 @@ const {app, BrowserWindow, extensions, session, ipcMain} = require('electron')
 const {makeImmutable} = require('../common/state/immutableUtil')
 const {getTargetAboutUrl, getSourceAboutUrl, isSourceAboutUrl, newFrameUrl, isTargetAboutUrl, isIntermediateAboutPage, isTargetMagnetUrl, getSourceMagnetUrl} = require('../../js/lib/appUrlUtil')
 const {isURL, getUrlFromInput, toPDFJSLocation, getDefaultFaviconUrl, isHttpOrHttps, getLocationIfPDF} = require('../../js/lib/urlutil')
-const {isSessionPartition} = require('../../js/state/frameStateUtil')
+const {isSessionPartition, getPartitionNumber, getPartitionFromNumber} = require('../../js/state/frameStateUtil')
 const {getOrigin} = require('../../js/lib/urlutil')
 const {getSetting} = require('../../js/settings')
 const settings = require('../../js/constants/settings')
@@ -35,7 +35,10 @@ const historyState = require('../common/state/historyState')
 const bookmarkOrderCache = require('../common/cache/bookmarkOrderCache')
 
 let currentPartitionNumber = 0
-const incrementPartitionNumber = () => ++currentPartitionNumber
+let currentPrivatePartitionNumber = 0
+const incrementPartitionNumber = (isPrivate) => isPrivate
+  ? ++currentPrivatePartitionNumber
+  : ++currentPartitionNumber
 
 const normalizeUrl = function (url) {
   if (isSourceAboutUrl(url)) {
@@ -70,28 +73,26 @@ const updateTab = (tabId, changeInfo = {}) => {
   }
 }
 
-const getPartitionNumber = (partition) => {
-  return Number(partition.split('persist:partition-')[1] || 0)
-}
-
 /**
  * Obtains the current partition.
  * Warning: This function has global side effects in that it increments the
  * global next partition number if isPartitioned is passed into the create options.
  */
 const getPartition = (createProperties) => {
-  let partition = session.defaultSession.partition
   if (createProperties.partition) {
-    partition = createProperties.partition
-  } else if (createProperties.isPrivate) {
-    partition = 'default'
-  } else if (createProperties.isPartitioned) {
-    partition = `persist:partition-${incrementPartitionNumber()}`
-  } else if (createProperties.partitionNumber) {
-    partition = `persist:partition-${createProperties.partitionNumber}`
+    return createProperties.partition
   }
-
-  return partition
+  if (createProperties.partitionNumber) {
+    return getPartitionFromNumber(createProperties.partitionNumber,
+      createProperties.isPrivate)
+  }
+  if (createProperties.isPrivate) {
+    return getPartitionFromNumber(incrementPartitionNumber(true), true)
+  }
+  if (createProperties.isPartitioned) {
+    return getPartitionFromNumber(incrementPartitionNumber(false), false)
+  }
+  return session.defaultSession.partition
 }
 
 const needsPartitionAssigned = (createProperties) => {

app/renderer/components/main/siteInfo.js

@@ -109,7 +109,7 @@ class SiteInfo extends React.Component {
   }
 
   get partitionInfo () {
-    if (this.props.partitionNumber) {
+    if (this.props.partitionNumber > 0) {
       // Figure out the partition info display
       let l10nArgs = {
         partitionNumber: this.props.partitionNumber

app/renderer/components/tabs/tab.js

@@ -257,7 +257,7 @@ class Tab extends React.Component {
     props.isNarrowestView = tabContentState.isNarrowestView(currentWindow, props.frameKey)
     props.isPlayIndicatorBreakpoint = tabContentState.isMediumView(currentWindow, props.frameKey) || props.isNarrowView
     props.title = frame.get('title')
-    props.showSessionIcon = partition && hasSeconardImage
+    props.showSessionIcon = partition > 0 && hasSeconardImage
     props.showPrivateIcon = props.isPrivateTab && hasSeconardImage
     props.showFavIcon = !((hasBreakpoint(breakpoint, 'extraSmall') && props.isActive) || frame.get('location') === 'about:newtab')
     props.showAudioIcon = breakpoint === 'default' && !!frame.get('audioPlaybackActive')

js/state/frameStateUtil.js

@@ -278,30 +278,37 @@ function isAncestorFrameKey (state, frame, parentFrameKey) {
 }
 
 function getPartitionNumber (partition) {
-  const regex = /(?:persist:)?partition-(\d+)/
+  const regex = /^(?:persist:)?partition-(\d+)$/
   const matches = regex.exec(partition)
-  return Number((matches && matches[1]) || 0)
+  const partitionNumber = Number((matches && matches[1]) || 0)
+  // Return negative numbers for private sessions to avoid conflicts with
+  // regular sessions
+  return isPrivatePartition(partition) ? -partitionNumber : partitionNumber
 }
 
 function isPrivatePartition (partition) {
-  return partition && !partition.startsWith('persist:')
+  if (partition === '') {
+    return true
+  }
+  if (partition) {
+    return !partition.startsWith('persist:')
+  }
+  return false
 }
 
 function isSessionPartition (partition) {
-  return partition && partition.startsWith('persist:partition-')
+  return partition &&
+    (partition.startsWith('persist:partition-') || partition.startsWith('partition-'))
 }
 
 function getPartition (frameOpts) {
   return getPartitionFromNumber(frameOpts.get('partitionNumber'), frameOpts.get('isPrivate'))
 }
 
 function getPartitionFromNumber (partitionNumber, incognito) {
-  if (!partitionNumber && !incognito) {
-    return 'persist:default'
-  } else if (incognito) {
-    return 'default'
-  }
-  return `persist:partition-${partitionNumber}`
+  const partition = partitionNumber ? `partition-${Math.abs(partitionNumber)}` : 'default'
+  const prefix = incognito ? '' : 'persist:'
+  return `${prefix}${partition}`
 }
 
 const frameOptsFromFrame = (frame) => {

test/unit/state/frameStateUtilTest.js

@@ -411,4 +411,40 @@ describe('frameStateUtil', function () {
       assert.equal(result, 1)
     })
   })
+
+  describe('partition utils', function () {
+    it('getPartitionNumber', function () {
+      const getPartitionNumber = frameStateUtil.getPartitionNumber
+      assert.equal(getPartitionNumber(''), 0)
+      assert.equal(getPartitionNumber('1'), 0)
+      assert.equal(getPartitionNumber('partition-1'), -1)
+      assert.equal(getPartitionNumber('partition-222'), -222)
+      assert.equal(getPartitionNumber('partition-222.1'), 0)
+      assert.equal(getPartitionNumber('persist:partition-1'), 1)
+      assert.equal(getPartitionNumber('unpersist:partition-1'), 0)
+      assert.equal(getPartitionNumber('persist:partition-10'), 10)
+      assert.equal(getPartitionNumber('persist:partition'), 0)
+    })
+    it('isPrivatePartition', function () {
+      const isPrivatePartition = frameStateUtil.isPrivatePartition
+      assert.equal(isPrivatePartition(''), true)
+      assert.equal(isPrivatePartition('partition-1'), true)
+      assert.equal(isPrivatePartition('persist:partition-1'), false)
+    })
+    it('isSessionPartition', function () {
+      const isSessionPartition = frameStateUtil.isSessionPartition
+      assert.equal(isSessionPartition(''), false)
+      assert.equal(isSessionPartition('default'), false)
+      assert.equal(isSessionPartition('partition-1'), true)
+      assert.equal(isSessionPartition('persist:partition-1'), true)
+    })
+    it('getPartitionFromNumber', function () {
+      const getPartitionFromNumber = frameStateUtil.getPartitionFromNumber
+      assert.equal(getPartitionFromNumber(0, true), 'default')
+      assert.equal(getPartitionFromNumber(0, false), 'persist:default')
+      assert.equal(getPartitionFromNumber(1, true), 'partition-1')
+      assert.equal(getPartitionFromNumber(-10, true), 'partition-10')
+      assert.equal(getPartitionFromNumber(1, false), 'persist:partition-1')
+    })
+  })
 })