wechaty-puppet-service/src/deprecated/file-box-chunk.ts

Puppet Service for Wechaty

import { FileBox }      from 'file-box'
import type { FileBoxInterface } from 'file-box'
import { PassThrough }  from 'stream'
import {
  Readable,
  Transform,
}                       from 'stronger-typed-streams'

import { puppet } from 'wechaty-grpc'

import { nextData } from './next-data.js'

/**
 * @deprecated Will be removed after Dec 31, 2022
 */
const decoder = () => new Transform<puppet.FileBoxChunk, any>({
  objectMode: true,
  transform: (chunk: puppet.FileBoxChunk, _: any, callback: any) => {
    if (!chunk.hasData()) {
      callback(new Error('no data'))
      return
    }
    const data = chunk.getData()
    callback(null, data)
  },
})

/**
 * @deprecated Will be removed after Dec 31, 2022
 */
async function unpackFileBoxFromChunk (
  stream: Readable<puppet.FileBoxChunk>,
): Promise<FileBox> {
  const chunk = await nextData(stream)
  if (!chunk.hasName()) {
    throw new Error('no name')
  }
  const fileName = chunk.getName()

  const fileStream = new PassThrough({ objectMode: true })
  const transformedStream = stream.pipe(decoder())
  transformedStream.pipe(fileStream)
  stream.on('error', e => fileStream.emit('error', e))
  transformedStream.on('error', e => fileStream.emit('error', e))

  const fileBox = FileBox.fromStream(fileStream, fileName)

  return fileBox
}

/**
 * @deprecated Will be removed after Dec 31, 2022
 */
const encoder = () => new Transform<any, puppet.FileBoxChunk>({
  objectMode: true,
  transform: (chunk: any, _: any, callback: any) => {
    const fileBoxChunk = new puppet.FileBoxChunk()
    fileBoxChunk.setData(chunk)
    callback(null, fileBoxChunk)
  },
})

/**
 * @deprecated Will be removed after Dec 31, 2022
 */
async function packFileBoxToChunk (
  fileBox: FileBoxInterface,
): Promise<Readable<puppet.FileBoxChunk>> {
  const stream = new PassThrough({ objectMode: true })

  const chunk = new puppet.FileBoxChunk()
  chunk.setName(fileBox.name)

  // FIXME: Huan(202010) write might return false
  stream.write(chunk)

  fileBox
    .pipe(encoder())
    .pipe(stream)

  return stream
}

export {
  unpackFileBoxFromChunk,
  packFileBoxToChunk,
}