@@ -38,11 +38,15 @@ public class GetAuthToken: UiPathCmdlet
3838 private const string WindowsCredentialsSet = "WindowsCredentials" ;
3939 private const string UnauthenticatedSet = "Unauthenticated" ;
4040
41- [ Parameter ( Mandatory = true , Position = 0 ) ]
42- public string URL { get ; set ; }
41+ private const string CurrentSessionSet = "CurrentSession" ;
4342
44- [ Parameter ( Mandatory = false ) ]
45- public string TenantName { get ; set ; }
43+ [ Parameter ( Mandatory = true , ParameterSetName = CurrentSessionSet ) ]
44+ public SwitchParameter CurrentSession { get ; set ; }
45+
46+ [ Parameter ( Mandatory = true , Position = 0 , ParameterSetName = UserPasswordSet ) ]
47+ [ Parameter ( Mandatory = true , Position = 0 , ParameterSetName = WindowsCredentialsSet ) ]
48+ [ Parameter ( Mandatory = true , Position = 0 , ParameterSetName = UnauthenticatedSet ) ]
49+ public string URL { get ; set ; }
4650
4751 [ Parameter ( Mandatory = true , ParameterSetName = UserPasswordSet ) ]
4852 public string Username { get ; set ; }
@@ -56,19 +60,32 @@ public class GetAuthToken: UiPathCmdlet
5660 [ Parameter ( Mandatory = true , ParameterSetName = UnauthenticatedSet ) ]
5761 public SwitchParameter Unauthenticated { get ; set ; }
5862
63+ [ Parameter ( Mandatory = false , ParameterSetName = UserPasswordSet ) ]
64+ [ Parameter ( Mandatory = false , ParameterSetName = WindowsCredentialsSet ) ]
65+ [ Parameter ( Mandatory = false , ParameterSetName = UnauthenticatedSet ) ]
66+ public string TenantName { get ; set ; }
67+
5968 /// <summary>
6069 /// Sets the current Organization Unit for the authentication token.
6170 /// This parameter is only valid for ORchestrator deployments with Organization Units feature enabled.
6271 /// </summary>
63- [ Parameter ]
72+ [ Parameter ( Mandatory = false , ParameterSetName = UserPasswordSet ) ]
73+ [ Parameter ( Mandatory = false , ParameterSetName = WindowsCredentialsSet ) ]
74+ [ Parameter ( Mandatory = false , ParameterSetName = UnauthenticatedSet ) ]
6475 public string OrganizationUnit { get ; set ; }
6576
66- [ Parameter ]
77+ [ Parameter ( Mandatory = false , ParameterSetName = UserPasswordSet ) ]
78+ [ Parameter ( Mandatory = false , ParameterSetName = WindowsCredentialsSet ) ]
79+ [ Parameter ( Mandatory = false , ParameterSetName = UnauthenticatedSet ) ]
6780 public SwitchParameter Session { get ; set ; }
6881
6982 protected override void ProcessRecord ( )
7083 {
71- try
84+ if ( ParameterSetName == CurrentSessionSet )
85+ {
86+ WriteObject ( AuthenticatedCmdlet . SessionAuthToken ) ;
87+ }
88+ else
7289 {
7390 AuthToken authToken = null ;
7491 if ( ParameterSetName == UserPasswordSet )
@@ -91,17 +108,15 @@ protected override void ProcessRecord()
91108 SetOrganizationUnit ( authToken , OrganizationUnit ) ;
92109 }
93110
111+ authToken . TenantName = TenantName ?? "Default" ;
112+
94113 if ( Session . IsPresent )
95114 {
96115 AuthenticatedCmdlet . SetAuthToken ( authToken ) ;
97116 }
98117
99118 WriteObject ( authToken ) ;
100119 }
101- catch ( Exception e )
102- {
103- WriteVerbose ( e . ToString ( ) ) ;
104- }
105120 }
106121
107122 private void SetOrganizationUnit ( AuthToken authToken , string organizationUnit )
0 commit comments