Refactor of win_user module

This commit is contained in:
Paul Durivage 2014-06-19 15:42:07 -05:00
parent 0ac996fbaa
commit f8f1b7e922

View file

@ -1,8 +1,8 @@
#!powershell #!powershell
# (c) 2014, Matt Martz <matt@sivel.net>, and others
#
# This file is part of Ansible # This file is part of Ansible
# #
# Copyright 2014, Paul Durivage <paul.durivage@rackspace.com>
#
# Ansible is free software: you can redistribute it and/or modify # Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by # it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or # the Free Software Foundation, either version 3 of the License, or
@ -21,12 +21,13 @@
$params = Parse-Args $args; $params = Parse-Args $args;
$result = New-Object psobject; $result = New-Object psobject @{
Set-Attr $result "changed" $false; changed = false
};
If (-not $params.username.GetType) If (-not $params.name.GetType)
{ {
Fail-Json $result "missing required arguments: username" Fail-Json $result "missing required arguments: name"
} }
If (-not $params.password.GetType) If (-not $params.password.GetType)
@ -34,43 +35,77 @@ If (-not $params.password.GetType)
Fail-Json $result "missing required arguments: password" Fail-Json $result "missing required arguments: password"
} }
$extra_args = "" $extra_args = $params "extra_args" ""
If ($params.extra_args.GetType)
{ If ($params.state) {
$extra_args = $params.extra_args; $state = $params.state.ToString().ToLower()
If (($state -ne 'present') -and ($state -ne 'absent')) {
Fail-Json $result "state is '$state'; must be 'present' or 'absent'"
}
}
Elseif (!$params.state) {
$state = "present"
} }
If ($params.creates.GetType -and $params.state.GetType -and $params.state -ne "absent") $username = Get-Attr $params "name"
{ $password = Get-Attr $params "password"
If (Does-User-Exist $params.username)
{ $user_obj = Get-User $username
Exit-Json $result; if (-not $user_obj) {
Fail-Json $result "Could not find user: $username"
}
if ($state -eq 'present') {
# Add or update user
try {
if ($user_obj) {
Update-Password $user_obj $password
}
else {
Create-User $username $password
}
$result.changed = $true
}
catch {
Fail-Json $result $_.Exception.Message
}
}
else {
# Remove user
try {
Delete-User $bob
$result.changed = $true
}
catch {
Fail-Json $result $_.Exception.Message
} }
} }
$logfile = [IO.Path]::GetTempFileName(); $adsi = [ADSI]"WinNT://$env:COMPUTERNAME"
if ($params.state.GetType -and $params.state -eq "absent")
{ function Get-User($user) {
NET USER $params.username $params.password /ADD $adsi.Children | where {$_.SchemaClassName -eq 'user' -and $_.Name -eq $user }
return
} }
Set-Attr $result "changed" $true; function Create-User([string]$user, [string]$passwd) {
$user = $adsi.Create("User", $user)
$user.SetPassword($passwd)
$user.SetInfo()
$user
return
}
$logcontents = Get-Content $logfile; function Update-Password($user, [string]$passwd) {
Remove-Item $logfile; $user.SetPassword($passwd)
$user.SetInfo()
}
Set-Attr $result "log" $logcontents; function Delete-User($user) {
$adsi.delete("user", $user.Name.Value)
}
Set-Attr $result "user" $user_obj; # Soemthing goes here.
Exit-Json $result; Exit-Json $result;
Function Does-User-Exist($username)
{
$objComputer = [ADSI]("WinNT://$env:COMPUTERNAME,computer")
$colUsers = ($objComputer.psbase.children |
Where-Object {$_.psBase.schemaClassName -eq "User"} |
Select-Object -expand Name)
$blnFound = $colUsers -eq $username
}