init.sh

#!/bin/sh

# Convenience script enhancing functionality of repo and envsetup.sh
#
# 2018 Vasyl Gello <vasek.gello@gmail.com>
#

# configuration block

# Default Gerrit checkout (fetch) method:
# Can be "http" / "https" (anonymous or authenticated)
# or "ssh" (authenticated)

GERRIT_CHECKOUT_METHOD="https"

# Gerrit SSH port (default is 29418)

GERRIT_SSH_PORT=29418

# functions

gerrit_reconstruct_topic()
(
  # $1 is Gerrit server hostname
  # $2 is quoted Gerrit query string ("topic:xxx branch:yyy status:open")
  # $3 is output type ("--git" or "--repopick")

  # Some helper functions

  trap_cleanup()
  {
    rm -f "$REC_TMPFILE"
    rm -f "$REC0_TMPFILE"
    rm -f "$RECREV_TMPFILE"
    rm -f "$RECHEADS_TMPFILE"
  }

  GERRIT_HOSTNAME="$1"
  GERRIT_QUERY="$2"
  OUTPUT_FORMAT="$3"

  if [ -z "$GERRIT_HOSTNAME" ] || [ -z "$GERRIT_QUERY" ]
  then
    echo "Usage: reconstruct.sh <Gerrit-hostname> \"<Gerrit-query>\" [--output-format=git|repopick]"
    echo ""
    echo "Output formats:"
    echo "    repopick: print changes in NUMBER/PATCHSET format (newer first)"
    echo "              to use with repopick"
    echo "         git: print changes in Project ref format (newer first)"
    echo "              to use with git fetch"
    echo ""
    echo "If --output-format is not specified, default setting is \"repopick\""
    return
  fi

  if [ -z "$OUTPUT_FORMAT" ] || [ "$OUTPUT_FORMAT" = "--output-format=repopick" ]
  then
    OUTPUT_FORMAT=0
  elif [ "$OUTPUT_FORMAT" = "--output-format=git" ]
  then
    OUTPUT_FORMAT=1
  else
    echo "Output format must be --output-format=git or --output-format=repopick, exiting..."
    return
  fi

  REC_TMPFILE=$(mktemp $HOME/REC-XXXXXXXXXXXXXXXX.txt)
  REC0_TMPFILE=$(mktemp $HOME/REC0-XXXXXXXXXXXXXXXX.txt)
  RECREV_TMPFILE=$(mktemp $HOME/REC-rev-XXXXXXXXXXXXXXXX.txt)
  RECHEADS_TMPFILE=$(mktemp $HOME/REC-heads-XXXXXXXXXXXXXXXX.txt)

  trap trap_cleanup 1 2 3 6

  # Get the JSON list

  if [ "$GERRIT_CHECKOUT_METHOD"="http" ] || [ "$GERRIT_CHECKOUT_METHOD"="http" ]
  then
    curl -s -G "$GERRIT_CHECKOUT_METHOD://$GERRIT_HOSTNAME/changes/" \
         --data-urlencode "q=$GERRIT_QUERY" \
         --data-urlencode "o=CURRENT_COMMIT" \
         --data-urlencode "o=CURRENT_REVISION" > "$REC_TMPFILE"

    # Prepare it for jq parse

    sed -i '1d' "$REC_TMPFILE"

    # Parse it with jq to get parent revisions

    i=0;
    while true
    do
      NUMBER=$(jq ".[$i]._number" "$REC_TMPFILE")
      [ -z "$NUMBER" ] || [ "$NUMBER" = "null" ] && break

      COMMIT=$(jq -r ".[$i].current_revision" "$REC_TMPFILE")
      PARENT=$(jq -r ".[$i].revisions.\""$COMMIT"\".commit.parents[0].commit" "$REC_TMPFILE")

      if [ $OUTPUT_FORMAT -eq 0 ]
      then
        PATCHSET=$(jq -r ".[$i].revisions.\""$COMMIT"\"._number" "$REC_TMPFILE")

        echo "$NUMBER/$PATCHSET $COMMIT $PARENT" >> "$RECREV_TMPFILE"
      else
        PROJECT=$(jq -r ".[$i].revisions.\""$COMMIT"\".fetch.\"anonymous http\".url" "$REC_TMPFILE")
        REF=$(jq -r ".[$i].revisions.\""$COMMIT"\".fetch.\"anonymous http\".ref" "$REC_TMPFILE")

        echo "$PROJECT+$REF $COMMIT $PARENT" >> "$RECREV_TMPFILE"
      fi

      i=$((i + 1))
    done
  elif [ "$GERRIT_CHECKOUT_METHOD"="ssh" ]
  then
    ssh -p $GERRIT_SSH_PORT "$GERRIT_HOSTNAME" \
      "gerrit query --format JSON --current-patch-set $GERRIT_QUERY" > "$REC_TMPFILE"

    # Prepare it for jq parse

    sed -i '/moreChanges/d' "$REC_TMPFILE"
    echo '{"changes":[' >> "$REC0_TMPFILE"
    sed 's|}$|},|' "$REC_TMPFILE" >> "$REC0_TMPFILE"
    echo "{\"project\":\"\",\"branch\":\"\",\"topic\":\"\",\"id\":\"\"}]}" >> "$REC0_TMPFILE"
    mv "$REC0_TMPFILE" "$REC_TMPFILE"

    # Parse it with jq to get parent revisions

    i=0;
    while true
    do
      NUMBER=$(jq ".changes[$i].number" "$REC_TMPFILE")
      [ -z "$NUMBER" ] || [ "$NUMBER" = "null" ] && break

      COMMIT=$(jq -r ".changes[$i].currentPatchSet.revision" "$REC_TMPFILE")
      PARENT=$(jq -r ".changes[$i].currentPatchSet.parents[0]" "$REC_TMPFILE")

      if [ $OUTPUT_FORMAT -eq 0 ]
      then
        PATCHSET=$(jq -r ".changes[$i].currentPatchSet.number" "$REC_TMPFILE")

        echo "$NUMBER/$PATCHSET $COMMIT $PARENT" >> "$RECREV_TMPFILE"
      else
        PROJECT=$(jq -r ".changes[$i].project" "$REC_TMPFILE")
        REF=$(jq -r ".changes[$i].currentPatchSet.ref" "$REC_TMPFILE")

        echo "$PROJECT+$REF $COMMIT $PARENT" >> "$RECREV_TMPFILE"
      fi

      i=$((i + 1))
    done
  else
    echo "ERROR: The GERRIT_CHECKOUT_METHOD setting must be \"http\", \"https\" or \"ssh\"!"
    return
  fi

  # Find the heads

  cat "$RECREV_TMPFILE" | while read LINE
  do
    COMMIT=$(echo "$LINE" | awk '{print $2}')
    PARENT=$(echo "$LINE" | awk '{print $3}')
    grep " $COMMIT$" "$RECREV_TMPFILE" 1>/dev/null 2>/dev/null
    [ $? -ne 0 ] && echo "$COMMIT" >> "$RECHEADS_TMPFILE"
  done

  # Rewind the found heads

  for HEAD_REV in $(tac "$RECHEADS_TMPFILE")
  do
    REV="$HEAD_REV"
    while true
    do
      if [ $OUTPUT_FORMAT -eq 0 ]
      then
        LINE=$(grep "^[0-9]*/[0-9]* $REV" "$RECREV_TMPFILE")
      else
        LINE=$(grep "+refs/changes/[0-9]*/[0-9]*/[0-9]* $REV" "$RECREV_TMPFILE")
      fi

      [ -z "$LINE" ] && break

      NUMBER=$(echo "$LINE" | awk '{print $1}')
      PARENT=$(echo "$LINE" | awk '{print $3}')

      if [ $OUTPUT_FORMAT -eq 0 ]
      then
        echo $NUMBER
      else
        NUMBER=$(echo "$NUMBER" | sed 's/+/ /')
        echo $NUMBER
      fi

      REV="$PARENT"
    done
  done

  # cleanup

  trap_cleanup
)

gerrit_cr()
(
  # $1 is review URL (review.lineageos.org)
  # $2 is quoted gerrit query ("status:open")
  # $3 is quoted code-review label ("-2", "-1", "0", "+1", "+2",  "n/a")
  # $4 is quoted verified label ("-1", "0", "+1", "n/a")
  # $5 is optional "--spam"

  # Some helper functions

  trap_cleanup()
  {
    rm -f "$CR_TMPFILE"
    rm -f "$CR0_TMPFILE"
  }

  GERRIT_HOSTNAME="$1"
  GERRIT_QUERY="$2"
  GERRIT_SPAMFLAG="$5"

  [ -z "$GERRIT_HOSTNAME" ] || \
  [ -z "$GERRIT_QUERY" ] || \
  [ -z "$3" ] || \
  [ -z "$4" ] && \
  echo "Usage: gerrit_cr <review-server-host> <\"gerrit-query\"> <\"CR-label\"> <\"V-label\"> [--spam]" && \
  return

  CR_LABEL=""
  for LABEL in "-2" "-1" "0" "+1" "+2" "n/a"
  do
    [ "$LABEL" = "$3" ] && CR_LABEL="$LABEL" && break
  done

  [ -z "$CR_LABEL" ] && echo "ERROR: Code-Review label must be \"-2\", \"-1\", \"0\", \"+1\", \"+2\" or \"n/a\" to skip the label" && return

  if [ "$CR_LABEL" = "n/a" ]
  then
    CR_LABEL=""
  else
    CR_LABEL="--code-review $CR_LABEL"
  fi

  V_LABEL=""
  for LABEL in "-1" "0" "+1" "n/a"
  do
    [ "$LABEL" = "$4" ] && V_LABEL="$LABEL" && break
  done

  [ -z "$V_LABEL" ] && echo "ERROR: Verified label must be \"-1\", \"0\", \"+1\" or \"n/a\" to skip the label" && return

  if [ "$V_LABEL" = "n/a" ]
  then
    V_LABEL=""
  else
    V_LABEL="--verified $V_LABEL"
  fi

  if [ -z "$GERRIT_SPAMFLAG" ]
  then
    GERRIT_SPAMFLAG="-n NONE"
  elif [ "$GERRIT_SPAMFLAG" = "--spam" ]
  then
    GERRIT_SPAMFLAG=""
  else
    echo "ERROR: Spam trigger can be \"--spam\" or empty string"
    return
  fi

  CR_TMPFILE=$(mktemp $HOME/CR-XXXXXXXXXXXXXXXX.txt)
  CR0_TMPFILE=$(mktemp $HOME/CR0-XXXXXXXXXXXXXXXX.txt)

  trap trap_cleanup 1 2 3 6

  # Retrieve list of changes

  if [ "$GERRIT_CHECKOUT_METHOD" = "http" ] || [ "$GERRIT_CHECKOUT_METHOD" = "https" ]
  then
    curl -s -G "$GERRIT_CHECKOUT_METHOD://$GERRIT_HOSTNAME/changes/" \
         --data-urlencode "q=$GERRIT_QUERY" \
         --data-urlencode "o=CURRENT_COMMIT" \
         --data-urlencode "o=CURRENT_REVISION" > "$CR_TMPFILE"

    # Prepare it for jq parse

    sed -i '1d' "$CR_TMPFILE"

    # Parse it with jq to get parent revisions

    i=0;
    while true
    do
      NUMBER=$(jq ".[$i]._number" "$CR_TMPFILE")
      [ -z "$NUMBER" ] || [ "$NUMBER" = "null" ] && break

      COMMIT=$(jq -r ".[$i].current_revision" "$CR_TMPFILE")

      echo "$COMMIT" >> "$CR0_TMPFILE"

      i=$((i + 1))
    done
  elif [ "$GERRIT_CHECKOUT_METHOD"="ssh" ]
  then
    ssh -p $GERRIT_SSH_PORT "$GERRIT_HOSTNAME" 'gerrit query --current-patch-set "'$GERRIT_QUERY'"' | \
      grep "^    revision:" | \
      sed 's,^    revision: ,,' > "$CR0_TMPFILE"
  else
    echo "ERROR: The GERRIT_CHECKOUT_METHOD setting must be \"http\", \"https\" or \"ssh\"!"
    return
  fi

  # Build list of change-ids

  IDS=""
  for ID in $(cat "$CR0_TMPFILE")
  do
    IDS="$ID $IDS"
  done

  # Perform actual review via SSH

  ssh -p $GERRIT_SSH_PORT "$1" "gerrit review $GERRIT_SPAMFLAG $CR_LABEL $V_LABEL $IDS"

  # cleanup

  trap_cleanup
)

fcb()
{
  # $1 is file1
  # $2 is file2

  [ -z "$1" ] || [ -z "$2" ] && "Usage: fcb <file1> <file2>" && return

  cmp -l "$1" "$2" |
    mawk 'function oct2dec(oct,    dec) {
              for (i = 1; i <= length(oct); i++) {
                  dec *= 8;
                  dec += substr(oct, i, 1)
              };
              return dec
          }
          {
              printf "%08X %02X %02X\n", $1-1, oct2dec($2), oct2dec($3)
          }'
}

repopick_topic()
{
  # $1 is topic-name
  # $2 is Gerrit URL

  [ -z "$1" ] && echo "Usage: repopick_topic <topic-name> [gerrit-url]" && return

  # check if topic-name is not a numeric ID
  echo "$1" | grep -ioe "[A-Za-z_-]" 1>/dev/null 2>/dev/null
  [ $? -ne 0 ] && echo "topic-name must not be the numeric change ID" && return

  # perform repopick
  if [ -z "$2" ]; then
    repopick -i -s _"$1" -t "$1"
  else
    repopick -i -s _"$1" -t "$1" -g "$2"
  fi
}

rebaseall()
{
  rmbuildbranch
  repo forall -c "BRANCHES=\$(git branch | grep -ioe \" _[0-9A-Za-z_-.]*$\"); [ \$? -eq 0 ] && for BRANCH in \$BRANCHES; do git checkout \$BRANCH; repo rebase .; done"
}

mkbuildbranch()
{
  repo abandon _build
  repo forall -c "BRANCHES=\$(git branch | grep -ioe \" _[0-9A-Za-z_-.]*$\"); [ \$? -eq 0 ] && repo start _build . && for BRANCH in \$BRANCHES; do git merge -m \"Merging the build\" \$BRANCH; done"
}

rmbuildbranch()
{
  repo abandon _build
}

killshell()
{
  history -c
  exit 0
}

# start script

if [[ "$(basename -- "$0")" == "init.sh" ]]; then
    echo "Don't run $0, source it" >&2
    exit 1
fi

. build/envsetup.sh 1>/dev/null 2>/dev/null