aboutsummaryrefslogtreecommitdiff
path: root/lib/Hash/Merge.pm6
blob: b24f25d894d3d9086b42c1ab4ba3eb0cf5609cd1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
#! /usr/bin/env false

use v6.c;
use MONKEY;

augment class Hash
{
    #| Merges a second hash into the hash the method is called on. Hash given as
    #| the argument is not modified.
    #| Traverses the full tree, replacing items in the original hash with the
    #| hash given in the argument. Does not replace positional elements by default,
    #| and instead appends the items from the supplied hash's array to the original
    #| hash's array. The object type of positionals is not retained and instead
    #| becomes an Array type.
    #| Use :no-append-array to replace arrays and positionals instead, which will
    #| also retain the original type and not convert to an Array
    #|
    method merge (%b; Bool:D :$no-append-array = False)
    {
        hashmerge self, %b, :$no-append-array;
    }

    sub hashmerge (%merge-into, %merge-source, Bool:D :$no-append-array)
    {
        for %merge-source.keys -> $key {
            if %merge-into{$key}:exists {
                if %merge-source{$key} ~~ Hash {
                    hashmerge %merge-into{$key}, %merge-source{$key}, :$no-append-array;
                } elsif %merge-source{$key} ~~ Positional {
                    if $no-append-array {
                        %merge-into{$key} = %merge-source{$key};
                    } else {
                        my @a;

                        @a.push: $_ for %merge-into{$key}.list;
                        @a.push: $_ for %merge-source{$key}.list;

                        %merge-into{$key} = @a;
                    }
                } else {
                    # Non-positionals, so strings or Bools or whatever
                    %merge-into{$key} = %merge-source{$key};
                }
            } else {
                %merge-into{$key} = %merge-source{$key};
            }
        }

        %merge-into;
    }
}