f-hj/slsk-client

const Peer = require('./peer.js')
const zlib = require('zlib')
const Messages = require('../messages.js')
const MessageFactory = require('../message-factory.js')
const downloadPeerFile = require('./download-peer-file.js')
const debug = require('debug')('slsk:peer:default:i')

let stack = require('../stack')

module.exports = class DefaultPeer extends Peer {
  constructor (socket, peer) {
    super(socket, peer)

    this.conn.on('connect', () => {
      if (peer.token) {
        let buf = Buffer.from('05000000' + '00' + peer.token, 'hex')
        this.conn.write(buf)
      }
    })

    let msgs = new Messages()

    this.conn.on('data', data => {
      msgs.write(data)
    })

    msgs.on('message', msg => {
      let size = msg.int32()
      if (size <= 4) return

      let code = msg.int32()
      switch (code) {
        case 4: {
          debug(`${peer.user} recv GetSharedFileList ${size}`)
          let res = MessageFactory
            .to.peer
            .sharedFileList(1, 'test', 1, '@@abcde\\test\\test-test.mp3')
          debug(`${peer.user} sending SharedFileList`)
          this.conn.write(res.getBuff())
          break
        }
        case 9: {
          debug(`${peer.user} recv FileSearchResult size ${size}`)
          let content = msg.data.slice(msg.pointer, size + 4)
          zlib.unzip(content, (err, buffer) => {
            if (err) {
              debug(err)
              return
            }

            let result = MessageFactory.from.peer.fileSearchResult(buffer)

            if (stack.search[result.currentToken]) {
              result.files.forEach(file => {
                stack.search[result.currentToken].cb({
                  user: file.user,
                  file: file.file,
                  size: file.size,
                  slots: result.slots === 1,
                  bitrate: file.attribs ? file.attribs[0] : undefined,
                  speed: result.speed
                })
              })
            }
          })
          break
        }
        case 36: {
          let numberOfFiles = msg.int32()
          debug(`${peer.user} recv FolderContentsRequest, files: ${numberOfFiles}`)
          break
        }
        case 40: {
          let dir = msg.int32()
          let token = msg.rawHexStr(4)
          let file = msg.str()
          debug(`${peer.user} recv TransferRequest ${file}`)
          stack.downloadTokens[token] = {
            user: peer.user,
            file
          }
          if (dir === 1) {
            stack.downloadTokens[token].size = msg.int32()
          }
          let res = MessageFactory
            .to.peer
            .transferResponse(token)

          setTimeout(() => {
            debug(`${peer.user} sending TransferResponse`)
            this.conn.write(res.getBuff())
          }, 200)
          break
        }
        case 41: {
          let token = msg.rawHexStr(4)
          let allowed = msg.int8()
          debug(`${peer.user} recv TransferResponse token: ${token} allowed: ${allowed}`)
          if (allowed === 0) {
            let reason = msg.str()
            debug(`reason: ${reason}, I will receive TransferRequest soon...`)
            delete stack.downloadTokens[token] // avoid memory leak
          } else if (allowed === 1) {
            debug(`Directly allowed. Connecting to peer with PeerInit + ${token}`)
            downloadPeerFile(peer.host, peer.port, token, peer.user, true)
          }
          break
        }
        case 46: {
          let filename = msg.str()
          debug(`${peer.user} UploadFailed ${filename}`)
          let down = stack.download[peer.user + '_' + filename]
          if (down && typeof down.cb === 'function') {
            down.cb(new Error('Peer error'))
          } else {
            debug(`Cannot cb for ${peer.user} ${filename}`)
          }
          break
        }
        case 50: {
          let filename = msg.str()
          let reason = msg.str()
          debug(`${peer.user} QueueFailed ${filename} reason ${reason}`)
          break
        }
        default: {
          debug(`${peer.user} unknown peer message code ${code}`)
        }
      }
    })
  }
}