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 auth = require('./auth')
const acl = require('./acl')

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 swarmOpts = {}
if (process.env.GIT_PEAR_AUTH && process.env.GIT_PEAR_AUTH !== 'native') {
  swarmOpts.keyPair = home.getKeyPair()
}
const swarm = new Hyperswarm(swarmOpts)

if (!home.isDaemonRunning()) {
  console.error('Please start git pear daemon')
  process.exit(1)
}

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

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

  let payload = { body: { url, method: 'get-repos' } }
  if (process.env.GIT_PEAR_AUTH && process.env.GIT_PEAR_AUTH !== 'native') {
    payload.header = await auth.getToken(payload.body)
  }

  const reposRes = await rpc.request('get-repos', Buffer.from(JSON.stringify(payload)))
  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 })

  payload = { body: { url, method: 'get-refs', data: repoName }}
  if (process.env.GIT_PEAR_AUTH && process.env.GIT_PEAR_AUTH !== 'native') {
    payload.header = await auth.getToken(payload.body)
  }
  const refsRes = await rpc.request('get-refs', Buffer.from(JSON.stringify(payload)))

  let commit 
  try {
    commit = await git.getCommit()
  } catch (e) { }
  await talkToGit(JSON.parse(refsRes.toString()), drive, repoName, rpc, commit)
})

async function talkToGit (refs, drive, repoName, rpc, commit) {
  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('+')

      dst = dst.replace('refs/heads/', '').replace('\n\n', '')

      try { home.createAppFolder(repoName) } catch (e) { }
      try { await git.createBareRepo(repoName) } catch (e) { }
      try { await git.addRemote(repoName) } catch (e) { }
      try { await git.push(dst) } catch (e) { }
      try { home.shareAppFolder(repoName) } catch (e) { }
      try { acl.setACL(repoName, acl.getACL(repoName)) } catch (e) { }

      let method
      if (isDelete) {
        method = 'd-branch'
      } else if (isForce) {
        console.warn('To', url)
        await git.push(src, isForce)
        src = src.replace('+', '')
        method = 'f-push'
      } else {
        console.warn('To', url)
        await git.push(src)
        method = 'push'
      }

      const publicKey = home.readPk()
      let payload = { body: {
        url: `pear://${publicKey}/${repoName}`,
        data: `${dst}#${commit}`,
        method
      } }
      if (process.env.GIT_PEAR_AUTH && process.env.GIT_PEAR_AUTH !== 'native') {
        payload.header = await auth.getToken(payload.body)
      }
      const res = await rpc.request(method, Buffer.from(JSON.stringify(payload)))

      process.stdout.write('\n\n')
      process.exit(0)
    } else if (chunk && chunk.search(/^list/) !== -1) { // list && list for-push
      for (const ref in refs) {
        console.warn(refs[ref] + '\t' + ref)
      }
      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) {
      for (const ref in refs) {
        console.warn(refs[ref] + '\t' + ref)
      }
      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
  })
}