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 +++++++++++++++---------
 1 file changed, 15 insertions(+), 9 deletions(-)

(limited to 'compiler')

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 =
-- 
cgit v1.2.3