SWCompression/Sources/swcomp/GZipCommand.swift

// Copyright (c) 2018 Timofey Solomko
// Licensed under MIT License
//
// See LICENSE for license information

import Foundation
import SWCompression
import SwiftCLI

class GZipCommand: Command {

    let name = "gz"
    let shortDescription = "Creates or extracts GZip archive"

    let compress = Flag("-c", "--compress", description: "Compress input file into GZip archive")
    let decompress = Flag("-d", "--decompress", description: "Decompress GZip archive")
    let info = Flag("-i", "--info", description: "Print information from GZip header")

    let useGZipName = Flag("-n", "--use-gzip-name",
                           description: "Use name saved inside GZip archive as output path, if possible")

    var optionGroups: [OptionGroup] {
        let actions = OptionGroup(options: [compress, decompress, info], restriction: .exactlyOne)
        return [actions]
    }

    let input = Parameter()
    let output = OptionalParameter()

    func execute() throws {
        if decompress.value {
            let inputURL = URL(fileURLWithPath: self.input.value)

            var outputURL: URL?
            if let outputPath = output.value {
                outputURL = URL(fileURLWithPath: outputPath)
            } else if inputURL.pathExtension == "gz" {
                outputURL = inputURL.deletingPathExtension()
            }

            let fileData = try Data(contentsOf: inputURL, options: .mappedIfSafe)

            if useGZipName.value {
                let header = try GzipHeader(archive: fileData)
                if let fileName = header.fileName {
                    outputURL = inputURL.deletingLastPathComponent()
                                    .appendingPathComponent(fileName, isDirectory: false)
                }
            }

            guard outputURL != nil else {
                print("""
                      ERROR: Unable to get output path. \
                      No output parameter was specified. \
                      Extension was: \(inputURL.pathExtension)
                      """)
                exit(1)
            }

            let decompressedData = try GzipArchive.unarchive(archive: fileData)
            try decompressedData.write(to: outputURL!)
        } else if compress.value {
            let inputURL = URL(fileURLWithPath: self.input.value)
            let fileName = inputURL.lastPathComponent

            let outputURL: URL
            if let outputPath = output.value {
                outputURL = URL(fileURLWithPath: outputPath)
            } else {
                outputURL = inputURL.appendingPathExtension("gz")
            }

            let fileData = try Data(contentsOf: inputURL, options: .mappedIfSafe)
            let compressedData = try GzipArchive.archive(data: fileData,
                                                        fileName: fileName.isEmpty ? nil : fileName,
                                                        writeHeaderCRC: true)
            try compressedData.write(to: outputURL)
        } else if info.value {
            let inputURL = URL(fileURLWithPath: self.input.value)
            let fileData = try Data(contentsOf: inputURL, options: .mappedIfSafe)

            let header = try GzipHeader(archive: fileData)
            print(header)
        }
    }

}