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
|
-- 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 )
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
let cache = if 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
|