#!/usr/bin/env python3
# encoding: utf-8
# === This file is part of Calamares - <http://github.com/calamares> ===
#
#   Copyright 2014, Aurélien Gâteau <agateau@kde.org>
#
#   Calamares is free software: you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.
#
#   Calamares is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#   GNU General Public License for more details.
#
#   You should have received a copy of the GNU General Public License
#   along with Calamares. If not, see <http://www.gnu.org/licenses/>.

import os
import subprocess
import sys
import tempfile

import libcalamares


HEADER = """# /etc/fstab: static file system information.
#
# Use 'blkid' to print the universally unique identifier for a device; this may
# be used with UUID= as a more robust way to name devices that works even if
# disks are added and removed. See fstab(5).
#
# <file system>                           <mount point>  <type>  <options>  <dump>  <pass>"""

FS_MAP = {
    "fat": "vfat",
    "linuxswap": "swap",
}


def mkdir_p( path ):
    if not os.path.exists( path ):
        os.makedirs( path )


def listUuids():
    dct = {}
    uuidDir = "/dev/disk/by-uuid"
    for uuid in os.listdir( uuidDir ):
        devicePath = os.readlink( os.path.join( uuidDir, uuid ) )
        devicePath = os.path.join( "/dev", os.path.basename( devicePath ) )
        dct[ devicePath ] = uuid
    return dct


def generateFstabLine( partition ):
    fs = partition[ "fs" ]
    mountPoint = partition[ "mountPoint" ]

    fs = FS_MAP.get( fs, fs )

    if not mountPoint and not fs == "swap":
        return

    options = "defaults"

    if mountPoint == "/":
        check = 1
    elif mountPoint:
        check = 2
    else:
        check = 0

    return "UUID={} {:<14} {:<7} {:<10} 0       {}".format(
        partition[ "uuid" ],
        mountPoint or "none",
        fs,
        options,
        check)


def run():
    partitions = libcalamares.globalStorage.value( "partitions" )
    rootMountPoint = libcalamares.globalStorage.value( "rootMountPoint" )

    # Create fstab
    mkdir_p( os.path.join( rootMountPoint, "etc" ) )
    fstabPath = os.path.join( rootMountPoint, "etc", "fstab" )
    with open( fstabPath, "w" ) as fl:
        print( HEADER, file = fl )
        for partition in partitions:
            line = generateFstabLine( partition )
            if line:
                print( line, file = fl )

    # Create mount points
    for partition in partitions:
        if partition[ "mountPoint" ]:
            mkdir_p( rootMountPoint + partition[ "mountPoint" ] )

    return "All done"