carbon/lib/util.js

import morph from 'morphmorph'
import omitBy from 'lodash.omitby'
import { htmlUnescape } from 'escape-goat'

const SETTINGS_KEY = 'CARBON_STATE'
const PRESETS_KEY = 'CARBON_PRESETS'
const THEMES_KEY = 'CARBON_THEMES'

const createAssigner = key => {
  const assign = morph.assign(key)

  return v => assign(localStorage, JSON.stringify(v))
}

const map = fn => obj => obj.map(fn)
export const omit = keys => object => omitBy(object, (_, k) => keys.indexOf(k) > -1)

export const saveSettings = morph.compose(
  createAssigner(SETTINGS_KEY),
  omit([
    'code',
    'backgroundImage',
    'backgroundImageSelection',
    'themes',
    'highlights',
    'fontUrl',
    'selectedLines',
    'name',
  ])
)
export const savePresets = morph.compose(
  createAssigner(PRESETS_KEY),
  map(omit(['backgroundImageSelection']))
)
export const saveThemes = createAssigner(THEMES_KEY)

const parse = v => {
  try {
    return JSON.parse(v)
  } catch (e) {
    // pass
  }
}

export const toggle = stateField => state => ({ [stateField]: !state[stateField] })

// https://gist.github.com/alexgibson/1704515
export const escapeHtml = s => {
  if (typeof s === 'string') {
    return s.replace(/</g, '&lt;').replace(/>/g, '&gt;').replace(/\//g, '&#x2F;')
  }
}

export const unescapeHtml = s => {
  if (typeof s === 'string') {
    return htmlUnescape(s).replace(/&#x2F;/g, '/')
  }
}

export const getSettings = morph.compose(parse, escapeHtml, morph.get(SETTINGS_KEY))

export const getPresets = morph.compose(parse, morph.get(PRESETS_KEY))

export const getThemes = morph.compose(parse, morph.get(THEMES_KEY))

export const clearSettings = () => localStorage.removeItem(SETTINGS_KEY)

export const fileToDataURL = blob =>
  new Promise(res => {
    const reader = new FileReader()
    reader.onload = e => res(e.target.result)
    reader.readAsDataURL(blob)
  })

export const fileToJSON = blob =>
  new Promise(res => {
    const reader = new FileReader()
    reader.onload = e => res(parse(e.target.result))
    reader.readAsText(blob)
  })

export const formatCode = async code => {
  const prettier = await import('prettier/standalone')
  const babylonParser = await import('prettier/parser-babel')

  return prettier.format(code, {
    parser: 'babel',
    plugins: [babylonParser],
    semi: false,
    singleQuote: true,
  })
}

export const stringifyRGBA = obj => `rgba(${obj.r},${obj.g},${obj.b},${obj.a})`

export const generateId = () => Math.random().toString(36).slice(2)

export function dataURLtoBlob(dataurl) {
  const [first, second] = dataurl.split(',')
  const mime = first.match(/:(.*?);/)[1]
  const bstr = atob(second)
  let n = bstr.length
  const u8arr = new Uint8Array(n)
  while (n--) {
    u8arr[n] = bstr.charCodeAt(n)
  }
  return new Blob([u8arr], { type: mime })
}