aboutsummaryrefslogtreecommitdiff
path: root/compiler/src/Compiler.hs
blob: 8819ffc1c39d186264c1f1b282c4104c263d4fc3 (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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
-- ldgallery - A static generator which turns a collection of tagged
--             pictures into a searchable web gallery.
--
-- Copyright (C) 2019-2020  Pacien TRAN-GIRARD
--
-- This program is free software: you can redistribute it and/or modify
-- it under the terms of the GNU Affero General Public License as
-- published by the Free Software Foundation, either version 3 of the
-- License, or (at your option) any later version.
--
-- This program is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-- GNU Affero General Public License for more details.
--
-- You should have received a copy of the GNU Affero General Public License
-- along with this program.  If not, see <https://www.gnu.org/licenses/>.

module Compiler
  ( compileGallery
  , writeJSON
  ) where


import Control.Monad (liftM2)
import Data.List (any)
import System.FilePath ((</>))
import qualified System.FilePath.Glob as Glob

import Data.Aeson (ToJSON)
import qualified Data.Aeson as JSON

import Config
import Input (readInputTree)
import Resource (buildGalleryTree, galleryCleanupResourceDir)
import Files
  ( FileName
  , FSNode(..)
  , readDirectory
  , isHidden
  , nodeName
  , filterDir
  , ensureParentDir )
import Processors
  ( itemFileProcessor, thumbnailFileProcessor
  , skipCached, withCached )


galleryConf :: String
galleryConf = "gallery.yaml"

indexFile :: String
indexFile = "index.json"

viewerMainFile :: String
viewerMainFile = "index.html"

viewerConfFile :: String
viewerConfFile = "viewer.json"

itemsDir :: String
itemsDir = "items"

thumbnailsDir :: String
thumbnailsDir = "thumbnails"


writeJSON :: ToJSON a => FileName -> a -> IO ()
writeJSON outputPath object =
  do
    putStrLn $ "Generating:\t" ++ outputPath
    ensureParentDir JSON.encodeFile outputPath object


galleryDirFilter :: CompilerConfig -> FSNode -> Bool
galleryDirFilter config =
      (not . isHidden)
  &&& (not . matchesFile (== galleryConf))
  &&& (not . containsOutputGallery)
  &&& ((matchesDir $ anyPattern $ includedDirectories config) |||
       (matchesFile $ anyPattern $ includedFiles config))
  &&& (not . ((matchesDir $ anyPattern $ excludedDirectories config) |||
              (matchesFile $ anyPattern $ excludedFiles config)))

  where
    (&&&) = liftM2 (&&)
    (|||) = liftM2 (||)

    matchesDir :: (FileName -> Bool) -> FSNode -> Bool
    matchesDir cond dir@Dir{} = maybe False cond $ nodeName dir
    matchesDir _ File{} = False

    matchesFile :: (FileName -> Bool) -> FSNode -> Bool
    matchesFile cond file@File{} = maybe False cond $ nodeName file
    matchesFile _ Dir{} = False

    anyPattern :: [String] -> FileName -> Bool
    anyPattern patterns filename = any (flip Glob.match filename) (map Glob.compile patterns)

    containsOutputGallery :: FSNode -> Bool
    containsOutputGallery File{} = False
    containsOutputGallery Dir{items} =
      any (matchesFile (== indexFile) ||| matchesFile (== viewerMainFile)) items


compileGallery :: FilePath -> FilePath -> Bool -> Bool -> IO ()
compileGallery inputDirPath outputDirPath rebuildAll cleanOutput =
  do
    fullConfig <- readConfig inputGalleryConf
    let config = compiler fullConfig

    inputDir <- readDirectory inputDirPath
    let sourceFilter = galleryDirFilter config
    let sourceTree = filterDir sourceFilter inputDir
    inputTree <- readInputTree sourceTree

    let cache = if rebuildAll then skipCached else withCached
    let itemProc = itemProcessor config cache
    let thumbnailProc = thumbnailProcessor config cache
    let galleryBuilder = buildGalleryTree itemProc thumbnailProc (tagsFromDirectories config)
    resources <- galleryBuilder (galleryName config) inputTree

    if cleanOutput then
      galleryCleanupResourceDir resources outputDirPath
    else
      return ()

    writeJSON outputIndex resources
    writeJSON outputViewerConf $ viewer fullConfig

  where
    inputGalleryConf = inputDirPath </> galleryConf
    outputIndex = outputDirPath </> indexFile
    outputViewerConf = outputDirPath </> viewerConfFile

    itemProcessor config cache =
      itemFileProcessor
        (pictureMaxResolution config) cache
        inputDirPath outputDirPath itemsDir
    thumbnailProcessor config cache =
      thumbnailFileProcessor
        (thumbnailMaxResolution config) cache
        inputDirPath outputDirPath thumbnailsDir