示例#1
0
        static async Task Main(string[] args)
        {
            try
            {
                GetSettings(args);

                // Output start.
                WriteLine("*** PolyDeploy Client ***");
                WriteLine();

                // Do we have a target uri.
                if (Options.TargetUri.Equals(string.Empty))
                {
                    throw new ArgumentException("No target uri has been set.");
                }

                // Do we have an api key?
                if (Options.APIKey.Equals(string.Empty))
                {
                    throw new ArgumentException("No api key has been set.");
                }

                // Do we have an encryption key?
                if (Options.EncryptionKey.Equals(string.Empty))
                {
                    throw new ArgumentException("No encryption key has been set.");
                }

                // Output identifying module archives.
                WriteLine("Identifying module archives...");

                // Read zip files in current directory.
                string        currentDirectory = Directory.GetCurrentDirectory();
                List <string> zipFiles         = new List <string>(Directory.GetFiles(currentDirectory, "*.zip"));

                // Is there something to do?
                if (zipFiles.Count <= 0)
                {
                    // No, exit.
                    WriteLine("No module archives found.");
                    WriteLine("Exiting.");
                    ReadLine();
                    Environment.Exit((int)ExitCode.NoModulesFound);
                }

                // Inform user of modules found.
                WriteLine(string.Format("Found {0} module archives in {1}:", zipFiles.Count, currentDirectory));

                foreach (string zipFile in zipFiles)
                {
                    WriteLine(string.Format("\t{0}. {1}", zipFiles.IndexOf(zipFile) + 1, Path.GetFileName(zipFile)));
                }
                WriteLine();

                if (!Options.NoPrompt)
                {
                    // Prompt to continue.
                    WriteLine("Would you like to continue? (y/n)");

                    // Continue?
                    if (!Confirm())
                    {
                        // No, exit.
                        WriteLine("Exiting.");
                        Environment.Exit((int)ExitCode.UserExit);
                    }
                    WriteLine();
                }

                // Get a session.
                string sessionGuid = await API.CreateSessionAsync();

                WriteLine(string.Format("Got session: {0}", sessionGuid));

                DateTime startTime = DateTime.Now;

                // Inform user of encryption.
                WriteLine("Starting encryption and upload...");

                TimeSpan interval             = TimeSpan.FromSeconds(2);
                DateTime apiNotFoundAbortTime = DateTime.Now.AddSeconds(Options.InstallationStatusTimeout);

                foreach (string zipFile in zipFiles)
                {
                    using (FileStream fs = new FileStream(zipFile, FileMode.Open))
                    {
                        WriteLine(string.Format("\t{0}", Path.GetFileName(zipFile)));
                        Write("\t\t...encrypting...");

                        using (Stream es = Crypto.Encrypt(fs, Options.EncryptionKey))
                        {
                            Write("uploading...");

                            var success = false;

                            do
                            {
                                try
                                {
                                    await API.AddPackageAsync(sessionGuid, es, Path.GetFileName(zipFile));

                                    success = true;
                                }
                                catch
                                {
                                    // Api is returning a 404 - wait and try again
                                    System.Threading.Thread.Sleep(interval);
                                    Write("error...retrying...");
                                }
                            } while (!success && DateTime.Now < apiNotFoundAbortTime);
                        }

                        WriteLine("done.");
                    }
                }

                WriteLine(string.Format("Finished encryption and upload in {0} ms.", (DateTime.Now - startTime).TotalMilliseconds));
                WriteLine();

                WriteLine("Starting installation...");

                DateTime             installStartTime = DateTime.Now;
                JavaScriptSerializer jsonSer          = new JavaScriptSerializer();

                // Start.
                (bool installSuccess, SortedList <string, dynamic> results) = await API.InstallAsync(sessionGuid);

                if (!installSuccess)
                {
                    Dictionary <string, dynamic> response;
                    var successfullyReachedApi = false;
                    apiNotFoundAbortTime = DateTime.Now.AddSeconds(Options.InstallationStatusTimeout);

                    // Attempt to get the status of the session from the remote api.
                    // This can fail shortly after an installation as the api has not yet been initialised,
                    // so attempt to get it for the given timespan.
                    do
                    {
                        // Get whether we can reach the api
                        (bool getSessionSuccess, Dictionary <string, dynamic> getSessionResponse) = await API.GetSessionAsync(sessionGuid);

                        successfullyReachedApi = getSessionSuccess;
                        response = getSessionResponse;

                        if (!successfullyReachedApi)
                        {
                            // Api is returning a 404 - wait and try again
                            System.Threading.Thread.Sleep(interval);
                        }
                    } while (!successfullyReachedApi && DateTime.Now < apiNotFoundAbortTime);

                    // If the api couldn't be reached by the given time, something has gone wrong
                    if (!successfullyReachedApi)
                    {
                        throw new HttpException("Remote API returned status 404");
                    }

                    int    status        = -1;
                    string previousPrint = null;

                    DateTime abortTime = DateTime.Now.AddMinutes(10);

                    // Get the installation status from the API until it is complete or until the abort time is reached
                    do
                    {
                        if (response.ContainsKey("Status"))
                        {
                            // Yes, get the status.
                            status = response["Status"];
                        }

                        // Is there a response key?
                        if (response.ContainsKey("Response"))
                        {
                            // Yes, get the response.
                            results = jsonSer.Deserialize <SortedList <string, dynamic> >(response["Response"]);
                        }

                        // As long as we have something.
                        if (status != -1 && results != null)
                        {
                            // Build feedback.
                            string print = BuildUpdateString(results);

                            // Same as previous feedback?
                            if (print != previousPrint)
                            {
                                WriteLine(print);
                                previousPrint = print;
                            }
                        }

                        // Is finished?
                        if (status == 2)
                        {
                            break;
                        }

                        System.Threading.Thread.Sleep(interval);

                        (bool success, Dictionary <string, dynamic> getSessionResponse) = await API.GetSessionAsync(sessionGuid);

                        response = getSessionResponse;

                        // The api should not be returning a 404 status at this point
                        if (!success)
                        {
                            throw new HttpException("Remote API returned status 404");
                        }
                    } while (status < 2 && DateTime.Now < abortTime);
                }
                else
                {
                    // Build feedback.
                    string print = BuildUpdateString(results);

                    // Print feedback.
                    WriteLine(print);
                }

                // Finished install.
                WriteLine(string.Format("Finished installation in {0} ms.", (DateTime.Now - installStartTime).TotalMilliseconds));
                ReadLine();

                int      succeeded    = ParseResults(results).succeeded;
                bool     allSucceeded = succeeded == results.Count;
                ExitCode exitCode     = allSucceeded ? ExitCode.Success : ExitCode.InstallFailure;
                Environment.Exit((int)exitCode);
            }
            catch (Exception ex)
            {
                // Output exception message and stack trace.
                WriteLine(string.Format("Exception caught at: {0}.", DateTime.Now.ToString()));
                WriteException(ex);

                ReadLine();
                Environment.Exit((int)ExitCode.Error);
            }
        }