# KiCad CI Pipeline with hfsntree Integration
# Include this file in your project's .gitlab-ci.yml and set PROJECT_NAME variable
#
# Example project .gitlab-ci.yml:
#   include:
#     - local: '.gitlab/kibot-ci.yml'
#   variables:
#     PROJECT_NAME: "my_project_name"

workflow:
  rules:
    - if: $CI_PIPELINE_SOURCE == "merge_request_event"
    - if: $CI_COMMIT_BRANCH && $CI_OPEN_MERGE_REQUESTS
      when: never
    - if: $CI_COMMIT_BRANCH

variables:
  GIT_STRATEGY: clone
  GIT_SUBMODULE_STRATEGY: recursive
  GIT_SUBMODULE_FORCE_HTTPS: "true"
  PACKAGE_REGISTRY_URL: "${CI_API_V4_URL}/projects/${CI_PROJECT_ID}/packages/generic/kicad"

stages:
  - preflight
  - fabrication
  - inventree
  - release

image:
  name: ghcr.io/inti-cmnb/kicad9_auto:1.8.4

# =============================================================================
# Rule Templates
# =============================================================================
.main_rules:
  rules:
    - if: '$CI_PIPELINE_SOURCE == "merge_request_event" && $CI_MERGE_REQUEST_TARGET_BRANCH_NAME == "main"'
    - if: $CI_COMMIT_BRANCH == "main"
    - if: $GITLAB_CI == 'false'

.dev_rules:
  rules:
    - if: '$CI_PIPELINE_SOURCE == "merge_request_event" && $CI_MERGE_REQUEST_TARGET_BRANCH_NAME == "dev"'
    - if: $CI_COMMIT_BRANCH == "dev"
    - if: $GITLAB_CI == 'false'

# =============================================================================
# Stage: Preflight
# =============================================================================
extract_version:
  stage: preflight
  script:
    - |
      # Version strategy:
      # - dev branch: always "dev"
      # - main branch: extract from commit message (e.g., "V1.2")

      if [[ "$CI_COMMIT_BRANCH" == "main" ]]; then
        # On main, extract version from commit message
        VERSION=$(echo "$CI_COMMIT_MESSAGE" | grep -oE 'V[0-9]+(\.[0-9]+)*' | head -1)
        if [[ -z "$VERSION" ]]; then
          VERSION="$CI_COMMIT_SHORT_SHA"
          echo "Warning: No version found in commit message, using SHA"
        fi
        echo "Main branch version: $VERSION"
      else
        # On dev or any other branch, use "dev"
        VERSION="dev"
        echo "Dev branch version: $VERSION"
      fi

      echo "VERSION=$VERSION" >> build.env
      echo "Extracted version: $VERSION"
  artifacts:
    reports:
      dotenv: build.env

run_erc:
  stage: preflight
  needs:
    - job: extract_version
      artifacts: true
  script:
    - |
      if [ -f "${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch" ]; then
        echo "Running ERC..."
        cat > /tmp/erc.kibot.yaml << 'EOF'
      kibot:
        version: 1
      preflight:
        erc:
          enabled: true
          warnings_as_errors: false
      EOF
        kibot -e ${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch \
          -c /tmp/erc.kibot.yaml \
          -d Validation/
      fi
  artifacts:
    when: always
    paths:
      - Validation/
    expire_in: 1 week

run_drc:
  stage: preflight
  needs:
    - job: extract_version
      artifacts: true
  script:
    - |
      if [ -f "${PROJECT_NAME}/${PROJECT_NAME}.kicad_pcb" ]; then
        echo "Running DRC..."
        # Extract part number from project name for text variables
        PART_NUM=$(echo "${PROJECT_NAME}" | cut -d"-" -f1)
        cat > /tmp/drc.kibot.yaml << EOF
      kibot:
        version: 1
      preflight:
        set_text_variables:
          - name: 'name'
            text: '${PART_NUM}'
          - name: 'rev'
            text: '${VERSION}'
          - name: 'rev_pcb'
            text: '${VERSION}'
        drc:
          enabled: true
          warnings_as_errors: false
      EOF
        kibot -e ${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch \
          -b ${PROJECT_NAME}/${PROJECT_NAME}.kicad_pcb \
          -c /tmp/drc.kibot.yaml \
          -d Validation/
      fi
  artifacts:
    when: always
    paths:
      - Validation/
    expire_in: 1 week

# =============================================================================
# Stage: Fabrication
# =============================================================================
.fabrication_base:
  stage: fabrication
  needs:
    - job: extract_version
      artifacts: true
    - job: run_erc
      artifacts: false
    - job: run_drc
      artifacts: false
  before_script:
    - |
      # Ensure submodules are initialized (needed for gitlab-ci-local)
      git submodule update --init --recursive 2>/dev/null || true

      # Export version and name for KiBot text variable injection
      export KIBOT_VAR_rev="${VERSION}"
      export KIBOT_VAR_rev_pcb="${VERSION}"
      export KIBOT_VAR_name=$(echo "${PROJECT_NAME}" | cut -d"-" -f1)
      echo "KiBot version variables set to: $VERSION"
      echo "KiBot name variable set to: $KIBOT_VAR_name"

      # Setup user config
      USER_FILE="$CI_PROJECT_DIR/.gitlab/configs/blank.kibot.yaml"
      if [ -f "${PROJECT_NAME}/kibot.yaml" ]; then
        USER_FILE="${PROJECT_NAME}/kibot.yaml"
      fi
      cp $USER_FILE $CI_PROJECT_DIR/.gitlab/configs/user.kibot.yaml
  artifacts:
    when: always
    paths:
      - Fabrication/
    expire_in: 1 week

generate_schematic:
  extends: .fabrication_base
  script:
    - |
      export KIBOT_VAR_rev="${VERSION}"
      export KIBOT_VAR_rev_pcb="${VERSION}"
      export KIBOT_VAR_name=$(echo "${PROJECT_NAME}" | cut -d"-" -f1)
      echo "Using version: $VERSION, name: $KIBOT_VAR_name"

      if [ -f "${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch" ]; then
        kibot -e ${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch \
          -c $CI_PROJECT_DIR/.gitlab/configs/sch.kibot.yaml \
          -d $CI_PROJECT_DIR/Fabrication/${PROJECT_NAME}
        mv Fabrication/${PROJECT_NAME}/*.pdf Fabrication/ 2>/dev/null || true
      fi

generate_bom:
  extends: .fabrication_base
  script:
    - |
      export KIBOT_VAR_rev="${VERSION}"
      export KIBOT_VAR_rev_pcb="${VERSION}"
      export KIBOT_VAR_name=$(echo "${PROJECT_NAME}" | cut -d"-" -f1)
      echo "Using version: $VERSION, name: $KIBOT_VAR_name"

      if [ -f "${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch" ]; then
        kibot -e ${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch \
          -c $CI_PROJECT_DIR/.gitlab/configs/sch.kibot.yaml \
          -d $CI_PROJECT_DIR/Fabrication/${PROJECT_NAME}
        mv Fabrication/${PROJECT_NAME}/*.csv Fabrication/ 2>/dev/null || true
        mv Fabrication/${PROJECT_NAME}/*.xlsx Fabrication/ 2>/dev/null || true
      fi

generate_3d:
  extends: .fabrication_base
  script:
    - |
      export KIBOT_VAR_rev="${VERSION}"
      export KIBOT_VAR_rev_pcb="${VERSION}"
      export KIBOT_VAR_name=$(echo "${PROJECT_NAME}" | cut -d"-" -f1)
      echo "Using version: $VERSION, name: $KIBOT_VAR_name"

      if [ -f "${PROJECT_NAME}/${PROJECT_NAME}.kicad_pcb" ]; then
        kibot -e ${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch \
          -c $CI_PROJECT_DIR/.gitlab/configs/mech.kibot.yaml \
          -d $CI_PROJECT_DIR/Fabrication/${PROJECT_NAME}
        mv Fabrication/${PROJECT_NAME}/*.step Fabrication/ 2>/dev/null || true
      fi

generate_gerbers:
  extends: .fabrication_base
  script:
    - |
      export KIBOT_VAR_rev="${VERSION}"
      export KIBOT_VAR_rev_pcb="${VERSION}"
      export KIBOT_VAR_name=$(echo "${PROJECT_NAME}" | cut -d"-" -f1)
      echo "Using version: $VERSION, name: $KIBOT_VAR_name"

      if [ -f "${PROJECT_NAME}/${PROJECT_NAME}.kicad_pcb" ]; then
        kibot -e ${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch \
          -c $CI_PROJECT_DIR/.gitlab/configs/pcb_main.kibot.yaml \
          -d $CI_PROJECT_DIR/Fabrication/${PROJECT_NAME}
        mv Fabrication/${PROJECT_NAME}/*.zip Fabrication/ 2>/dev/null || true
      fi

generate_position:
  extends: .fabrication_base
  script:
    - |
      export KIBOT_VAR_rev="${VERSION}"
      export KIBOT_VAR_rev_pcb="${VERSION}"
      export KIBOT_VAR_name=$(echo "${PROJECT_NAME}" | cut -d"-" -f1)
      echo "Using version: $VERSION, name: $KIBOT_VAR_name"

      if [ -f "${PROJECT_NAME}/${PROJECT_NAME}.kicad_pcb" ]; then
        kibot -e ${PROJECT_NAME}/${PROJECT_NAME}.kicad_sch \
          -c $CI_PROJECT_DIR/.gitlab/configs/pos.kibot.yaml \
          -d $CI_PROJECT_DIR/Fabrication/${PROJECT_NAME}
        mv Fabrication/${PROJECT_NAME}/*cpl*.csv Fabrication/ 2>/dev/null || true
      fi

generate_panel:
  extends: .fabrication_base
  script:
    - |
      export KIBOT_VAR_rev="${VERSION}"
      export KIBOT_VAR_rev_pcb="${VERSION}"
      export KIBOT_VAR_name=$(echo "${PROJECT_NAME}" | cut -d"-" -f1)
      echo "Using version: $VERSION, name: $KIBOT_VAR_name"

      # Find panel configuration
      PANEL_JSON=$(find ${PROJECT_NAME} -name "*_panel.json" 2>/dev/null | head -1)

      if [ -n "$PANEL_JSON" ] && [ -f "$PANEL_JSON" ]; then
        echo "Found panel config: $PANEL_JSON"

        # Create panel directory
        PANEL_NAME=$(basename "${PANEL_JSON%.json}")
        mkdir -p panels/${PANEL_NAME}
        PCB_FILE="${PROJECT_NAME}/${PROJECT_NAME}.kicad_pcb"

        # Run pre-panel script if exists
        if [ -f ".gitlab/.scripts/pre_panel.py" ]; then
          python3 .gitlab/.scripts/pre_panel.py $PCB_FILE $PANEL_JSON
        fi

        # Generate panel
        kikit panelize -p $PANEL_JSON $PCB_FILE panels/${PANEL_NAME}/${PANEL_NAME}.kicad_pcb

        # Copy required files for panel
        cp .gitlab/templates/micromelon_default/micromelon_default.kicad_sch panels/${PANEL_NAME}/${PANEL_NAME}.kicad_sch 2>/dev/null || true
        cp ${PROJECT_NAME}/fp-lib-table panels/${PANEL_NAME}/ 2>/dev/null || true
        # Copy libs folder for 3D models
        cp -r ${PROJECT_NAME}/libs panels/${PANEL_NAME}/ 2>/dev/null || true

        # Run post-panel script if exists
        if [ -f ".gitlab/.scripts/post_panel.py" ]; then
          python3 .gitlab/.scripts/post_panel.py panels/${PANEL_NAME}/${PANEL_NAME}.kicad_pro $PCB_FILE
        fi

        # Generate panel gerbers
        # Set name variable explicitly for panel (extract part number from project name)
        export KIBOT_VAR_name=$(echo "${PROJECT_NAME}" | cut -d"-" -f1)
        cd panels
        kibot -e ${PANEL_NAME}/${PANEL_NAME}.kicad_sch \
          -c $CI_PROJECT_DIR/.gitlab/configs/panel.kibot.yaml \
          -d $CI_PROJECT_DIR/Fabrication/panels
        cd $CI_PROJECT_DIR

        mv Fabrication/panels/*.zip Fabrication/ 2>/dev/null || true
      else
        echo "No panel configuration found, skipping panel generation"
      fi

# =============================================================================
# Stage: InvenTree
# =============================================================================
.inventree_base:
  stage: inventree
  image: python:3.11-slim
  before_script:
    - |
      cd hfsntree
      pip install -e . --quiet
      cd $CI_PROJECT_DIR

inventree_dev:
  extends: .inventree_base
  rules:
    - !reference [.dev_rules, rules]
  needs:
    - job: extract_version
      artifacts: true
    - job: generate_schematic
      artifacts: true
    - job: generate_bom
      artifacts: true
    - job: generate_3d
      artifacts: true
    - job: generate_gerbers
      artifacts: true
    - job: generate_position
      artifacts: true
    - job: generate_panel
      artifacts: true
      optional: true
  script:
    - |
      echo "Running InvenTree upload for dev branch (revision: dev)"
      cd hfsntree
      python main.py batch $CI_PROJECT_DIR/Fabrication --version-override dev -y

inventree_main:
  extends: .inventree_base
  rules:
    - !reference [.main_rules, rules]
  needs:
    - job: extract_version
      artifacts: true
    - job: generate_schematic
      artifacts: true
    - job: generate_bom
      artifacts: true
    - job: generate_3d
      artifacts: true
    - job: generate_gerbers
      artifacts: true
    - job: generate_position
      artifacts: true
    - job: generate_panel
      artifacts: true
      optional: true
  artifacts:
    when: always
    paths:
      - Fabrication/
    expire_in: 1 week
  script:
    - |
      cd hfsntree

      # MRs targeting main upload to "dev" revision for pre-ordering parts
      # Actual commits to main upload to the versioned revision
      if [[ "$CI_PIPELINE_SOURCE" == "merge_request_event" ]]; then
        echo "Running InvenTree upload for MR (revision: dev)"
        python main.py batch $CI_PROJECT_DIR/Fabrication --version-override dev -y
      else
        echo "Running InvenTree upload for main branch (revision: $VERSION)"
        python main.py batch $CI_PROJECT_DIR/Fabrication -y

        # Generate Samsung P&P files only on actual release
        echo "Generating Samsung pick-and-place files..."
        python main.py samsung $CI_PROJECT_DIR/Fabrication

        # Deactivate old revisions (keep only current release + dev)
        echo "Deactivating old InvenTree revisions..."
        python main.py deactivate-old $CI_PROJECT_DIR/Fabrication
      fi
      cd $CI_PROJECT_DIR

# =============================================================================
# Stage: Release
# =============================================================================
upload_packages:
  stage: release
  rules:
    - if: $CI_COMMIT_TAG
      when: never
    - if: $CI_COMMIT_BRANCH == "main"
  needs:
    - job: extract_version
      artifacts: true
    - job: inventree_main
      artifacts: true
  script:
    - apt-get update && apt-get -y install zip
    - |
      # Create single release zip with all fabrication outputs
      RELEASE_ZIP="${PROJECT_NAME}_${VERSION}.zip"
      zip -r "Fabrication/${RELEASE_ZIP}" Fabrication/
      echo "Created release package: ${RELEASE_ZIP}"
    - echo 'import urllib.request,sys,os;f=open(sys.argv[1],"rb");req=urllib.request.Request(sys.argv[2],data=f.read(),method="PUT");req.add_header("JOB-TOKEN",os.environ["CI_JOB_TOKEN"]);urllib.request.urlopen(req);print("Uploaded",sys.argv[1])' > /tmp/upload.py
    - |
      # Upload release package
      RELEASE_ZIP="${PROJECT_NAME}_${VERSION}.zip"
      url="${PACKAGE_REGISTRY_URL}/${VERSION}/${RELEASE_ZIP}"
      echo "Uploading: Fabrication/${RELEASE_ZIP} to $url"
      python3 /tmp/upload.py "Fabrication/${RELEASE_ZIP}" "$url"
  artifacts:
    when: always
    paths:
      - Fabrication/
    expire_in: 1 week

create_release:
  stage: release
  image: registry.gitlab.com/gitlab-org/release-cli:latest
  rules:
    - if: $CI_COMMIT_TAG
      when: never
    - if: $CI_COMMIT_BRANCH == "main"
  needs:
    - job: extract_version
      artifacts: true
    - job: upload_packages
      artifacts: true
  script:
    - echo "Creating release for version $VERSION"
  release:
    tag_name: '$VERSION'
    name: 'Release $VERSION'
    description: 'Automated release for $VERSION'