Question

我有以下代码，我尝试使用C＃为PowerShell创建自定义Cmdlet。我想用自定义cmdlet做的是，用户应该使用两个参数调用它，第一个应该是-Text或-File或-Dir，下一个应该是一个值，一个指定值的字符串用于文本，文件或目录。只要我能看到它就可以正常工作。但我只是好奇是否有另一种简单的方法或更优雅的方法可以用来实现我想要的东西。或者我的解决方案最简单吗？顺便说一下，SHA256Text，SHA256File和SHA256Directory只是我编写的自定义函数，所以不要担心它们。

using System;
using System.IO;
using System.Text;
using System.Security.Cryptography;
using System.Management.Automation;

namespace PSSL
{
    [Cmdlet(VerbsCommon.Get, "SHA256")]
    public class GetSHA256 : PSCmdlet
    {
        #region Members

        private bool text;
        private bool file;
        private bool directory;
        private string argument;

        #endregion

        #region Parameters

        [Parameter(Mandatory = true, Position = 0, ParameterSetName = "Text")]
        public SwitchParameter Text
        {
            get { return text; }
            set { text = value; }
        }

        [Parameter(Mandatory = true, Position = 0, ParameterSetName = "File")]
        public SwitchParameter File
        {
            get { return file; }
            set { file = value; }
        }

        [Parameter(Mandatory = true, Position = 0, ParameterSetName = "Directory")]
        public SwitchParameter Dir
        {
            get { return directory; }
            set { directory = value; }
        }

        [Parameter(Mandatory = true, Position = 1)]
        [ValidateNotNullOrEmpty]
        public string Argument
        {
            get { return argument; }
            set { argument = value; }
        }

        #endregion

        #region Override Methods

        protected override void ProcessRecord()
        {
            switch(ParameterSetName)
            {
                case "Text":
                    SHA256Text(argument);
                    break;

                case "File":
                    SHA256File(argument);
                    break;

                case "Directory":
                    SHA256Directory(argument);
                    break;

                default:
                    throw new ArgumentException("Error: Bad parameter name.");
            }
        }

        #endregion
    }
}

Answer 1

您可以使用parameter sets确保用户只能与-Text一起指定-File，-Dir或-Argument中的一个。然后，您应该使您的开关（和参数）成为强制性而非可选性。这意味着您的ProcessRecord方法将知道已指定了一个开关并且已提供参数。因此，您可以删除使用输出，这些输出将通过Powershell的内置Get-Help cmdlet提供。

使用SwitchParameter的PowerShell自定义Cmdlet

1 个答案: