#!/usr/bin/env python3
# Copyright (c) 2006-2022 Arastra, Inc.  All rights reserved.
# Arastra, Inc. Confidential and Proprietary.

from AaaHomeLinks import createHomeLinks # pylint: disable=import-self
import argparse
import pwd
import sys

def main():
   helpText = ( 'Make symlinks to persistent files on /mnt/flash'
                ' at /home/<user>.' )
   parser = argparse.ArgumentParser( description=helpText )
   parser.add_argument( '-u', '--user', help='User created by Aaa CLI',
                        required=True )
   args = parser.parse_args()

   errMsg = ""
   try:
      pwEnt = pwd.getpwnam( args.user )
   except KeyError:
      # pylint: disable-next=consider-using-f-string
      errMsg = "user: %s not found" % args.user
   else:
      errors = createHomeLinks( args.user, pwEnt.pw_uid, pwEnt.pw_gid, quiet=False)
      if errors != 0:
         # pylint: disable-next=consider-using-f-string
         errMsg = ( "Errors setting up symlinks for %s. Error count: %d" %
                    ( args.user, errors ) )

   if errMsg:
      print( errMsg, file=sys.stderr )
      sys.exit( 1 )

main()
