PSResourceGet/src/code/LocalResponseUtil.cs at 7c18db27fd6cb29a432306121110b3141f072b10 · PowerShell/PSResourceGet · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

using Microsoft.PowerShell.PSResourceGet.UtilClasses;
using System;
using System.Collections;
using System.Collections.Generic;

namespace Microsoft.PowerShell.PSResourceGet.Cmdlets
{
    internal class LocalResponseUtil : ResponseUtil
    {
        #region Members

        internal override PSRepositoryInfo Repository { get; set; }
        private readonly string _fileTypeKey = "filetype";

        #endregion

        #region Constructor

        public LocalResponseUtil(PSRepositoryInfo repository) : base(repository)
        {
            this.Repository = repository;
        }

        #endregion

        #region Overriden Methods
        public override IEnumerable<PSResourceResult> ConvertToPSResourceResult(FindResults responseResults, bool isResourceRequestedWithWildcard = false)
        {
            foreach (Hashtable response in responseResults.HashtableResponse)
            {
                if (!response.ContainsKey(_fileTypeKey))
                {
                    yield return new PSResourceResult(returnedObject: null, new InvalidOrEmptyResponse("Package response did not contain metadata file type key"), isTerminatingError: false);
                    continue;
                }

                string fileTypeString = response[_fileTypeKey].ToString();
                if (!Enum.TryParse(fileTypeString, out Utils.MetadataFileType fileType) || _fileTypeKey.Equals(Utils.MetadataFileType.None))
                {
                    yield return new PSResourceResult(returnedObject: null, new XmlParsingException("MetadataFileType key in package metadata could not be parsed successfully"), isTerminatingError: false);
                }

                response.Remove(_fileTypeKey);
                PSResourceResult pkgInfo = null;

                switch (fileType)
                {
                    case Utils.MetadataFileType.ModuleManifest:
                        if (!PSResourceInfo.TryConvertFromHashtableForPsd1(
                            pkgMetadata: response,
                            psGetInfo: out PSResourceInfo pkgWithPsd1,
                            out string psd1ErrorMsg,
                            repository: Repository))
                        {
                            yield return new PSResourceResult(returnedObject: null, new ConvertToPSResourceException(psd1ErrorMsg), isTerminatingError: false);
                        }

                        pkgInfo = new PSResourceResult(returnedObject: pkgWithPsd1, new ConvertToPSResourceException(psd1ErrorMsg), isTerminatingError: false);
                        break;

                    case Utils.MetadataFileType.ScriptFile:
                        if (!PSResourceInfo.TryConvertFromHashtableForPs1(
                            pkgMetadata: response,
                            psGetInfo: out PSResourceInfo pkgWithPs1,
                            out string ps1ErrorMsg,
                            repository: Repository))
                        {
                            yield return new PSResourceResult(returnedObject: null, new ConvertToPSResourceException(ps1ErrorMsg), isTerminatingError: false);
                        }

                        pkgInfo = new PSResourceResult(returnedObject: pkgWithPs1, new ConvertToPSResourceException(ps1ErrorMsg), isTerminatingError: false);
                        break;

                    case Utils.MetadataFileType.Nuspec:
                        if (!PSResourceInfo.TryConvertFromHashtableForNuspec(
                            pkgMetadata: response,
                            psGetInfo: out PSResourceInfo pkgWithNuspec,
                            out string nuspecErrorMsg,
                            repository: Repository))
                        {
                            yield return new PSResourceResult(returnedObject: null, new ConvertToPSResourceException(nuspecErrorMsg), isTerminatingError: false);
                        }

                        pkgInfo = new PSResourceResult(returnedObject: pkgWithNuspec, new ConvertToPSResourceException(nuspecErrorMsg), isTerminatingError: false);
                        break;
                }

                yield return pkgInfo;
            }

            yield break;
        }

        #endregion
    }
}