#!/bin/bash
# Copyright (c) Abstract Machines
# SPDX-License-Identifier: Apache-2.0

# This script contains commands to be executed by the CI tool.
NPROC=$(nproc)
GO_VERSION=1.24.3
PROTOC_VERSION=30.2
PROTOC_GEN_VERSION=v1.36.6
PROTOC_GRPC_VERSION=v1.5.1
GOLANGCI_LINT_VERSION=v1.60.3

function version_gt() { test "$(printf '%s\n' "$@" | sort -V | head -n 1)" != "$1"; }

update_go() {
    CURRENT_GO_VERSION=$(go version | sed 's/[^0-9.]*\([0-9.]*\).*/\1/')
    if version_gt $GO_VERSION $CURRENT_GO_VERSION; then
        echo "Updating go version from $CURRENT_GO_VERSION to $GO_VERSION ..."
        # remove other Go version from path
        sudo rm -rf /usr/bin/go
        sudo rm -rf /usr/local/go
        sudo rm -rf /usr/local/bin/go
        sudo rm -rf /usr/local/golang
        sudo rm -rf $GOROOT $GOPAT $GOBIN
        wget https://go.dev/dl/go$GO_VERSION.linux-amd64.tar.gz
        sudo tar -C /usr/local -xzf go$GO_VERSION.linux-amd64.tar.gz
        export GOROOT=/usr/local/go
        export PATH=$PATH:/usr/local/go/bin
    fi
    export GOBIN=$HOME/go/bin
    export PATH=$PATH:$GOBIN
    go version
}

setup_protoc() {
    # Execute `go get` for protoc dependencies outside of project dir.
    echo "Setting up protoc..."
    PROTOC_ZIP=protoc-$PROTOC_VERSION-linux-x86_64.zip
    curl -0L https://github.com/google/protobuf/releases/download/v$PROTOC_VERSION/$PROTOC_ZIP -o $PROTOC_ZIP
    unzip -o $PROTOC_ZIP -d protoc3
    sudo mv protoc3/bin/* /usr/local/bin/
    sudo mv protoc3/include/* /usr/local/include/
    rm -rf $PROTOC_ZIP protoc3

    go install google.golang.org/protobuf/cmd/protoc-gen-go@$PROTOC_GEN_VERSION
    go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@$PROTOC_GRPC_VERSION

    export PATH=$PATH:/usr/local/bin/protoc
}

setup_mg() {
    echo "Setting up SuperMQ..."
    for p in $(ls *.pb.go); do
        mv $p $p.tmp
    done
    for p in $(ls pkg/*/*.pb.go); do
        mv $p $p.tmp
    done
    make proto
    for p in $(ls *.pb.go); do
        if ! cmp -s $p $p.tmp; then
            echo "Proto file and generated Go file $p are out of sync!"
            exit 1
        fi
    done
    for p in $(ls pkg/*/*.pb.go); do
        if ! cmp -s $p $p.tmp; then
            echo "Proto file and generated Go file $p are out of sync!"
            exit 1
        fi
    done
    echo "Compile check for rabbitmq..."
    SMQ_MESSAGE_BROKER_TYPE=rabbitmq make http
    echo "Compile check for redis..."
    SMQ_ES_TYPE=redis make http
    make -j$NPROC
}

setup_lint() {
    # binary will be $(go env GOBIN)/golangci-lint
    curl -sSfL https://raw.githubusercontent.com/golangci/golangci-lint/main/install.sh | sh -s -- -b $(go env GOBIN) $GOLANGCI_LINT_VERSION
}

setup() {
    echo "Setting up..."
    update_go
    setup_protoc
    setup_mg
    setup_lint
}

run_test() {
    echo "Running lint..."
    golangci-lint run
    echo "Running tests..."
    echo "" > coverage.txt
    for d in $(go list ./... | grep -v 'vendor\|cmd'); do
        GOCACHE=off
        go test -mod=vendor -v -race -tags test -coverprofile=profile.out -covermode=atomic $d
        if [ -f profile.out ]; then
            cat profile.out >> coverage.txt
            rm profile.out
        fi
    done
}

push() {
    if test -n "$BRANCH_NAME" && test "$BRANCH_NAME" = "main"; then
        echo "Pushing Docker images..."
        make -j$NPROC latest
    fi
}

set -e
setup
run_test
push
