# /bin/bash

dockfile=""

tempf() {
   case "$(uname -s)" in
      Darwin )
         local f="$(mktemp)"
         mkdir -p $HOME/.tmp
         mv "$f" $HOME/.tmp
         echo "$HOME/.tmp/${f##*/}"
      ;;
      * )
         mktemp "$@"
      ;;
   esac
}

function main() {
   set -e
   set -u
   thispkg="${PWD##*$GOPATH/}"
   thisexec="$(basename "$thispkg")"
   function cleanup() {
      if [ -z "${DONOTDELETE:-""}" ]; then
         echo rm "$dockfile"
      fi
   }
   trap cleanup EXIT

   dockfile="$(tempf)"
   echo "$dockfile"

   if [ -n "$(ls ./*.go 2>/dev/null )" ]; then
      echo "PACKING GO..."
      go_dockerfiles
   elif [ -n "$(ls ./*.rb 2> /dev/null)" ]; then
      echo "PACKING RUBY..."
      rb_dockerfiles
   elif [ -n "$(ls ./*.py 2>/dev/null )" ]; then
      if [ "$(ls ./main.py 2>/dev/null )" != "" ] && [[ "$(head -n 1 ./main.py)" == *python2* ]]; then
         echo "PACKING PYTHON2..."
         py2_dockerfiles
      else
         echo "PACKING PYTHON3..."
         py3_dockerfiles
      fi
   fi

   if [ "$#" -lt 1 ]; then
      args="-t ${thispkg##*/}:latest"
   else
      args="$*"
   fi

   docker build $args -f "$dockfile" .
}

function py3_dockerfiles() {
   echo "PY3"

   img="${1:-python}"

   if [[ "$(ls ./*req*.txt 2>/dev/null)" == "" ]]; then
      echo "NEED requirements.txt"
      exit 1
   fi
   reqFile="$(ls ./*req*.txt | head -n 1)"

   if [[ "$img" == "python" ]]; then
      from="python:3.7.0-alpine3.8"
   elif [[ "$img" == "pypy" ]]; then
      from="pypy:3-6.0.0-slim-jessie"
   else
      echo "ERR py3_dockerfile can be '', 'python', or 'pypy', not ${img}"
   fi

   if [ "$(uname -s)" == "Darwin" ]; then
      from="${from}"
   fi

   f=()
   f+=('')
   f+=('FROM alpine as certs')
   f+=('RUN apk update && apk add --no-cache ca-certificates')
   f+=('')
   f+=('#========================')
   f+=('')
   f+=('FROM '"${from}"'')
   f+=('WORKDIR /main/')
   f+=('COPY --from=certs /etc/ssl/certs /etc/ssl/certs')
   f+=('')
   if [[ "$img" == "python" ]]; then
      echo ""
      #f+=('RUN apk add linux-headers')
      #f+=('RUN apk add g++ gcc make')
   else
      echo ""
      #f+=('RUN apt-get update')
      #f+=('RUN apt-get install -y linux-headers')
      #f+=('RUN apt-get install -y g++ gcc make')
   fi
   f+=('')
   f+=('COPY ./'"$(basename ${reqFile})"' /main/req.txt')
   f+=('RUN pip install -r /main/req.txt')
   f+=('')
   f+=('COPY . .')
   if [[ "$img" == "python" ]]; then
      f+=('ENTRYPOINT ["python3", "/main/main.py"]')
   else
      f+=('ENTRYPOINT ["pypy3", "/main/main.py"]')
   fi
   f+=('')
   DOCKERFILE=""
   for i in "${f[@]}"; do
      DOCKERFILE+="${i}"$'\n'
   done
   echo "$DOCKERFILE" > "$dockfile"
echo $dockfile
cat $dockfile
}

function py2_dockerfiles() {
   echo "PY2"

   if [[ "$(ls ./*req*.txt 2>/dev/null)" == "" ]]; then
      echo "NEED requirements.txt"
      exit 1
   fi
   reqFile="$(ls ./*req*.txt | head -n 1)"

   from="python:2.7.15-alpine3.8"
   if [ "$(uname -s)" == "Darwin" ]; then
      from="${from}"
   fi

   f=()
   f+=('')
   f+=('FROM alpine as certs')
   f+=('RUN apk update && apk add --no-cache ca-certificates')
   f+=('')
   f+=('#========================')
   f+=('')
   f+=('FROM '"${from}"'')
   f+=('WORKDIR /main/')
   f+=('COPY --from=certs /etc/ssl/certs /etc/ssl/certs')
   f+=('COPY ./'"$(basename ${reqFile})"' /main/req.txt')
   f+=('RUN pip2.7 install -r /main/req.txt')
   f+=('COPY . .')
   f+=('ENTRYPOINT ["python2.7", "/main/main.py"]')
   f+=('')
   DOCKERFILE=""
   for i in "${f[@]}"; do
      DOCKERFILE+="${i}"$'\n'
   done
   echo "$DOCKERFILE" > "$dockfile"
echo $dockfile
cat $dockfile
}

function go_dockerfiles() {
   if ! [ -e "./exec-$thisexec" ]; then
      GOOS=linux CGO_ENABLED=0 go build -o ./exec-$thisexec -a -installsuffix cgo
   fi

from="golang:1.13-alpine"
if [ "$(uname -s)" == "Darwin" ] && ssh eng hostname; then
   from="registry-app.eng.qops.net:5001/imported/alpine:3.12"
fi

#FROM '"$from"' as builder
#MAINTAINER breel@qualtrics.com
#
#RUN apk add --no-cache git
#
#WORKDIR /go/'"$thispkg"'
#COPY . .
#RUN /bin/sh -c "CGO_ENABLED=0 go build -o /go/bin/'"$thisexec"' -a -installsuffix cgo"
#
#FROM busybox:glibc
#COPY --from=builder /go/'"$thispkg"' /main
#COPY --from=builder /go/bin /main

#========================

DOCKERFILE='
FROM '"$from"' as certs
RUN apk update && apk add --no-cache ca-certificates

FROM busybox:glibc
RUN mkdir -p /var/log
WORKDIR /main
COPY --from=certs /etc/ssl/certs /etc/ssl/certs

COPY . .

RUN test -e /main/exec-'"$thisexec"'

ENV GOPATH=""
ENV MNT="/mnt/"
ENTRYPOINT ["/main/exec-'"$thisexec"'"]
CMD []
'
echo "$DOCKERFILE" > "$dockfile"
echo $dockfile
cat $dockfile
}

function rb_dockerfiles() {
from="ruby:2.7.0-alpine3.11"
if [ "$(uname -s)" == "Darwin" ]; then
   from="registry-app.eng.qops.net:5001/imported/alpine/ruby:alpine"
fi

cp=( $(find . -mindepth 1 -maxdepth 1 -type f | grep -v "$dockfile" | grep -v '\.[^\.]*\.sw[a-z]$' | grep -v '\.tar$' ) )
cpd=( $(find . -mindepth 1 -maxdepth 1 -type d | grep -v \\.git || true) )
gf=""
if [ -e ./Gemfile ]; then
gf='
COPY ./Gemfile /main/Gemfile
RUN bundle config build.google-protobuf --with-cflags=-D__va_copy=va_copy \
   && BUNDLE_FORCE_RUBY_PLATFORM=1 bundler install --gemfile=/main/Gemfile
'
fi

main="$(find . -maxdepth 1 -name main.rb || find . -maxdepth 1 -name "*.rb" | head -n 1)"

DOCKERFILE='
FROM alpine as certs
RUN apk update && apk add --no-cache ca-certificates

#========================

FROM '"$from"' as builder
RUN mkdir -p /var/log
RUN apk add --no-cache ruby ruby-bundler make gcc libc-dev
'"$gf"'
#CMD []
#ENTRYPOINT ["sh"]

#========================

FROM '"$from"'
MAINTAINER breel@qualtrics.com
RUN rm -rf /usr/local/lib/ruby /usr/local/bundle/
RUN mkdir -p /usr/local/lib/ /usr/local/bundle/
RUN apk update && apk add --no-cache ca-certificates
#COPY --from=certs /etc/ssl/certs /etc/ssl/certs
COPY --from=builder /usr/local/lib/ /usr/local/lib/
COPY --from=builder /usr/local/bundle/ /usr/local/bundle/
WORKDIR /main
ENV HOME=/main
ENV MNT=/mnt/
COPY '"${cp[@]}"' /main/
'"$(for dir in ${cpd[@]}; do printf "COPY $dir /main/${dir##*/}\n"; done)"'
ENTRYPOINT ["ruby", "/main/'"${main##*/}"'"]
CMD []
'
echo "$DOCKERFILE" > "$dockfile"
echo $dockfile
cat $dockfile
}

main $@