// Licensed to the Software Freedom Conservancy (SFC) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The SFC licenses this file
// to you 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.

const Type = {
  CONTEXT: 'context',
  STORAGE_KEY: 'storageKey',
}

class PartitionDescriptor {
  #type

  constructor(type) {
    this.#type = type
  }
}

class BrowsingContextPartitionDescriptor extends PartitionDescriptor {
  #context = null

  constructor(context) {
    super(Type.CONTEXT)
    this.#context = context
  }

  asMap() {
    const map = new Map()
    map.set('type', Type.CONTEXT)
    map.set('context', this.#context)
    return map
  }
}

class StorageKeyPartitionDescriptor extends PartitionDescriptor {
  #map = new Map()

  constructor() {
    super(Type.STORAGE_KEY)
  }

  userContext(userContext) {
    this.#map.set('userContext', userContext)
    return this
  }

  sourceOrigin(sourceOrigin) {
    this.#map.set('sourceOrigin', sourceOrigin)
    return this
  }

  asMap() {
    return this.#map
  }
}

module.exports = { BrowsingContextPartitionDescriptor, StorageKeyPartitionDescriptor }
