☰

使用脚本编辑类似 INI 的文件

tag-icon • tag-icon shell-script text-processing configuration
使用脚本编辑类似 INI 的文件

我正在编写一个脚本来自动在 Docker 中设置 Puppet 代理配置文件。

基本上，我需要确保以下部分位于/etc/puppet/puppet.conf：

[agent]
server=$PUPPETMASTER_HOSTNAME
masterport=$PUPPETMASTER_PORT

到目前为止，我在 Puppet 代理 runit 脚本中所做的事情是：

function write_puppet_config () {
    read -d '' puppet_config <<EOF
[agent]
server=$1
masterport=$2
EOF

    echo -e "$puppet_config" >> /etc/puppet/puppet.conf
}

# default puppet master port is 8410
test -z "$PUPPET_MASTER_TCP_PORT" && export PUPPET_MASTER_TCP_PORT="8410"

# if there is a puppet master host defined, rewrite the config to match
if [ ! -z "$PUPPET_MASTER_TCP_HOST" ]; then 
    write_puppet_config "$PUPPET_MASTER_TCP_HOST" "$PUPPET_MASTER_TCP_PORT"
fi

问题应该是很明显的。如果 Puppet 配置已经指定了配置，我只是附加另一[agent]部分，这是不好的。

我可以打开条件逻辑（即： grep 如果存在，然后用 sed 重写它），但是有没有办法从命令行进行编辑？我基本上想运行一个命令，其中显示“如果没有代理部分，请添加它，然后确保服务器和主端口在该部分中设置为正确的值。”

我知道 XML 存在这样的结构化工具，但是 INI 样式文件呢？

答案1

看一下crudini，这是为此设计的一个shell工具

conf=/etc/puppet/puppet.conf
crudini --set "$conf" agent server "$PUPPET_MASTER_TCP_HOST"
crudini --set "$conf" agent masterport "$PUPPET_MASTER_TCP_PORT"

或单个原子调用，例如：

echo "
[agent]
server=$1
masterport=$2" |

crudini --merge /etc/puppet/puppet.conf

答案2

以下是一些脚本示例。这些是最低限度的，不用担心错误检查、命令行选项等。我已经表明我是否自己运行了脚本来验证其正确性。

红宝石

inifile为此脚本安装ruby​​gem。这个脚本是经测试。

#!/usr/bin/env ruby
# filename: ~/config.rb

require 'inifile'

PUPPETMASTER_HOSTNAME='hello'
PUPPETMASTER_PORT='world'

ini = IniFile::load('/etc/puppet/puppet.conf')
ini['agent']['server'] = PUPPETMASTER_HOSTNAME
ini['agent']['masterport'] = PUPPETMASTER_PORT
ini.save

用法：

$ chmod 700 ~/config.rb
$ sudo ~/config.rb     # or, if using rvm, rvmsudo ~/config.rb

珀尔

Config::IniFiles使用您的操作系统包管理器进行安装cpan（如果有可用的包）。这个脚本是未经测试的因为我已经停止perl 在我的系统上使用。可能还需要一些工作，欢迎指正。

#!/usr/bin/env perl
# filename: ~/config.pl

use Config::IniFiles;

my $PUPPETMASTER_HOSTNAME='perl';
my $PUPPETMASTER_PORT='1234';

my $ini = Config::IniFiles->new(-file => '/etc/puppet/puppet.conf');

if (! $ini->SectionExists('agent')) {
    $ini->AddSection('agent');
}

if ($ini->exists('agent', 'server')) {
    $ini->setval('agent', 'server', $PUPPETMASTER_HOSTNAME);
}
else {
    $ini->newval('agent', 'server', $PUPPETMASTER_HOSTNAME);
}

if ($ini->exists('agent', 'masterport')) {
    $ini->setval('agent', 'masterport', $PUPPETMASTER_PORT);
}
else {
    $ini->newval('agent', 'masterport', $PUPPETMASTER_PORT);
}

$ini->RewriteConfig();

用法：

$ chmod 700 ~/config.pl
$ sudo ~/config.pl

awk

该脚本对 Bash 和 *nix 更友好，并使用 *nix 操作系统的通用实用程序awk.这个脚本是经测试。

#!/usr/bin/env awk
# filename: ~/config.awk

BEGIN {
    in_agent_section=0;
    is_host_done=0;
    is_port_done=0;
    host = "awk.com";
    port = "4567";
}

in_agent_section == 1 {
    if ($0 ~ /^server[[:space:]]*=/) {
        print "server="host;
        is_host_done = 1;
        next;
    }
    else if ($0 ~ /^masterport[[:space:]]*=/) {
        print "masterport="port;
        is_port_done = 1;
        next;
    }
    else if ($0 ~ /^\[/) {
        in_agent_section = 0;
        if (! is_host_done) {
            print "server="host;
        }
        if (! is_port_done) {
            print "masterport="port;
        }
    }
}

/^\[agent\]/ {
    in_agent_section=1;
}

{ print; }

用法：

$ awk -f ~/config.awk < /etc/puppet/puppet.conf > /tmp/puppet.conf
$ sudo mv /tmp/puppet.conf /etc/puppet/puppet.conf

答案3

如果您有能力安装外部工具，那么我会推荐 Augeas - 这是您需要的唯一用于处理配置文件的工具。它将配置表示为树。阅读更多这里。

相关内容