#!/usr/bin/python
# Copyright (c) 2017 Arista Networks, Inc.  All rights reserved.
# Arista Networks, Inc. Confidential and Proprietary.

import EosCloudInitLib as ecil
import AwsCloudInit as awsci
import AzureCloudInit as azureci
import GcpCloudInit as gcpci
import KvmInit as kvmi
import os
import VeosHypervisor

if __name__ == '__main__':
   doCleanup = True
   platform = VeosHypervisor.getPlatform()
   if platform == 'AWS':
      awsci.cloudInit()
   elif platform == 'Azure':
      azureci.cloudInit()
   elif platform == 'GCP':
      gcpci.cloudInit()
   elif platform == 'KVM':
      kvmi.cloudInit()
   else:
      print "Skipping CloudInit"
      doCleanup = False

   if ecil.isModeSfe():
      if ecil.setupSfeKernelParams():
         print "Kernel command line params changed. Rebooting system"
         os.system( "/sbin/reboot" )
   
   if doCleanup:
      ecil.cleanupExit( "CloudInit done !" )
