#!/bin/bash

CERT_DIR=certs
ROOT_DIR_DB=/data/b2g/mozilla
CERT=cert9.db
KEY=key4.db
PKCS11=pkcs11.txt
DB_DIR=`adb.exe shell "ls -d ${ROOT_DIR_DB}/*.default 2>/dev/null" | sed "s/default.*$/default/g"`

if [ "${DB_DIR}" = "" ]; then
  echo "Profile directory does not exists. Please start the b2g process at
least once before running this script."
  exit 1
fi

function log
{
    GREEN="\E[32m"
    RESET="\033[00;00m"
    echo -e "${GREEN}$1${RESET}"
}

# cleanup
rm -f ./$CERT
rm -f ./$KEY
rm -f ./$PKCS11

# pull files from phone
log "getting ${CERT}"
adb.exe pull ${DB_DIR}/${CERT} .
log "getting ${KEY}"
adb.exe pull ${DB_DIR}/${KEY} .
log "getting ${PKCS11}"
adb.exe pull ${DB_DIR}/${PKCS11} .

# clear password and add certificates
log "set password (hit enter twice to set an empty password)"
certutil -d 'sql:.' -N

log "adding certificats"
for i in ${CERT_DIR}/*
do
  log "Adding certificate $i"
  certutil -d 'sql:.' -A -n "`basename $i`" -t "C,C,TC" -i $i
done

# push files to phone
log "stopping b2g"
adb.exe shell stop b2g

log "copying ${CERT}"
adb.exe push ./${CERT} ${DB_DIR}/${CERT}
log "copying ${KEY}"
adb.exe push ./${KEY} ${DB_DIR}/${KEY}
log "copying ${PKCS11}"
adb.exe push ./${PKCS11} ${DB_DIR}/${PKCS11}

log "starting b2g"
adb.exe shell start b2g

log "Finished."