stages:
  - deploy
  - deploy-streamlit

variables:
  BACKEND_APP_NAME: "slop.unturf.com"
  BACKEND_APP_DIR: "/opt/${BACKEND_APP_NAME}"
  BACKEND_VENV_DIR: "${BACKEND_APP_DIR}/venv"
  STREAMLIT_APP_NAME: "streamlit.slop.unturf.com"
  STREAMLIT_APP_DIR: "/opt/${STREAMLIT_APP_NAME}"
  STREAMLIT_VENV_DIR: "${STREAMLIT_APP_DIR}/venv"
  CADDYFILE_DIR: "/etc/caddy"
  CADDY_SNIPPETS_DIR: "/etc/caddy/caddy_snippets"

deploy:
  stage: deploy
  tags:
    - master.unturf.com
  only:
    - main  # Deploy only from the 'main' branch
  script:
    # Step 1: Define variables with substitutions
    - |
      SERVICE_NAME="uwsgi-${BACKEND_APP_NAME//./-}.service"  # Replace dots with dashes
      UWSGI_SOCKET="${BACKEND_APP_DIR}/uwsgi.sock"           # Place socket in app directory

    # Step 2: Synchronize the code to the backend application directory
    - |
      if [ ! -d "${BACKEND_APP_DIR}" ]; then
        mkdir -p "${BACKEND_APP_DIR}"
      fi
      rsync -a --exclude=data/ --exclude=venv/ --delete "${CI_PROJECT_DIR}/" "${BACKEND_APP_DIR}/"

    # Step 3: Set up Python virtual environment and install dependencies
    - |
      if [ ! -d "${BACKEND_VENV_DIR}" ]; then
        python3 -m venv "${BACKEND_VENV_DIR}"
      fi
      source "${BACKEND_VENV_DIR}/bin/activate"
      pip install --upgrade pip
      pip install -r "${BACKEND_APP_DIR}/requirements.txt"

    # Step 4: Install uWSGI in virtual environment
    - |
      source "${BACKEND_VENV_DIR}/bin/activate"
      pip install uwsgi

    # Step 5: Create or update the systemd user service file for uWSGI
    - |
      mkdir -p ~/.config/systemd/user/

      # Set the MODULE for slop_with_models.py
      MODULE="slop_with_models:app"

      cat > ~/.config/systemd/user/${SERVICE_NAME} <<EOF
      [Unit]
      Description=uWSGI instance to serve ${BACKEND_APP_NAME}
      After=network.target

      [Service]
      WorkingDirectory=${BACKEND_APP_DIR}
      Environment="PATH=${BACKEND_VENV_DIR}/bin"
      Environment="MODEL_ENDPOINT_1=https://hermes.ai.unturf.com/v1"
      Environment="MODEL_ENDPOINT_2=https://hermes2.ai.unturf.com/v1"
      Environment="MODEL_API_KEY_1=${MODEL_API_KEY_1:-not-needed}"
      Environment="MODEL_API_KEY_2=${MODEL_API_KEY_2:-not-needed}"
      ExecStart=${BACKEND_VENV_DIR}/bin/uwsgi \\
        --master \\
        --enable-threads \\
        --processes 4 \\
        --http-socket ${UWSGI_SOCKET} \\
        --chmod-socket=660 \\
        --chown-socket=gitlab-runner:caddy \\
        --vacuum \\
        --die-on-term \\
        --chdir ${BACKEND_APP_DIR} \\
        --logto2 ${BACKEND_APP_DIR}/uwsgi.log \\
        --module ${MODULE}

      Restart=always
      KillSignal=SIGQUIT
      Type=notify
      NotifyAccess=all

      [Install]
      WantedBy=default.target
      EOF

    # Step 6: Reload systemd user daemon and restart the uWSGI service
    - systemctl --user daemon-reload
    - systemctl --user restart ${SERVICE_NAME}
    - systemctl --user enable ${SERVICE_NAME}

    # Step 7: Handle Caddy configurations for backend
    - |
      mkdir -p "${CADDY_SNIPPETS_DIR}"
      cat > "${CADDY_SNIPPETS_DIR}/${BACKEND_APP_NAME}.caddy" <<EOF
      ${BACKEND_APP_NAME} {
          encode gzip
          tls {
              on_demand
          }
          handle {
              reverse_proxy unix//${UWSGI_SOCKET}
          }
      }
      EOF

    # Deployment completed
    - echo "Backend deployment completed successfully."

deploy-streamlit:
  stage: deploy-streamlit
  tags:
    - master.unturf.com
  only:
    - main  # Deploy only from the 'main' branch
  script:
    # Step 1: Define variables
    - |
      SERVICE_NAME="streamlit-${STREAMLIT_APP_NAME//./-}.service"  # Replace dots with dashes
      STREAMLIT_PORT="8501"  # Default Streamlit port, can be changed

    # Step 2: Synchronize the code to the Streamlit application directory
    - |
      if [ ! -d "${STREAMLIT_APP_DIR}" ]; then
        mkdir -p "${STREAMLIT_APP_DIR}"
      fi
      rsync -a --exclude=data/ --exclude=venv/ --delete "${CI_PROJECT_DIR}/" "${STREAMLIT_APP_DIR}/"

    # Step 3: Set up Python virtual environment and install dependencies
    - |
      if [ ! -d "${STREAMLIT_VENV_DIR}" ]; then
        python3 -m venv "${STREAMLIT_VENV_DIR}"
      fi
      source "${STREAMLIT_VENV_DIR}/bin/activate"
      pip install --upgrade pip
      pip install -r "${STREAMLIT_APP_DIR}/requirements.txt"
      pip install streamlit

    # Step 4: Create or update the systemd user service file for Streamlit
    - |
      mkdir -p ~/.config/systemd/user/

      cat > ~/.config/systemd/user/${SERVICE_NAME} <<EOF
      [Unit]
      Description=Streamlit instance to serve ${STREAMLIT_APP_NAME}
      After=network.target

      [Service]
      WorkingDirectory=${STREAMLIT_APP_DIR}
      Environment="PATH=${STREAMLIT_VENV_DIR}/bin"
      ExecStart=${STREAMLIT_VENV_DIR}/bin/streamlit run streamlit_slop_with_models.py --server.port ${STREAMLIT_PORT} --server.address 127.0.0.1
      Restart=always
      Type=simple

      [Install]
      WantedBy=default.target
      EOF

    # Step 5: Reload systemd user daemon and restart the Streamlit service
    - systemctl --user daemon-reload
    - systemctl --user restart ${SERVICE_NAME}
    - systemctl --user enable ${SERVICE_NAME}

    # Step 6: Handle Caddy configurations for Streamlit
    - |
      mkdir -p "${CADDY_SNIPPETS_DIR}"
      cat > "${CADDY_SNIPPETS_DIR}/${STREAMLIT_APP_NAME}.caddy" <<EOF
      ${STREAMLIT_APP_NAME} {
          encode gzip
          tls {
              on_demand
          }
          handle {
              reverse_proxy 127.0.0.1:${STREAMLIT_PORT}
          }
      }
      EOF

      # Concatenate all snippets into the main Caddyfile
      echo '# Main Caddyfile - concatenated snippets' > "${CADDYFILE_DIR}/Caddyfile"
      cat "${CADDY_SNIPPETS_DIR}"/*.caddy >> "${CADDYFILE_DIR}/Caddyfile"

      # Validate the Caddyfile
      caddy validate --config "${CADDYFILE_DIR}/Caddyfile"

    # Step 7: Reload Caddy service
    - sudo /bin/systemctl restart caddy.service

    # Deployment completed
    - echo "Streamlit deployment completed successfully."