procontor.sh

#!/bin/bash

# Definir cores como constantes
WHITE='\033[1;37m'
RED='\033[0;31m'
BLUE='\033[0;36m'
CLEAN='\033[0m'

error_exit() {
    echo -e "${RED}Error: $1${CLEAN}" >&2
    exit 1
}

show_help() {
    echo -e "${WHITE}Usage:${CLEAN}"
    echo -e "${WHITE}  ./procontor.sh [OPTIONS]${CLEAN}"
    echo -e "${WHITE}Options:${CLEAN}"
    echo -e "${BLUE}  -h, --help${CLEAN}             Show this help message"
    echo -e "${BLUE}  -p, --project PROJECT${CLEAN}  Specify the project name"
    echo -e "${BLUE}  -s, --size SIZE${CLEAN}        Specify the container size (e.g., 500M, 5G)"
    echo -e "${WHITE}Description:${CLEAN}"
    echo -e "  Procontor is a Project Container Creator designed to create LUKS-encrypted containers for handling sensitive projects and subsequently mount them for further use. Please remember the password you've set for your container!"
    echo -e "${WHITE}Note:${CLEAN}"
    echo -e "  The ideal container naming should follow this pattern: year-month-day_projectType_client_yourInitials. For instance: 2023-10-02_Web_evilcorp_kd"
    exit 0
}

if [ "$#" -eq 0 ]; then
    show_help
fi

project=""
size=""

while [[ "$#" -gt 0 ]]; do
    case "$1" in
        -h|--help)
            show_help
            ;;
        -p|--project)
            if [ -n "$2" ]; then
                project="$2"
                shift 2
            else
                error_exit "No project name provided after -p or --project."
            fi
            ;;
        -s|--size)
            if [ -n "$2" ]; then
                size="$2"
                shift 2
            else
                error_exit "No size provided after -s or --size."
            fi
            ;;
        *)
            echo -e "${RED}Unknown option: $1${CLEAN}"
            show_help
            ;;
    esac
done

# Verificar se os parâmetros obrigatórios foram fornecidos
if [ -z "$project" ]; then
    error_exit "Project name is required. Use -p or --project to specify it."
fi

if [ -z "$size" ]; then
    error_exit "Container size is required. Use -s or --size to specify it."
fi

# Verificar se o valor após a flag -s é válido
if [[ ! "$size" =~ ^[0-9]+[GgMm]$ ]]; then
    error_exit "Invalid size format. Please use 'G' for gigabytes or 'M' for megabytes."
fi

if ! command -v cryptsetup &> /dev/null; then
    error_exit "cryptsetup is not installed. Please install it."
fi

create_container() {
    echo -e "${BLUE}01 Creating a ${size} container for $project${CLEAN}"
    dd if=/dev/zero of="$project" bs=1 count=0 seek="$size" || error_exit "Failed to create the container."
    echo ""
}

encrypt_container() {
    echo -e "${BLUE}02 Encrypting the $project container with LUKS${CLEAN}"
    sudo cryptsetup -y --cipher aes-xts-plain64 --hash sha512 --key-size 512 -v luksFormat "$project" || error_exit "Failed to encrypt the container."
    echo ""
}

create_device() {
    echo -e "${BLUE}03 Creating the device as project-device${CLEAN}"
    sudo cryptsetup luksOpen "$project" project-device || error_exit "Failed to create the device."
    echo -e "${BLUE}The project-device is in /dev/mapper:${CLEAN}"
    ls /dev/mapper/
    echo ""
}

create_filesystem() {
    echo -e "${BLUE}04 Creating a file system in the device${CLEAN}"
    sudo mkfs.xfs /dev/mapper/project-device || error_exit "Failed to create the filesystem."
    echo ""
}

mount_device() {
    echo -e "${BLUE}05 Mounting the device at /mnt/project-mount${CLEAN}"
    sudo mount /dev/mapper/project-device /mnt/project-mount/ || error_exit "Failed to mount the device."
    lsblk
    echo ""
}

set_ownership() {
    echo -e "${BLUE}06 Changing the project-mount ownership to $USER ${CLEAN}"
    sudo chown $USER -R /mnt/project-mount/ || error_exit "Failed to change ownership."
}

while true
do
    echo -e "                                  _             "
    echo -e "  _ __  _ __ ___   ___ ___  _ __ | |_ ___  _ __ "
    echo -e " | '_ \| '__/ _ \ / __/ _ \| '_ \| __/ _ \| '__|"
    echo -e " | |_) | | | (_) | (_| (_) | | | | || (_) | |   "
    echo -e " | .__/|_|  \___/ \___\___/|_| |_|\__\___/|_|   "
    echo -e " |_|                                            "
    echo ""
    echo -e "     by k4rkarov (v3.1)"
    echo ""
    echo ""
    create_container
    encrypt_container
    create_device
    create_filesystem
    mount_device
    set_ownership
    echo ""
    echo ""
    echo -e "${WHITE}All done.${CLEAN}"
    echo ""
    break
done