works, troubleshooting

auth.py

@@ -4,12 +4,15 @@ LOGIN_URL = "http://localhost:7777/user/login"
 EDIT_MILESTONE_URL = "http://localhost:7777/setyotontowi/experiments/milestones/1/edit"
 session = requests.Session()
 
-def login(username, password):
+USERNAME = "setyotontowi"
+PASSWORD = "nasipadang"
+
+def login():
     """Log in and store session cookies."""
     print("Logging in...")
     payload = {
-        "user_name": username,
-        "password": password
+        "user_name": USERNAME,
+        "password": PASSWORD
     }
     response = session.post(LOGIN_URL, data=payload)
     if response.status_code == 200:

content.md

@@ -7,4 +7,5 @@ Team Sigma Σ
 - [ ] Lambang: [Klinik | PCARE - Analisis Bridging PCARE ke SIMRS](http://192.168.1.33:5050/apps/team/detail?teamId=22&updateTeamId=3619)
 - [x] Hilyas: [RSI Banjarnegara | Tambah filter sesuai kelas pasien di tab tindakan edoc seperti di SIMRS](http://192.168.1.33:5050/apps/team/detail?teamId=8&updateTeamId=2580)
 - [x] Setyotontowi: [Add Something](http://192.168.1.33:5050/apps/team/detail?teamID=20&updateTeamId=1232)
+- [x] Setyotontowi: [Add Something](http://192.168.1.33:5050/apps/team/detail?teamID=40&updateTeamId=1232)
 <!-- /feature_sigma -->

gogmilestone2.py

@@ -1,12 +1,10 @@
 from flask import Flask, request, jsonify
 from auth import login, get_csrf_token
-from milestone import get_milestone, append_item_to_identifier
+from milestone import update_milestone, get_milestone, append_item_to_identifier
 from utils import find_username_id, generate_evoluz_link, find_task_type
 
 app = Flask(__name__)
 
-USERNAME = "setyotontowi"
-PASSWORD = "nasipadang"
 EVOLUZ_URL = "http://192.168.1.33:5050/apps/team/detail?"
 
 @app.route('/webhook', methods=['POST'])
@@ -47,7 +45,7 @@ def index():
     return "Webhook server is running."
 
 if __name__ == '__main__':
-    if login(USERNAME, PASSWORD):
+    if login():
         print("Initial login successful.")
     else:
         print("Initial login failed.")

milestone.py

@@ -1,21 +1,77 @@
 import re
 import html
+from utils import find_squad
+from auth import login, get_csrf_token, EDIT_MILESTONE_URL, session
 
-def get_milestone(session, edit_milestone_url):
+def update_milestone(pr_data):
+    login()
+    """Update milestone with PR title using CSRF token."""
+    csrf_token = get_csrf_token()
+    if not csrf_token:
+        print("Failed to get CSRF token.")
+        return False
+
+    meta = get_milestone(session)
+    content = create_milestone_content(meta.get("content"), pr_data)
+
+    print(f"Updating milestone with content:\n")
+
+    with open("content.md", "w") as file:
+        file.write(content)
+
+    payload = {
+        "_csrf": csrf_token,
+        "title": meta.get("title"),
+        "content": content,
+        "deadline": meta.get("deadline")
+    }
+    headers = {
+        "Content-Type": "application/x-www-form-urlencoded"
+    }
+
+
+    response = session.post(EDIT_MILESTONE_URL, data=payload, headers=headers)
+
+    if response.status_code == 200:
+        print("Milestone updated successfully!")
+        return True
+    elif response.status_code == 401:  # Unauthorized → Need to re-login
+        print("Session expired, re-logging in...")
+        if login():
+            return update_milestone(pr_data)
+    else:
+        print(f"Failed to update milestone: {response.status_code} - {response.text}")
+        return False
+
+def create_milestone_content(content, pr_data):
+    if content:
+        # From PR Data, get branch type, and username to assign it to identifier. like feature owi, then it should be
+        # in feature_lambda
+        type_task = pr_data.get("task_type")
+        squad = "sigma"
+        identifier = f'{type_task}_{squad}'
+        content = append_item_to_identifier(content, identifier, pr_data)
+        return f"{content}\n"
+    else:
+        return f"- [] {pr_data['title']}"
+
+def get_milestone(session):
     """Fetch milestone details."""
-    response = session.get(edit_milestone_url)
+    response = session.get(EDIT_MILESTONE_URL)
     if response.status_code == 200:
-        title_match = re.search(r'<input[^>]*name=["\']title["\'][^>]*value=["\'](.*?)["\']', response.text, re.IGNORECASE)
+        title_match = re.search(r'<input[^>]*name=["\']title["\'][^>]*value=["\'](.* ?)["\']', response.text, re.IGNORECASE)
         content_match = re.search(r'<textarea[^>]*name=["\']content["\'][^>]*>(.*?)</textarea>', response.text, re.IGNORECASE | re.DOTALL)
+        deadline_match = re.search(r'<input[^>]*name=["\']deadline["\'][^>]*value=["\'](.*?)["\']', response.text, re.IGNORECASE)
         title = title_match.group(1) if title_match else None
         content = content_match.group(1).strip() if content_match else None
-        return {'title': title, 'content': content}
+        deadline = deadline_match.group(1) if deadline_match else None
+        return {'title': title, 'content': content, 'deadline': deadline}
     else:
         print(f"Failed to fetch milestone: {response.status_code}")
         return None
 
 def append_item_to_identifier(encoded_content, identifier, new_item):
-    """Append a new item to the block identified by the given identifier."""
+    # Extract the block delimited by the identifier markers
     content = html.unescape(encoded_content)
     block_pattern = rf'(<!--\s*{identifier}\s*-->)(.*?)(<!--\s*/{identifier}\s*-->)'
     block_match = re.search(block_pattern, content, re.DOTALL)
@@ -23,12 +79,18 @@ def append_item_to_identifier(encoded_content, identifier, new_item):
         print(f"Block with identifier '{identifier}' not found.")
         return False
 
+    # Extract the parts of the block
     start_marker = block_match.group(1)
     block_content = block_match.group(2).strip()
     end_marker = block_match.group(3)
 
+    # Append the new item to the block content
     formatted_item = f"- [x] {new_item['username']}: [{new_item['title']}]({new_item['link_evoluz']})"
     updated_block_content = f"{block_content}\n{formatted_item}"
 
+    # Reconstruct the content with the updated block
     updated_content = f"{start_marker}\n{updated_block_content}\n{end_marker}"
-    return updated_content
+    content = content.replace(block_match.group(0), updated_content)
+
+    print(f"New item appended to '{identifier}' block.")
+    return content

testcase.md

@@ -0,0 +1,12 @@
+### Happy Path
+- [ ] Create PR, it should add the data to milestone
+
+### Side case
+- [ ] Sprint Transition
+- [ ] Add new people
+
+### Negative
+- [ ] Unmerged PR
+
+### Thinking
+- if repo is production

uninstall.sh

@@ -0,0 +1,36 @@
+#!/bin/bash
+
+# filepath: /home/setyotontowi/BackendProject/gogsmilestone/uninstall.sh
+
+# Dynamically determine the project directory
+PROJECT_DIR=$(dirname "$(realpath "$0")")
+
+echo "Uninstalling the project from $PROJECT_DIR..."
+
+# Step 1: Stop and disable the systemd service
+SERVICE_FILE="/etc/systemd/system/gogmilestone.service"
+if [ -f "$SERVICE_FILE" ]; then
+    echo "Stopping the gogmilestone service..."
+    sudo systemctl stop gogmilestone
+
+    echo "Disabling the gogmilestone service..."
+    sudo systemctl disable gogmilestone
+
+    echo "Removing the gogmilestone service file..."
+    sudo rm -f "$SERVICE_FILE"
+
+    echo "Reloading systemd daemon..."
+    sudo systemctl daemon-reload
+else
+    echo "Service file not found. Skipping service removal."
+fi
+
+# Step 2: Delete the virtual environment
+if [ -d "$PROJECT_DIR/venv" ]; then
+    echo "Removing the virtual environment..."
+    rm -rf "$PROJECT_DIR/venv"
+else
+    echo "Virtual environment not found. Skipping removal."
+fi
+
+echo "Uninstallation complete."

utils.py

@@ -35,4 +35,23 @@ def find_task_type(branch_name):
             return None
     except Exception as e:
         print(f"Error extracting task type: {e}")
+        return None
+
+def find_squad(username):
+    """Find the squad for a given username from the username_squad file."""
+    file_path = "username_squad"
+    try:
+        with open(file_path, "r", encoding="utf-8") as file:
+            for line in file:
+                if line.strip():
+                    user, squad = line.strip().split(":")
+                    if user.lower() == username.lower():
+                        return squad.lower()
+        print(f"Username '{username}' not found in squad file.")
+        return None
+    except FileNotFoundError:
+        print(f"Squad file not found: {file_path}")
+        return None
+    except Exception as e:
+        print(f"An error occurred while finding squad: {e}")
         return None