trailofbits/tubertc

#!/bin/sh

NODEJS_ROOT="null"

log () {
    printf "[info] $1\n"
}

error () {
    printf "[error] $1\n"
}

help () {
    printf "Usage: $0 [-h|--help] [-f|--force] [--run]\n"
    printf "  -h,--help   Shows this screen\n"
    printf "  -f,--force  Forces the download of a node.js instance\n"
    printf "  --run       Opens a browser after start of node.js server\n"
    exit 1
}

install_nodejs () {
    ARCH="`uname -m`"
    PLATFORM="`uname`"
    NODEJS_URL="null"
    DOWNLOAD=0

    if [ "$1" = "1" ]; then
        DOWNLOAD=1
    fi

    if [ "$DOWNLOAD" = "0" ]; then
        log "Checking if node is installed..."
        node -v 2>&1 > /dev/null
        if [ "$?" = "0" ]; then
            log "node is found and working"
            log "Checking if npm is installed..."
            npm -v 2>&1 > /dev/null
            if [ "$?" = "0" ]; then
                log "npm is found and working"
                log "No need to install node.js"
                return 0
            fi
            log "Failed to detect npm"
        fi
    else
        log "Ignoring node/npm command check, downloading node.js"
    fi
    
    NODEJS_VERSION="v8.11.4"
    NODEJS_SHASUM_URL="https://nodejs.org/dist/${NODEJS_VERSION}/SHASUMS256.txt"

    log "Detecting architecture and platform..."
    if [ $PLATFORM = "Darwin" ]; then
        if [ $ARCH = "x86_64" ]; then
            log "64-bit Mac OS X detected"
            NODEJS_URL="https://nodejs.org/dist/${NODEJS_VERSION}/node-${NODEJS_VERSION}-darwin-x64.tar.gz"
        fi
    elif [ $PLATFORM = "Linux" ]; then
        if [ $ARCH = "x86_64" ]; then
            log "64-bit Linux detected"
            NODEJS_URL="https://nodejs.org/dist/${NODEJS_VERSION}/node-${NODEJS_VERSION}-linux-x64.tar.gz"
        elif [ $ARCH = "i686" ]; then
            log "32-bit Linux detected"
            NODEJS_URL="https://nodejs.org/dist/${NODEJS_VERSION}/node-${NODEJS_VERSION}-linux-x86.tar.gz"
        fi
    fi

    if [ $NODEJS_URL = "null" ]; then
        error "ERROR: Unsupported architecture: $PLATFORM $ARCH"
        exit -1
    fi

    FILENAME="`basename $NODEJS_URL`"
    BASENAME=${FILENAME%.tar.gz}

    if [ ! -d $BASENAME ]; then
        log "Cannot find $BASENAME directory in $PWD"
        DOWNLOAD=1
    fi

    if [ "$DOWNLOAD" = "1" ]; then
        log "Downloading node.js from $NODEJS_URL"
        curl $NODEJS_URL -o "$FILENAME" 2> /dev/null 
        curl $NODEJS_SHASUM_URL -o SHASUMS256.txt 2> /dev/null

        EXPECTED_HASH=`grep $FILENAME SHASUMS256.txt | awk '{print $1}'`
        SHA256_HASH=`shasum -a 256 $FILENAME | awk '{print $1}'`

        if [ "$SHA256_HASH" != "$EXPECTED_HASH" ]; then
           error "ERROR: SHA256 sums do not match Expected $EXPECTED_HASH got $SHA256_HASH"
           exit 1
        fi
        tar zxf $FILENAME
        log "Download complete, extracted into $PWD/$BASENAME"
    else
        log "Found node.js instance at $PWD/$BASENAME"
    fi

    NODEJS_ROOT=$PWD/$BASENAME

    return 1
}

FORCE_NODEJS_DOWNLOAD=0
INSTALL_AND_EXIT=0
SPAWN_BROWSER=0

# Shell script argument parsing from https://gist.github.com/jehiah/855086
while [ "$1" != "" ]; do
    PARAM=`echo $1 | awk -F= '{print $1}'`
    VALUE=`echo $1 | awk -F= '{print $2}'`

    case $PARAM in
        -h | --help)
            help
            exit
            ;;
        -f | --force)
            FORCE_NODEJS_DOWNLOAD=1
            ;;
        -i | --install)
            FORCE_NODEJS_DOWNLOAD=1
            INSTALL_AND_EXIT=1
            ;;
        --run)
            SPAWN_BROWSER=1
            ;;
        *)
            printf "Error: Unknown parameter \"$PARAM\"\n"
            help
            exit 1
            ;;
    esac
    shift
done

log "Preparing to run TubeRTC..."

install_nodejs $FORCE_NODEJS_DOWNLOAD
if [ "$?" = "1" ]; then
    log "Adding $NODEJS_ROOT/bin to PATH"
    PATH="$PATH:$NODEJS_ROOT/bin"
fi

log "Installing node.js dependencies for TubeRTC"
npm install
if [ ! "$?" = "0" ]; then
    error "Failed to resolve node.js dependencies!"
    exit 1
fi

if [ "$INSTALL_AND_EXIT" = "1" ]; then
  log "Installed tubertc and dependncies"
  exit 0
fi

log "Running TubeRTC"

# Enable monitor mode so we can spawn another process if needed
set -m
npm start &

if [ "$SPAWN_BROWSER" = "1" ]; then
    sleep 2
    log "Spawning browser..."
    if !(open "http://localhost:8080"); then
        if !(xdg-open "http://localhost:8080"); then
            if !(start "http://localhost:8080"); then
                log "TubeRTC running at http://localhost:8080"
            fi
        fi
    fi
fi

fg