name: publish-docker

on:
  push:
    paths-ignore:
      - ".github/**"
      - "docs/**"
      - "resources/**"
      - "benchmark/**"
      - "tests/**"
      - "**/*.md"
    branches:
      - main
    tags:
      - "v*.*.*"

jobs:
  publish_docker_image:
    runs-on: [self-hosted, linux-3090]
    environment: 'prod'
    env:
      TAG_PREFIX: "openmmlab/lmdeploy"
    steps:
      - name: Checkout repository
        uses: actions/checkout@v3
      - name: Get lmdeploy commit id
        run: |
          export commit_id=$(git log --format="%H" -n 1)
          echo "commit id = $commit_id"
          echo "COMMIT_ID=$commit_id" >> $GITHUB_ENV
      - name: Build and push the latest Docker image
        run: |
          export TAG=$TAG_PREFIX:latest
          echo $TAG
          docker build docker/ -t ${TAG} --build-arg COMMIT_ID=${COMMIT_ID}
          docker push $TAG
          echo "TAG=${TAG}" >> $GITHUB_ENV
      - name: Push docker image with released tag
        if: startsWith(github.ref, 'refs/tags/') == true
        run: |
          export LMDEPLOY_VERSION=$(python3 -c "import sys; sys.path.append('lmdeploy');from version import __version__;print(__version__)")
          echo $LMDEPLOY_VERSION
          export RELEASE_TAG=${TAG_PREFIX}:v${LMDEPLOY_VERSION}
          echo $RELEASE_TAG
          docker tag $TAG $RELEASE_TAG
          docker push $RELEASE_TAG