1// Copyright 2015 The Kyua Authors.
2// All rights reserved.
3//
4// Redistribution and use in source and binary forms, with or without
5// modification, are permitted provided that the following conditions are
6// met:
7//
8// * Redistributions of source code must retain the above copyright
9//   notice, this list of conditions and the following disclaimer.
10// * Redistributions in binary form must reproduce the above copyright
11//   notice, this list of conditions and the following disclaimer in the
12//   documentation and/or other materials provided with the distribution.
13// * Neither the name of Google Inc. nor the names of its contributors
14//   may be used to endorse or promote products derived from this software
15//   without specific prior written permission.
16//
17// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28
29// Dumps all environment variables.
30//
31// This helper program allows comparing the printed environment variables
32// to what 'kyua report --verbose' may output.  It does so by sorting the
33// variables and allowing the caller to customize how the output looks
34// like (indentation for each line and for continuation lines).
35
36#include <cstdlib>
37#include <iostream>
38
39#include "utils/env.hpp"
40#include "utils/text/operations.ipp"
41
42namespace text = utils::text;
43
44
45int
46main(const int argc, const char* const* const argv)
47{
48    if (argc != 3) {
49        std::cerr << "Usage: dump_env <prefix> <continuation-prefix>\n";
50        return EXIT_FAILURE;
51    }
52    const char* prefix = argv[1];
53    const char* continuation_prefix = argv[2];
54
55    const std::map< std::string, std::string > env = utils::getallenv();
56    for (std::map< std::string, std::string >::const_iterator
57             iter = env.begin(); iter != env.end(); ++iter) {
58        const std::string& name = (*iter).first;
59        const std::vector< std::string > value = text::split(
60            (*iter).second, '\n');
61
62        if (value.empty()) {
63            std::cout << prefix << name << "=\n";
64        } else {
65            std::cout << prefix << name << '=' << value[0] << '\n';
66            for (std::vector< std::string >::size_type i = 1;
67                 i < value.size(); ++i) {
68                std::cout << continuation_prefix << value[i] << '\n';
69            }
70        }
71    }
72
73    return EXIT_SUCCESS;
74}
75