support for skel-files

1 files changed, 18 insertions, 1 deletions

diff --git a/fsmgr.hs b/fsmgr.hs
index 0dc1790..429b0a1 100644
--- a/fsmgr.hs
+++ b/fsmgr.hs
@@ -23,7 +23,7 @@ import qualified Options.Applicative as Opt
          ;import Options.Applicative hiding (action, command)
 
 import ConfigFile
-import System.Directory     (createDirectoryIfMissing)
+import System.Directory     (getCurrentDirectory, createDirectoryIfMissing)
 import System.Posix.Process (getProcessID)
 import System.Posix.Types   (CUid (..))
 import System.Posix.User    (getEffectiveUserID)
@@ -63,6 +63,12 @@ buildRoot config@DiskImageConfig{..} finalOut = do
     forM_ (unpack <$> binaries) $ \b -> do
       p <- fromMaybe (fail $ "not found in ${PATH}: " ++ b) <$> liftIO (pathLocate b)
       cmd_ "cp -L" [p] [mountpoint </> "usr/local/bin" </> b]
+    {- 3.5 skel -}
+    forM_ (unpack <$> skelFiles) $ \f -> do
+      homeDir <- getHomeDir
+      target <- absPath mountpoint <&> (</> "etc/skel")
+      cmd_ "mkdir -p" [target]
+      cmd_ (Cwd homeDir) "cp -r --preserve=mode,timestamps -L --parents -t" [target] [f]
     {- 4. custom setup commands -}
     forM_ chrootCommands $ \c -> do
       cmd_ "chroot" [mountpoint] "/bin/sh -c" [unpack c]
@@ -78,6 +84,17 @@ buildRoot config@DiskImageConfig{..} finalOut = do
     cmd_ "btrfstune -S1" [out]
     cmd_ "mv" [out, finalOut]
 
+absPath :: MonadIO m => FilePath -> m FilePath
+absPath f@('/':_) = return f
+absPath f = do
+  wd <- liftIO getCurrentDirectory
+  return $ wd </> f
+
+getHomeDir :: Action FilePath
+getHomeDir = do
+  Stdout homeDir <- cmd ["sh", "-c", "getent passwd \"$SUDO_USER\"|(IFS=: read _ _ _ _ _ d _; printf %s \"$d\")"]
+  return homeDir
+
 buildInitialImage :: DiskImageConfig -> FilePath -> FilePath -> Action ()
 buildInitialImage DiskImageConfig{..} mountpoint out = do
   case initialImage of