aboutsummaryrefslogtreecommitdiff
path: root/compiler/src/Compiler.hs
blob: b84dedf25ae9754d8270125a0d8a4e982138c0df (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
-- 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
  ) 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
  , isOutdated )
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 :: ([Glob.Pattern], [Glob.Pattern]) -> FSNode -> Bool
galleryDirFilter (inclusionPatterns, exclusionPatterns) =
      (not . isHidden)
  &&& (matchName True $ anyPattern inclusionPatterns)
  &&& (not . isConfigFile)
  &&& (not . containsOutputGallery)
  &&& (not . (matchName False $ anyPattern exclusionPatterns))

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

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

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

    isConfigFile = matchName False (== galleryConf)
    isGalleryIndex = matchName False (== indexFile)
    isViewerIndex = matchName False (== viewerMainFile)
    containsOutputGallery File{} = False
    containsOutputGallery Dir{items} = any (isGalleryIndex ||| isViewerIndex) items


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

    inputDir <- readDirectory inputDirPath
    let inclusionPatterns = map Glob.compile $ includeFiles config
    let exclusionPatterns = map Glob.compile $ excludeFiles config
    let sourceFilter = galleryDirFilter (inclusionPatterns, exclusionPatterns)
    let sourceTree = filterDir sourceFilter inputDir
    inputTree <- readInputTree sourceTree

    invalidateCache <- isOutdated False inputGalleryConf outputIndex
    let cache = if invalidateCache || rebuildAll then skipCached else withCached

    let itemProc = itemProcessor (pictureMaxResolution config) cache
    let thumbnailProc = thumbnailProcessor (thumbnailMaxResolution config) cache
    let galleryBuilder = buildGalleryTree itemProc thumbnailProc (tagsFromDirectories config)
    resources <- galleryBuilder (galleryName config) inputTree

    galleryCleanupResourceDir resources outputDirPath
    writeJSON outputIndex resources
    writeJSON outputViewerConf $ viewer fullConfig

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

    itemProcessor maxRes cache =
      itemFileProcessor maxRes cache inputDirPath outputDirPath itemsDir
    thumbnailProcessor thumbRes cache =
      thumbnailFileProcessor thumbRes cache inputDirPath outputDirPath thumbnailsDir