// Copyright (c) 2012 The LevelDB Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file. See the AUTHORS file for names of contributors.

#include <stdio.h>
#include "leveldb/dumpfile.h"
#include "leveldb/env.h"
#include "leveldb/status.h"

namespace leveldb {
namespace {

class StdoutPrinter : public WritableFile
{
public:
    virtual Status Append(const Slice& data)
    {
        fwrite(data.data(), 1, data.size(), stdout);
        return Status::OK();
    }
    virtual Status Close()
    {
        return Status::OK();
    }
    virtual Status Flush()
    {
        return Status::OK();
    }
    virtual Status Sync()
    {
        return Status::OK();
    }
};

bool HandleDumpCommand(Env* env, char** files, int num)
{
    StdoutPrinter printer;
    bool ok = true;
    for (int i = 0; i < num; i++)
    {
        Status s = DumpFile(env, files[i], &printer);
        if (!s.ok())
        {
            fprintf(stderr, "%s\n", s.ToString().c_str());
            ok = false;
        }
    }
    return ok;
}

}  // namespace
}  // namespace leveldb

static void Usage()
{
    fprintf(
        stderr,
        "Usage: leveldbutil command...\n"
        "   dump files...         -- dump contents of specified files\n"
    );
}

int main(int argc, char** argv)
{
    leveldb::Env* env = leveldb::Env::Default();
    bool ok = true;
    if (argc < 2)
    {
        Usage();
        ok = false;
    }
    else
    {
        std::string command = argv[1];
        if (command == "dump")
        {
            ok = leveldb::HandleDumpCommand(env, argv + 2, argc - 2);
        }
        else
        {
            Usage();
            ok = false;
        }
    }
    return (ok ? 0 : 1);
}
