From c22ea3de0fc5a42449b4bae80afa1a94c7afa41e Mon Sep 17 00:00:00 2001
From: pacien
Date: Mon, 3 Feb 2020 15:10:58 +0100
Subject: compiler: reject input and output directories that coincide

GitHub: closes #80
---
 compiler/app/Main.hs | 24 +++++++++++++++---------
 ldgallery.1.md       |  7 +++++--
 2 files changed, 20 insertions(+), 11 deletions(-)

diff --git a/compiler/app/Main.hs b/compiler/app/Main.hs
index 1864dee..594a5b7 100644
--- a/compiler/app/Main.hs
+++ b/compiler/app/Main.hs
@@ -20,11 +20,13 @@ module Main where
 
 import GHC.Generics (Generic)
 import Paths_ldgallery_compiler (version, getDataFileName)
+import Control.Monad (when)
 import Data.Version (showVersion)
 import Data.Int (Int64)
 import Data.Aeson (ToJSON)
 import Data.Time.Clock.System (getSystemTime, systemSeconds)
 import System.FilePath ((</>))
+import System.Directory (canonicalizePath)
 import System.Console.CmdArgs
 
 import Compiler
@@ -88,11 +90,9 @@ main =
   do
     opts <- cmdArgs options
     buildGallery opts
-    if (withViewer opts) then do
+    when (withViewer opts) $ do
       copyViewer (outputDir opts)
       writeViewerConfig (outputDir opts </> "config.json")
-    else
-      return ()
 
   where
     gallerySubdir :: String
@@ -100,12 +100,18 @@ main =
 
     buildGallery :: Options -> IO ()
     buildGallery opts =
-      compileGallery
-        (inputDir opts)
-        (galleryOutputDir opts)
-        [outputDir opts]
-        (rebuilAll opts)
-        (cleanOutput opts)
+      checkDistinctPaths (inputDir opts) (outputDir opts)
+      >>  compileGallery
+            (inputDir opts)
+            (galleryOutputDir opts)
+            [outputDir opts]
+            (rebuilAll opts)
+            (cleanOutput opts)
+      where
+        checkDistinctPaths a b = do
+          canonicalA <- canonicalizePath a
+          canonicalB <- canonicalizePath b
+          when (canonicalA == canonicalB) $ error "Input and output paths refer to the same location."
 
     galleryOutputDir :: Options -> FilePath
     galleryOutputDir opts =
diff --git a/ldgallery.1.md b/ldgallery.1.md
index 5612cbf..8928693 100644
--- a/ldgallery.1.md
+++ b/ldgallery.1.md
@@ -25,10 +25,13 @@ ldgallery [\--input-dir _./_] [\--output-dir _./out_] [\--with-viewer]
 Available options are:
 
 -i, \--input-dir _DIR_
-: Gallery source directory.  Defaults to the current directory.
+: Gallery source directory.
+  Defaults to the current directory.
 
 -o, \--output-dir _DIR_
-: Generated gallery output path.  Defaults to ./out.
+: Generated gallery output path.
+  Must be distinct from the source directory.
+  Defaults to ./out.
 
 -r, \--rebuild-all
 : Invalidate cache and recompile everything.
-- 
cgit v1.2.3