Update .gitlab-ci.yml file

.gitlab-ci.yml

@@ -12,23 +12,59 @@
 
 stages:
   - build
+  - deploy
 
 variables:
-  DOCKER_BUILDKIT: 1  # Включаем BuildKit для оптимизации
+  DOCKER_HOST: tcp://docker:2375
+  DOCKER_DRIVER: overlay2
+  DOCKER_TLS_CERTDIR: ""
+  DOCKER_BUILDKIT: 1
 
-build:
+build-docker-image:
+  image: docker:latest
   stage: build
-  image: docker:24  # Обновлённый Docker для улучшений
   services:
-    - docker:24-dind
-  script:
-    - docker build --no-cache --pull --progress=plain -t flask-backend flask_ml_api
-
+    - docker:dind
   before_script:
-    - export DOCKER_TMPDIR=/builds/tmp  # Указываем временный каталог для Docker
+    - docker login -u "$CI_REGISTRY_USER" -p "$CI_REGISTRY_PASSWORD" "$CI_REGISTRY"
+    - export DOCKER_TMPDIR=/mnt/volume
+  script:
+    - >
+      if [[ "$CI_COMMIT_BRANCH" == "$CI_DEFAULT_BRANCH" ]]; then
+        tag="latest";
+      else
+        tag="$CI_COMMIT_REF_SLUG";
+      fi
+    - docker build --pull --no-cache --progress=plain -t "$CI_REGISTRY_IMAGE:$tag" .
+    - docker push "$CI_REGISTRY_IMAGE:$tag"
   cache:
     key: build-cache
     paths:
-      - /builds/tmp  # Кэшируем временные файлы
+      - /mnt/volume
+  rules:
+    - exists:
+        - Dockerfile
+
+deploy-to-bw-cloud:
+  image: alpine:latest
+  stage: deploy
+  before_script:
+    - apk add --no-cache sshpass
+  script:
+    - >
+      if [[ "$CI_COMMIT_BRANCH" == "$CI_DEFAULT_BRANCH" ]]; then
+        tag="latest";
+      else
+        tag="$CI_COMMIT_REF_SLUG";
+      fi
+    - sshpass -p "$BW_CLOUD_PASSWORD" ssh -o StrictHostKeyChecking=no "$BW_CLOUD_USER@$BW_CLOUD_HOST" "
+        docker pull $CI_REGISTRY_IMAGE:$tag &&
+        docker stop flask-container || true &&
+        docker rm flask-container || true &&
+        docker run -d --name flask-container -p 5001:5000 $CI_REGISTRY_IMAGE:$tag
+      "
+  only:
+    - main
+