penx/apps/extension/components/content/hooks.ts

import React, { useState } from 'react'
import { selectedSpaceKey, spacesKey } from '@/lib/helper'
import { useStorage } from '@plasmohq/storage/hook'

// import { selectedSpaceKey, spacesKey, storageDocKey } from '~/common/helper'

export function useSelectedSpace() {
  const [selectedSpace, setSelectedSpace] = useStorage(selectedSpaceKey, '')
  return { selectedSpace, setSelectedSpace }
}

export function useMySpaces() {
  const [mySpaces, setMySpaces] = useStorage<any[]>(spacesKey, [])
  return { mySpaces, setMySpaces }
}

export function useForceUpdate() {
  const [_, setCount] = useState(0)

  const forceUpdate = () => {
    setCount((count) => count + 1)
  }

  return {
    forceUpdate,
  }
}