#!/usr/bin/env python3
from flask import request
from flask_restful import Resource
import subprocess
import json

from selfprivacy_api.resources.services import api

# List all restic backups
class ListAllBackups(Resource):
    def get(self):
        backupListingCommand = """
            restic -r b2:{0}:/sfbackup snapshots --password-file /var/lib/restic/rpass --json
        """.format(
            request.headers.get("X-Repository-Name")
        )

        backupListingProcessDescriptor = subprocess.Popen(
            backupListingCommand,
            shell=True,
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT,
        )

        snapshotsList = backupListingProcessDescriptor.communicate()[0]

        return snapshotsList.decode("utf-8")


# Create a new restic backup
class AsyncCreateBackup(Resource):
    def put(self):
        backupCommand = """
            restic -r b2:{0}:/sfbackup --verbose backup /var --password-file /var/lib/restic/rpass > /tmp/backup.log
        """.format(
            request.headers.get("X-Repository-Name")
        )

        backupProcessDescriptor = subprocess.Popen(
            backupCommand, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT
        )

        return {
            "status": 0,
            "message": "Backup creation has started",
        }

class CheckBackupStatus(Resource):
    def get(self):
        backupStatusCheckCommand = """
            tail -1 /tmp/backup.log
        """

        backupStatusCheckProcessDescriptor = subprocess.Popen(
            backupStatusCheckCommand, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT
        )

        backupProcessStatus = backupStatusCheckProcessDescriptor.communicate()[0].decode("utf-8")

        try:
            json.loads(backupProcessStatus)
        except ValueError:
            return {
                "message": backupProcessStatus
            }
        return backupProcessStatus


api.add_resource(ListAllBackups, "/restic/backup/list")
api.add_resource(AsyncCreateBackup, "/restic/backup/create")
api.add_resource(CheckBackupStatus, "/restic/backup/status")