summaryrefslogtreecommitdiff
path: root/katana.raku
blob: 293033328218fd252412590dba7bfc0fc214d401 (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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#!/usr/bin/env raku

sub prefix:<❱>(*@args) {
  say "❱ {@args}";
  my \proc = run @args, :out, :err;

  .say if .chars > 0 for proc.out.slurp(:close), proc.err.slurp(:close);
}

sub prefix:<⁉>(*@args) { ❱ @args unless @args[*-1].IO.f }

class Image {
  has Str $.filename;
  has Str $!source;
  has Str $!dist-dir;

  submethod BUILD(IO::Path :$source, Str :$dist-dir) {
    $!filename = $source.basename;
    $!source = $source.path;
    $!dist-dir = $dist-dir;
  }

  method generate(Int :$thumb-geometry, Str :$bg-blur) {
    my $thumb = "$!dist-dir/thumb/{$.filename}";
    ⁉ [|<convert -auto-orient -geometry>, $thumb-geometry, $!source, $thumb];

    my $blur = "$!dist-dir/blur/{$.filename}";
    ⁉ [|<convert -flip -geometry>, $thumb-geometry/4, '-blur', $bg-blur, $thumb, $blur];

    my $large = "$!dist-dir/large/{$.filename}";
    ⁉ ['cp', $!source, $large];
  }

  method tag(Str $class) {
    "<img class='{$class}' src='./{$class}/{$.filename}' />"
  }
}

sub dist-dirs(Str \dist-dir --> List) { dist-dir <<~>> </large /blur /thumb> }

sub ensure-directories(Str \dist-dir) {
  mkdir dist-dir unless dist-dir.IO.d;
  mkdir $_ unless .IO.d for dist-dirs dist-dir;
}

sub walk-dir(Str \dir, :&onFile, :&onDir = sub (\dir) {} ) {
  return unless dir.IO.d;
  for dir(dir) -> $file {
    given $file {
      when .d { walk-dir $file.path, :&onFile, :&onDir }
      when .f { &onFile($file.path) }
    }
  }
  &onDir(dir);
}

sub cleanup-nonexistent (Str \dist-dir, @images) {
  my $images = set @images.map:{ $_.filename };
  walk-dir dist-dir,
    onFile => sub (\file) {
      unlink file if file.IO.basename ∉ $images
    };
}

sub make-mr-proper(Str \dist-dir) {
  walk-dir dist-dir,
    onFile => sub (\file) { unlink file },
    onDir => sub (\dir) { rmdir dir };
}

multi MAIN(
  Bool :$mr-proper,           #= Clean output dir
  Str :$in-dir = './in',      #= Input dir
  Str :$dist-dir = './dist',  #= Output dir
  Int :$thumb-geometry = 800, #= Thumbnail geometry
  Str :$bg-blur = '0x8',      #= Background blur factor
  Bool :$randomize = True,    #= Randomize order of images
  Str :$title = 'Yay',        #= Album title
  Int :$degree = 4,           #= Degree of parallelism
) {

  my @images = dir($in-dir, test => { "$in-dir/$_".IO.f }).map:{
    Image.new: source => $_, :$dist-dir
  };
  say "Found {@images.elems} images";

  @images = @images.pick: * if $randomize;
  cleanup-nonexistent $dist-dir, @images;
  make-mr-proper $dist-dir if $mr-proper;
  ensure-directories $dist-dir;

  @images.hyper(:$degree).map:{ .generate: :$thumb-geometry, :$bg-blur };
}