Config.pm6

#! /usr/bin/env false

use v6.c;

use Hash::Merge;

use Config::Exception::MissingParserException;
use Config::Exception::FileNotFoundException;
use Config::Parser;

class Config is Associative is export
{
    has Hash $!content = {};
    has Str $!path = "";
    has Str $!parser = "";

    #| Clear the config.
    method clear()
    {
        $!content = {};
        $!path = "";
        $!parser = "";
    }

    #| Return the entire config hash.
    multi method get()
    {
        return $!content;
    }

    #| Fallback method in case the key is Nil. Will always return the default
    #| value.
    multi method get(Nil $key, Any $default = Nil)
    {
        $default;
    }

    #| Get a value from the config object. To get a nested
    #| key, use a . to descent a level.
    multi method get(Str $key, Any $default = Nil)
    {
        self.get($key.split(".").list, $default);
    }

    #| Get a value from the config object using a list
    #| to indicate the nested key to get.
    multi method get(List $keyparts, Any $default = Nil)
    {
        my $index = $!content;

        for $keyparts.list -> $part {
            return $default unless defined($index{$part});

            $index = $index{$part};
        }

        $index;
    }

    #| Get the name of the parser module to use for the
    #| given path.
    method get-parser(Str $path, Str $parser = "" --> Str)
    {
        return $parser if $parser ne "";
        return $!parser if $!parser ne "";

        my $type = self.get-parser-type($path);

        "Config::Parser::" ~ $type;
    }

    #| Get the type of parser required for the given path.
    method get-parser-type(Str $path --> Str)
    {
        given ($path) {
            when .ends-with(".yml") { return "yaml"; };
        }

        my $file = $path;

        if (defined($path.index("/"))) {
            $file = $path.split("/")[*-1];
        }

        if (defined($file.index("."))) {
            return $file.split(".")[*-1].lc;
        }

        return "";
    }

    #| Check wether a given key exists.
    multi method has(Str $key) {
        self.has($key.split(".").list);
    }

    #| Check wether a given key exists using a list to supply
    #| the nested key to check.
    multi method has(List $keyparts)
    {
        my $index = $!content;

        for $keyparts.list -> $part {
            return False unless defined($index{$part});

            $index = $index{$part};
        }

        defined($index);
    }

    #| Return a sorted list of all available keys in the current Config.
    method keys()
    {
        my @keys;

        for $!content.keys -> $key {
            @keys.append: self.extract-keys($key);
        }

        @keys.sort;
    }

    #| Reload the configuration. Requires the configuration to
    #| have been loaded from a file.
    multi method read()
    {
        if ($!path eq "") {
            return False;
        }

        return self.read($!path);
    }

    #| Load a configuration file from the given path. Optionally
    #| set a parser module name to use. If not set, Config will
    #| attempt to deduce the parser to use.
    multi method read(
        Str $path,
        Str $parser = "",
        Bool :$skip-not-found = False
    ) {
        Config::Exception::FileNotFoundException.new(
            path => $path
        ).throw() unless ($path.IO.f || $skip-not-found);

        $!parser = self.get-parser($path, $parser);

        try {
            CATCH {
                when X::CompUnit::UnsatisfiedDependency {
                    Config::Exception::MissingParserException.new(
                        parser => $!parser
                    ).throw();
                }
            }

            require ::($!parser);

            self.read(::($!parser).read($path));
        }

        return True;
    }

    #| Read a list of paths. Will fail on the first file that fails to load for
    #| whatever reason. If no files could be loaded, the method will return
    #| False.
    multi method read(
        List $paths,
        Str $parser = "",
        Bool :$skip-not-found = False
    ) {
        my Bool $read = False;

        for $paths.list -> $path {
            next if $skip-not-found && !$path.IO.f;

            self.read($path, $parser);

            $read = True;
        }

        return $read;
    }

    #| Read a plain Hash into the configuration.
    multi method read(Hash $hash)
    {
        $!content = merge-hash($!content, $hash);

        return True;
    }

    #| Set a single key to a given value;
    multi method set(Str $key, Any $value)
    {
        self.set($key.split(".").list, $value);
    }

    multi method set(List $keyparts, Any $value)
    {
        my $index := $!content;

        for $keyparts.list -> $part {
            $index{$part} = {} unless defined($index{$part});

            $index := $index{$part};
        }

        $index = $value;

        self;
    }

    multi method unset(Str $key)
    {
        self.unset($key.split(".").Array);
    }

    multi method unset(@parts)
    {
        my %index := $!content;
        my $target = @parts.pop;

        for @parts.list -> $part {
            %index{$part} = {} unless defined(%index{$part});

            %index := %index{$part};
        }

        %index{$target}:delete if %index{$target}:exists;

        self;
    }

    #| Write the current configuration to the given path. If
    #| no parser is given, it tries to use the parser that
    #| was used when loading the configuration.
    method write(Str $path, Str $parser = "")
    {
        my $chosen-parser = self.get-parser($path, $parser);

        require ::($chosen-parser);
        return ::($chosen-parser).write($path, $!content);
    }

    multi method AT-KEY(::?CLASS:D: $key)
    {
        self.get($key);
    }

    multi method EXISTS-KEY(::?CLASS:D: $key)
    {
        self.has($key);
    }

    multi method DELETE-KEY(::?CLASS:D: $key)
    {
        self.unset($key);
    }

    multi method ASSIGN-KEY(::?CLASS:D: $key, $new)
    {
        self.set($key, $new);
    }

    multi method BIND-KEY(::?CLASS:D: $key, \new)
    {
        self.set($key, new);
    }

    submethod extract-keys($key)
    {
        my $value = self.get($key);
        return $key if $value !~~ Iterable;

        my @keys;

        for $value.keys -> $nested-key {
            @keys.append: self.extract-keys("{$key}.{$nested-key}");
        }

        return @keys;
    }
}