ssh암복호화 해결 해야함

ansible/migrations/0001_initial.py

@@ -0,0 +1,26 @@
+# Generated by Django 4.2.14 on 2025-05-20 07:34
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='AnsibleTask',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=200)),
+                ('playbook_content', models.TextField()),
+                ('inventory_content', models.TextField()),
+                ('status', models.CharField(default='pending', max_length=50)),
+                ('output', models.TextField(blank=True)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+            ],
+        ),
+    ]

ansible/models.py

@@ -1,3 +1,12 @@
 from django.db import models
 
-# Create your models here.
+class AnsibleTask(models.Model):
+    name = models.CharField(max_length=200)
+    playbook_content = models.TextField()          # ✅ YAML 문자열
+    inventory_content = models.TextField()         # ✅ 인벤토리 형식 문자열
+    status = models.CharField(max_length=50, default='pending')  # 'pending', 'running', 'success', 'failed', 'error'
+    output = models.TextField(blank=True)          # ✅ 실행 결과 로그
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f"{self.name} ({self.status})"

ansible/serializers.py

@@ -0,0 +1,32 @@
+from rest_framework import serializers
+from .models import AnsibleTask
+
+
+# ✅ 기본 Serializer: 목록 / 생성용
+class AnsibleTaskSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = AnsibleTask
+        fields = [
+            "id",
+            "name",
+            "playbook_content",
+            "inventory_content",
+            "status",
+            "output",
+            "created_at",
+        ]
+        read_only_fields = ("id", "status", "output", "created_at")
+
+
+# ✅ 상세용 Serializer: 실행 결과만 확인
+class AnsibleTaskDetailSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = AnsibleTask
+        fields = [
+            "id",
+            "name",
+            "status",
+            "output",
+            "created_at",
+        ]
+        read_only_fields = fields

ansible/services.py

@@ -0,0 +1,49 @@
+# services.py
+import os
+import tempfile
+import subprocess
+from .models import AnsibleTask
+from django.conf import settings
+
+
+def run_ansible_job(task: AnsibleTask, ssh_key: str):
+    task.status = "running"
+    task.save()
+
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, mode="w") as playbook_file, \
+             tempfile.NamedTemporaryFile(delete=False, mode="w") as inventory_file, \
+             tempfile.NamedTemporaryFile(delete=False, mode="w") as private_key_file:
+
+            playbook_file.write(task.playbook_content.strip())
+            playbook_file.close()
+
+            inventory_file.write(task.inventory_content.strip())
+            inventory_file.close()
+
+            private_key_file.write(ssh_key.strip() + "\n")
+            private_key_file.close()
+            os.chmod(private_key_file.name, 0o600)
+
+            command = [
+                "ansible-playbook",
+                playbook_file.name,
+                "-i", inventory_file.name,
+                "--private-key", private_key_file.name,
+                "-u", "ubuntu",
+            ]
+
+            result = subprocess.run(command, capture_output=True, text=True)
+            task.status = "success" if result.returncode == 0 else "failed"
+            task.output = result.stdout + "\n" + result.stderr
+
+    except Exception as e:
+        task.status = "error"
+        task.output = f"\u274c 실행 중 예외 발생: {str(e)}"
+
+    finally:
+        for f in [playbook_file.name, inventory_file.name, private_key_file.name]:
+            if os.path.exists(f):
+                os.remove(f)
+
+        task.save()

ansible/urls.py

@@ -0,0 +1,10 @@
+from django.urls import path, include
+from rest_framework.routers import DefaultRouter
+from .views import AnsibleTaskViewSet
+
+router = DefaultRouter()
+router.register(r'tasks', AnsibleTaskViewSet, basename='ansibletask')
+
+urlpatterns = [
+    path('', include(router.urls)),
+]

ansible/views.py

@@ -1,3 +1,101 @@
-from django.shortcuts import render
+from rest_framework.views import APIView
+from rest_framework.response import Response
+from rest_framework import status
+from rest_framework.permissions import IsAuthenticated
+from rest_framework_simplejwt.views import TokenObtainPairView
 
-# Create your views here.
+from .serializers import RegisterSerializer, CustomTokenObtainPairSerializer
+from cryptography.fernet import Fernet
+from django.conf import settings
+import base64
+import hashlib
+
+# Fernet 키를 settings.SECRET_KEY에서 파생
+hashed = hashlib.sha256(settings.SECRET_KEY.encode()).digest()
+fernet_key = base64.urlsafe_b64encode(hashed[:32])
+fernet = Fernet(fernet_key)
+
+
+class RegisterView(APIView):
+    def post(self, request):
+        serializer = RegisterSerializer(data=request.data)
+        if serializer.is_valid():
+            user = serializer.save()
+            return Response({"message": "User registered successfully."}, status=status.HTTP_201_CREATED)
+        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
+
+
+class MeView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request):
+        user = request.user
+        serializer = RegisterSerializer(user)
+        return Response(serializer.data)
+
+    def put(self, request):
+        user = request.user
+        serializer = RegisterSerializer(user, data=request.data, partial=True)
+        if serializer.is_valid():
+            serializer.save()
+            return Response(serializer.data)
+        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
+
+
+class CustomTokenObtainPairView(TokenObtainPairView):
+    serializer_class = CustomTokenObtainPairSerializer
+
+
+class SSHKeyUploadView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    def post(self, request):
+        private_key = request.data.get("private_key")
+        key_name = request.data.get("key_name")
+
+        if not private_key or not key_name:
+            return Response(
+                {"error": "private_key와 key_name 모두 필요합니다."},
+                status=status.HTTP_400_BAD_REQUEST
+            )
+
+        user = request.user
+        try:
+            encrypted_key = fernet.encrypt(private_key.encode()).decode()  # ✅ decode 추가
+            user.encrypted_private_key = encrypted_key
+            user.encrypted_private_key_name = key_name
+            user.save(update_fields=["encrypted_private_key", "encrypted_private_key_name"])
+            return Response({"message": "SSH key 저장 완료."})
+        except Exception as e:
+            return Response({"error": str(e)}, status=500)
+
+    def delete(self, request):
+        user = request.user
+        user.encrypted_private_key = None
+        user.encrypted_private_key_name = None
+        user.last_used_at = None
+        user.save(update_fields=["encrypted_private_key", "encrypted_private_key_name", "last_used_at"])
+        return Response({"message": "SSH key deleted."}, status=200)
+
+
+class SSHKeyInfoView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request):
+        user = request.user
+        return Response({
+            "has_key": bool(user.encrypted_private_key),
+            "encrypted_private_key_name": user.encrypted_private_key_name,
+            "last_used_at": user.last_used_at
+        })
+
+
+# ✅ 실제 암호화된 키를 반환하는 API
+class SSHKeyRetrieveView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request):
+        user = request.user
+        if not user.encrypted_private_key:
+            return Response({"error": "SSH 키가 등록되어 있지 않습니다."}, status=404)
+        return Response({"ssh_key": user.encrypted_private_key})