filepizza/client/stores/UploadStore.js

import UploadActions from '../actions/UploadActions'
import UploadFile from '../UploadFile'
import alt from '../alt'
import peer from '../peer'
import socket from '../socket'

export default alt.createStore(class UploadStore {

  constructor() {
    this.bindActions(UploadActions)

    this.status = 'ready'
    this.token = null
    this.file = null

    this.peerProgress = {}
  }

  onUploadFile(file) {
    if (this.status !== 'ready') return
    this.status = 'processing'
    this.file = new UploadFile(file)

    socket.emit('upload', {
      name: this.file.name,
      size: this.file.size,
      type: this.file.type
    }, (token) => {
      this.setState({
        status: 'uploading',
        token: token
      })
    })
  }

  onSendToDownloader(peerID) {
    if (this.status !== 'uploading') return

    let conn = peer.connect(peerID, {
      reliable: true
    })

    let totalChunks = this.file.countChunks()
    let i = 0

    let sendNextChunk = () => {
      this.peerProgress[peerID] = i/totalChunks
      if (i === totalChunks) return
      let packet = this.file.getChunk(i)
      conn.send(packet)
      i++
      this.peerProgress[peerID] = i/totalChunks
    }

    let numInProgess = () => {
      var keys = Object.keys(this.peerProgress)
      var num = 0
      for (var k = 0; k < keys.length; k++) {
        if (this.peerProgress[keys[k]] < 1) {
          num++
        }
      }
      return num
    }

    let numCompleted = () => {
      var keys = Object.keys(this.peerProgress)
      var num = 0
      for (var k = 0; k < keys.length; k++) {
        if (this.peerProgress[keys[k]] == 1) {
          num++
        }
      }
      return num
    }

    conn.on('open', () => {
      sendNextChunk()
      this.setState({ inProgress: numInProgess()})
    })

    conn.on('data', (data) => {
      if (data === 'more') sendNextChunk()
    })

    conn.on('close', () => {
      this.setState({
        inProgress: numInProgess(),
        completed: numCompleted()
      })
    })
  }

}, 'UploadStore')