import { Effect } from 'effect'
import type { Page } from 'puppeteer'
import { getSelected } from '.'
import { getBatteryVariations } from '../variations/battery'

export const getBattery = (page: Page) =>
  Effect.gen(function* () {
    const batteryVariations = yield* getBatteryVariations(page)

    const selectedCapacity = yield* getSelected(batteryVariations ?? [])

    return selectedCapacity.label
  })