mirror of https://github.com/jumpserver/jumpserver
246 lines
9.2 KiB
Python
246 lines
9.2 KiB
Python
import os
|
|
import shutil
|
|
import zipfile
|
|
|
|
from django.conf import settings
|
|
from django.shortcuts import get_object_or_404
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import status
|
|
|
|
from common.exceptions import JMSException
|
|
from orgs.mixins.api import OrgBulkModelViewSet
|
|
from rbac.permissions import RBACPermission
|
|
from ..exception import PlaybookNoValidEntry
|
|
from ..models import Playbook
|
|
from ..serializers.playbook import PlaybookSerializer
|
|
|
|
__all__ = ["PlaybookViewSet", "PlaybookFileBrowserAPIView"]
|
|
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
|
|
|
|
def unzip_playbook(src, dist):
|
|
fz = zipfile.ZipFile(src, 'r')
|
|
for file in fz.namelist():
|
|
fz.extract(file, dist)
|
|
|
|
|
|
class PlaybookViewSet(OrgBulkModelViewSet):
|
|
serializer_class = PlaybookSerializer
|
|
permission_classes = (RBACPermission,)
|
|
model = Playbook
|
|
search_fields = ('name', 'comment')
|
|
|
|
def perform_destroy(self, instance):
|
|
if instance.job_set.exists():
|
|
raise JMSException(code='playbook_has_job', detail={"msg": _("Currently playbook is being used in a job")})
|
|
instance_id = instance.id
|
|
super().perform_destroy(instance)
|
|
dest_path = os.path.join(settings.DATA_DIR, "ops", "playbook", instance_id.__str__())
|
|
shutil.rmtree(dest_path)
|
|
|
|
def get_queryset(self):
|
|
queryset = super().get_queryset()
|
|
queryset = queryset.filter(creator=self.request.user)
|
|
return queryset
|
|
|
|
def perform_create(self, serializer):
|
|
instance = serializer.save()
|
|
if 'multipart/form-data' in self.request.headers['Content-Type']:
|
|
src_path = os.path.join(settings.MEDIA_ROOT, instance.path.name)
|
|
dest_path = os.path.join(settings.DATA_DIR, "ops", "playbook", instance.id.__str__())
|
|
try:
|
|
unzip_playbook(src_path, dest_path)
|
|
except RuntimeError as e:
|
|
raise JMSException(code='invalid_playbook_file', detail={"msg": "Unzip failed"})
|
|
|
|
if 'main.yml' not in os.listdir(dest_path):
|
|
raise PlaybookNoValidEntry
|
|
|
|
else:
|
|
if instance.create_method == 'blank':
|
|
dest_path = os.path.join(settings.DATA_DIR, "ops", "playbook", instance.id.__str__())
|
|
os.makedirs(dest_path)
|
|
with open(os.path.join(dest_path, 'main.yml'), 'w') as f:
|
|
f.write('## write your playbook here')
|
|
|
|
|
|
class PlaybookFileBrowserAPIView(APIView):
|
|
rbac_perms = ()
|
|
permission_classes = (RBACPermission,)
|
|
rbac_perms = {
|
|
'GET': 'ops.change_playbook',
|
|
'POST': 'ops.change_playbook',
|
|
'DELETE': 'ops.change_playbook',
|
|
'PATCH': 'ops.change_playbook',
|
|
}
|
|
protected_files = ['root', 'main.yml']
|
|
|
|
def get(self, request, **kwargs):
|
|
playbook_id = kwargs.get('pk')
|
|
playbook = get_object_or_404(Playbook, id=playbook_id)
|
|
work_path = playbook.work_dir
|
|
file_key = request.query_params.get('key', '')
|
|
if file_key:
|
|
file_path = os.path.join(work_path, file_key)
|
|
with open(file_path, 'r') as f:
|
|
try:
|
|
content = f.read()
|
|
except UnicodeDecodeError:
|
|
content = _('Unsupported file content')
|
|
return Response({'content': content})
|
|
else:
|
|
expand_key = request.query_params.get('expand', '')
|
|
nodes = self.generate_tree(playbook, work_path, expand_key)
|
|
return Response(nodes)
|
|
|
|
def post(self, request, **kwargs):
|
|
playbook_id = kwargs.get('pk')
|
|
playbook = get_object_or_404(Playbook, id=playbook_id)
|
|
work_path = playbook.work_dir
|
|
|
|
parent_key = request.data.get('key', '')
|
|
if parent_key == 'root':
|
|
parent_key = ''
|
|
if os.path.dirname(parent_key) == 'root':
|
|
parent_key = os.path.basename(parent_key)
|
|
full_path = os.path.join(work_path, parent_key)
|
|
|
|
is_directory = request.data.get('is_directory', False)
|
|
content = request.data.get('content', '')
|
|
name = request.data.get('name', '')
|
|
|
|
def find_new_name(p, is_file=False):
|
|
if not p:
|
|
if is_file:
|
|
p = 'new_file.yml'
|
|
else:
|
|
p = 'new_dir'
|
|
np = os.path.join(full_path, p)
|
|
n = 0
|
|
while os.path.exists(np):
|
|
n += 1
|
|
np = os.path.join(full_path, '{}({})'.format(p, n))
|
|
return np
|
|
|
|
if is_directory:
|
|
new_file_path = find_new_name(name)
|
|
os.makedirs(new_file_path)
|
|
else:
|
|
new_file_path = find_new_name(name, True)
|
|
with open(new_file_path, 'w') as f:
|
|
f.write(content)
|
|
|
|
relative_path = os.path.relpath(os.path.dirname(new_file_path), work_path)
|
|
new_node = {
|
|
"name": os.path.basename(new_file_path),
|
|
"title": os.path.basename(new_file_path),
|
|
"id": os.path.join(relative_path, os.path.basename(new_file_path))
|
|
if not os.path.join(relative_path, os.path.basename(new_file_path)).startswith('.')
|
|
else os.path.basename(new_file_path),
|
|
"isParent": is_directory,
|
|
"pId": relative_path if not relative_path.startswith('.') else 'root',
|
|
"open": True,
|
|
}
|
|
if not is_directory:
|
|
new_node['iconSkin'] = 'file'
|
|
return Response(new_node)
|
|
|
|
def patch(self, request, **kwargs):
|
|
playbook_id = kwargs.get('pk')
|
|
playbook = get_object_or_404(Playbook, id=playbook_id)
|
|
work_path = playbook.work_dir
|
|
|
|
file_key = request.data.get('key', '')
|
|
new_name = request.data.get('new_name', '')
|
|
|
|
if file_key in self.protected_files and new_name:
|
|
return Response({'msg': '{} can not be rename'.format(file_key)}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
if os.path.dirname(file_key) == 'root':
|
|
file_key = os.path.basename(file_key)
|
|
|
|
content = request.data.get('content', '')
|
|
is_directory = request.data.get('is_directory', False)
|
|
|
|
if not file_key or file_key == 'root':
|
|
return Response(status=status.HTTP_400_BAD_REQUEST)
|
|
file_path = os.path.join(work_path, file_key)
|
|
|
|
# rename
|
|
if new_name:
|
|
new_file_path = os.path.join(os.path.dirname(file_path), new_name)
|
|
if new_file_path == file_path:
|
|
return Response(status=status.HTTP_200_OK)
|
|
if os.path.exists(new_file_path):
|
|
return Response({'msg': '{} already exists'.format(new_name)}, status=status.HTTP_400_BAD_REQUEST)
|
|
os.rename(file_path, new_file_path)
|
|
# edit content
|
|
else:
|
|
if not is_directory:
|
|
with open(file_path, 'w') as f:
|
|
f.write(content)
|
|
return Response(status=status.HTTP_200_OK)
|
|
|
|
def delete(self, request, **kwargs):
|
|
playbook_id = kwargs.get('pk')
|
|
playbook = get_object_or_404(Playbook, id=playbook_id)
|
|
work_path = playbook.work_dir
|
|
file_key = request.query_params.get('key', '')
|
|
if not file_key:
|
|
return Response({'msg': 'key is required'}, status=status.HTTP_400_BAD_REQUEST)
|
|
if file_key in self.protected_files:
|
|
return Response({'msg': ' {} can not be delete'.format(file_key)}, status=status.HTTP_400_BAD_REQUEST)
|
|
file_path = os.path.join(work_path, file_key)
|
|
if os.path.isdir(file_path):
|
|
shutil.rmtree(file_path)
|
|
else:
|
|
os.remove(file_path)
|
|
return Response({'msg': 'ok'})
|
|
|
|
@staticmethod
|
|
def generate_tree(playbook, root_path, expand_key=None):
|
|
nodes = [{
|
|
"name": playbook.name,
|
|
"title": playbook.name,
|
|
"id": 'root',
|
|
"isParent": True,
|
|
"open": True,
|
|
"pId": '',
|
|
"temp": False
|
|
}]
|
|
for path, dirs, files in os.walk(root_path):
|
|
dirs.sort()
|
|
files.sort()
|
|
|
|
relative_path = os.path.relpath(path, root_path)
|
|
for d in dirs:
|
|
node = {
|
|
"name": d,
|
|
"title": d,
|
|
"id": os.path.join(relative_path, d) if not os.path.join(relative_path, d).startswith(
|
|
'.') else d,
|
|
"isParent": True,
|
|
"open": True,
|
|
"pId": relative_path if not relative_path.startswith('.') else 'root',
|
|
"temp": False
|
|
}
|
|
if expand_key == node['id']:
|
|
node['open'] = True
|
|
nodes.append(node)
|
|
for f in files:
|
|
node = {
|
|
"name": f,
|
|
"title": f,
|
|
"iconSkin": 'file',
|
|
"id": os.path.join(relative_path, f) if not os.path.join(relative_path, f).startswith(
|
|
'.') else f,
|
|
"isParent": False,
|
|
"open": False,
|
|
"pId": relative_path if not relative_path.startswith('.') else 'root',
|
|
"temp": False
|
|
}
|
|
nodes.append(node)
|
|
return nodes
|