gitpear/src/git-remote-pear.js

#!/usr/bin/env node

const { spawn } = require('child_process')
const ProtomuxRPC = require('protomux-rpc')

const RAM = require('random-access-memory')
const Corestore = require('corestore')
const Hyperswarm = require('hyperswarm')
const Hyperdrive = require('hyperdrive')
const crypto = require('hypercore-crypto')

const git = require('./git.js')
const home = require('./home')

const fs = require('fs')

const url = process.argv[3]
const matches = url.match(/pear:\/\/([a-f0-9]{64})\/(.*)/)

if (!matches || matches.length < 3) {
  console.error('Invalid URL')
  process.exit(1)
}

const targetKey = matches[1]
const repoName = matches[2]

const store = new Corestore(RAM)
const swarm = new Hyperswarm({ keypair: home.getKeyPair() })

// let daemonPid
// if (!home.isDaemonRunning()) {
//   const opts = {
//    detached: true,
//    stdio: [ 'ignore', home.getOutStream(), home.getErrStream() ]
//   }
//   const daemon = spawn('git-peard', opts)
//   daemonPid = daemon.pid
//   home.storeDaemonPid(daemonPid)
//   // TODO: remove in case of error or exit but allow unref
//   // daemon.on('error', home.removeDaemonPid)
//   // daemon.on('exit', home.removeDaemonPid)
//   console.error('started daemon', daemonPid)
//   daemon.unref()
// }

swarm.join(crypto.discoveryKey(Buffer.from(targetKey, 'hex')), { server: false })

swarm.on('connection', async (socket) => {
  store.replicate(socket)
  const rpc = new ProtomuxRPC(socket)

  const reposRes = await rpc.request('get-repos')
  const repositories = JSON.parse(reposRes.toString())
  if (!repositories) {
    console.error('Failed to retrieve repositories')
    process.exit(1)
  }

  if (!repositories[repoName]) {
    console.error('Failed to retrieve repository')
    process.exit(1)
  }

  const driveKey = Buffer.from(repositories[repoName], 'hex')
  if (!driveKey) {
    console.error('Failed to retrieve pack key')
    process.exit(1)
  }

  const packStore = store.namespace(repoName)
  const drive = new Hyperdrive(packStore, driveKey)
  await drive.ready()
  swarm.join(drive.discoveryKey, { server: false, client: true })
  await swarm.flush()

  await drive.core.update({ wait: true })

  const refsRes = await rpc.request('get-refs', Buffer.from(repoName))

  await talkToGit(JSON.parse(refsRes.toString()), drive, repoName, rpc)
})

async function talkToGit (refs, drive, repoName, rpc) {
  for (const ref in refs) {
    console.warn(refs[ref] + '\t' + ref)
  }
  process.stdin.setEncoding('utf8')
  const didFetch = false
  process.stdin.on('readable', async function () {
    const chunk = process.stdin.read()
    if (chunk === 'capabilities\n') {
      process.stdout.write('list\n')
      process.stdout.write('push\n')
      process.stdout.write('fetch\n\n')
    } else if (chunk && chunk.search(/^push/) !== -1) {
      const [_command, path] = chunk.split(' ')
      let [src, dst] = path.split(':')

      const isDelete = !src
      const isForce = src.startsWith('+')

      if (!home.isShared(repoName)) {
        home.shareAppFolder(name)
      }
      
      console.error('dst', JSON.stringify(dst))
      dst = dst.replace('refs/heads/', '').replace('\n\n', '')
      console.error('dst', JSON.stringify(dst))
      await git.push(dst)

      let command
      if (isDelete) {
        command = 'delete-branch-from-repo'
      } else if (isForce) {
        command = 'force-push-to-repo'
      } else {
        command = 'push-to-repo'
      }

      const publicKey = home.readPk()
      const res = await rpc.request(command, Buffer.from(`${publicKey}/${repoName}:${dst}`))

      // process.kill(daemonPid || home.getDaemonPid())
      // home.removeDaemonPid()

      console.error('res', res.toString())
      process.stdout.write(res.toString())
      process.stdout.write('\n\n')
      process.exit(0)
    } else if (chunk && chunk.search(/^list/) !== -1) { // list && list for-push
      Object.keys(refs).forEach(function (branch, i) {
        process.stdout.write(refs[branch] + ' ' + branch + '\n')
      })
      process.stdout.write('\n')
    } else if (chunk && chunk.search(/^fetch/) !== -1) {
      const lines = chunk.split(/\n/).filter(l => l !== '')

      const targets = []
      await lines.forEach(async function (line) {
        if (line === '') return

        line = line.split(/\s/)

        if (targets.includes(line[1])) return

        targets.push(line[1])
      })

      for (let i = 0; i < targets.length; i++) {
        const sha = targets[i]

        const exist = await drive.exists(`/packs/${sha}.pack`)
        if (!exist) process.exit(1)

        const driveStream = drive.createReadStream(`/packs/${sha}.pack`, { start: 0 })
        await git.unpackStream(driveStream)
      }

      process.stdout.write('\n\n')
      process.exit(0)
    } else if (chunk && chunk !== '' && chunk !== '\n') {
      console.warn('unhandled command: "' + chunk + '"')
    }

    if (chunk === '\n') {
      process.stdout.write('\n')
      if (!didFetch) {
        // If git already has all the refs it needs, we should exit now.
        process.exit()
      }
    }
  })
  process.stdout.on('error', function () {
    // stdout was closed
  })
}