//----------------------------------------------------------------------------
//
// Copyright (c) Intel Corporation, 2012  All Rights Reserved.
//
//----------------------------------------------------------------------------

using System;
using System.IO;
using Intel.Manageability;
using Intel.Manageability.Exceptions;
using Common.Utils;
using System.Runtime.InteropServices;

namespace CertificateManagementSample
{
    class Program
    {
        [DllImport("kernel32.dll", CallingConvention = CallingConvention.StdCall)]
        [return: MarshalAs(UnmanagedType.Bool)]
        public static extern bool SetDefaultDllDirectories(int directoryFlags);

        static void Main(string[] args)
        {
            // set default dll lookup directory to system
            SetDefaultDllDirectories(0x00000800); //LOAD_LIBRARY_SEARCH_SYSTEM32

            IAMTInstance amt = null;
            ConnectionInfoEX ci = null;

            try
            {
                // Check if JSON path was provided as an argument. If not, default path would be used.
                try
                {
                    ci = ProgramInput.DeserializeJsonToStruct(args.Length > 0 ? args[0] : null);
                }
                catch (IOException e)
                {
                    Console.WriteLine($"Could not read argument file: {e.Message}");
                    return;
                }

                amt = AMTInstanceFactory.CreateEX(ci);
            }
            catch (ManageabilityException e)
            {
                ci?.Dispose();
                Console.WriteLine(e.Message);
                return;
            }

            try
            {
                // Add a certificate including a private key.
                CertificateManagementFunctionality.AddCertificate(amt);
                // Add a trusted root certificate.
                CertificateManagementFunctionality.AddTrustedRootCertificate(amt);
                // Add a certificate chain.
                CertificateManagementFunctionality.AddCertificateChain(amt);
                // Get All Certificates.
                CertificateManagementFunctionality.GetAllCertificates(amt);
                // Get Trusted Root Certificates.
                CertificateManagementFunctionality.GetTrustedRootCertificates(amt);
                // Get Non Trusted Root Certificates.
                CertificateManagementFunctionality.GetNonTrustedRootCertificates(amt);
                // Get certificate chain.
                CertificateManagementFunctionality.GetCertificateChain(amt);
                // Remove a certificate.
                CertificateManagementFunctionality.RemoveCertificate(amt);
                // Remove all NonTrustedRoot certificates.
                CertificateManagementFunctionality.RemoveNonTrustedRootCertificates(amt);
                // Remove all TrustedRoot certificates.
                CertificateManagementFunctionality.RemoveTrustedRootCertificates(amt);
            }
            catch (Exception ex)
            {
                Console.WriteLine(ex.Message);
            }
            finally
            {
                amt?.Dispose();
            }
        }
    }
}