/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * Copyright:
 *      2014-2016, Danny Robson <danny@nerdcruft.net>
 */

#include <cruft/crypto/hash/md2.hpp>
#include <cruft/crypto/hash/md4.hpp>
#include <cruft/crypto/hash/md5.hpp>
#include <cruft/crypto/hash/ripemd.hpp>
#include <cruft/crypto/hash/sha1.hpp>
#include <cruft/crypto/hash/sha2.hpp>

#include <cruft/util/hash/adler.hpp>
#include <cruft/util/hash/bsdsum.cpp>
#include <cruft/util/hash/crc.hpp>

#include <cruft/util/io.hpp>
#include <cruft/util/stream.hpp>

#include <iostream>
#include <cstdlib>
#include <cstring>

///////////////////////////////////////////////////////////////////////////////
static
const char* NAMES[] = {
    "adler32",
    "bsdsum",
    "crc",
    "MD2",
    "MD4",
    "MD5",
    "RIPEMD",
    "SHA1",
    "SHA256",
};


///////////////////////////////////////////////////////////////////////////////
std::ostream&
print_digest (std::ostream &os, uint32_t t)
{
    return os << std::hex << t << std::dec;
}


//-----------------------------------------------------------------------------
template <size_t S>
std::ostream&
print_digest (std::ostream &os, std::array<uint8_t,S> digest)
{
    cruft::stream::scoped::flags f (os);
    os << std::hex;

    for (auto c: digest)
        os << +(c >> 4) << +(c & 0x0F);

    return os;
}


///////////////////////////////////////////////////////////////////////////////
static void
compute (const std::string &name, const cruft::view<const uint8_t*> data)
{
    #define stream(TYPE, ...) do {                          \
        if (name != #TYPE)                                  \
            break;                                          \
                                                            \
        print_digest (                                      \
            std::cout,                                      \
            cruft::hash::TYPE{} (data)                       \
        ) << '\n';                                          \
        return;                                             \
    } while (0);

        stream (adler32);
        //stream (bsdsum);
        //stream (crc32);

        //stream (MD2);
        //stream (MD4);
        //stream (MD5);
        //stream (RIPEMD);
        //stream (SHA1);
        //stream (SHA256);

    #undef stream
}


///////////////////////////////////////////////////////////////////////////////
enum {
    ARG_CMD,
    ARG_HASH,
    ARG_INPUT,

    NUM_ARGS
};


//-----------------------------------------------------------------------------
void
print_usage (int argc, char **argv)
{
    (void)argc;

    std::cerr << argv[ARG_CMD] << " [";
    for (auto name: NAMES)
        std::cerr << name << "|";
    std::cerr << "] <input>\n";
}


//-----------------------------------------------------------------------------
int
main (int argc, char **argv)
{
    if (argc != NUM_ARGS) {
        print_usage (argc, argv);
        return EXIT_FAILURE;
    }

    if (strcmp (argv[ARG_INPUT], "-")) {
        const cruft::mapped_file src (argv[ARG_INPUT]);
        compute (argv[ARG_HASH], cruft::view{src});

        return EXIT_SUCCESS;
    } else {
        //compute (argv[ARG_HASH], std::cin);
        return EXIT_SUCCESS;
    }
}