Commit c0f00105 authored by Hans Leidekker's avatar Hans Leidekker Committed by Alexandre Julliard

wmic: Pad output with spaces.

parent 46289779
...@@ -147,14 +147,14 @@ static int output_error( int msg ) ...@@ -147,14 +147,14 @@ static int output_error( int msg )
return output_string( GetStdHandle(STD_ERROR_HANDLE), fmtW, buffer ); return output_string( GetStdHandle(STD_ERROR_HANDLE), fmtW, buffer );
} }
static int output_header( const WCHAR *prop ) static int output_header( const WCHAR *prop, ULONG column_width )
{ {
static const WCHAR bomW[] = {0xfeff}, fmtW[] = {'%','s','\r','\n',0}; static const WCHAR bomW[] = {0xfeff}, fmtW[] = {'%','-','*','s','\r','\n',0};
int len; int len;
DWORD count; DWORD count;
WCHAR buffer[8192]; WCHAR buffer[8192];
len = snprintfW( buffer, ARRAY_SIZE(buffer), fmtW, prop ); len = snprintfW( buffer, ARRAY_SIZE(buffer), fmtW, column_width, prop );
if (!WriteConsoleW( GetStdHandle(STD_OUTPUT_HANDLE), buffer, len, &count, NULL )) /* redirected */ if (!WriteConsoleW( GetStdHandle(STD_OUTPUT_HANDLE), buffer, len, &count, NULL )) /* redirected */
{ {
...@@ -166,10 +166,10 @@ static int output_header( const WCHAR *prop ) ...@@ -166,10 +166,10 @@ static int output_header( const WCHAR *prop )
return count; return count;
} }
static int output_line( const WCHAR *str ) static int output_line( const WCHAR *str, ULONG column_width )
{ {
static const WCHAR fmtW[] = {'%','s','\r','\n',0}; static const WCHAR fmtW[] = {'%','-','*','s','\r','\n',0};
return output_string( GetStdHandle(STD_OUTPUT_HANDLE), fmtW, str ); return output_string( GetStdHandle(STD_OUTPUT_HANDLE), fmtW, column_width, str );
} }
static int query_prop( const WCHAR *class, const WCHAR *propname ) static int query_prop( const WCHAR *class, const WCHAR *propname )
...@@ -186,6 +186,9 @@ static int query_prop( const WCHAR *class, const WCHAR *propname ) ...@@ -186,6 +186,9 @@ static int query_prop( const WCHAR *class, const WCHAR *propname )
WCHAR *prop = NULL; WCHAR *prop = NULL;
BOOL first = TRUE; BOOL first = TRUE;
int len, ret = -1; int len, ret = -1;
IWbemClassObject *obj;
ULONG count, width = 0;
VARIANT v;
WINE_TRACE("%s, %s\n", debugstr_w(class), debugstr_w(propname)); WINE_TRACE("%s, %s\n", debugstr_w(class), debugstr_w(propname));
...@@ -210,29 +213,41 @@ static int query_prop( const WCHAR *class, const WCHAR *propname ) ...@@ -210,29 +213,41 @@ static int query_prop( const WCHAR *class, const WCHAR *propname )
hr = IWbemServices_ExecQuery( services, wql, query, flags, NULL, &result ); hr = IWbemServices_ExecQuery( services, wql, query, flags, NULL, &result );
if (hr != S_OK) goto done; if (hr != S_OK) goto done;
for (;;) for (;;) /* get column width */
{ {
IWbemClassObject *obj; IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count );
ULONG count; if (!count) break;
VARIANT v;
if (!prop && !(prop = find_prop( obj, propname )))
{
output_error( STRING_INVALID_QUERY );
goto done;
}
if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR)
{
VariantChangeType( &v, &v, 0, VT_BSTR );
width = max( strlenW( V_BSTR( &v ) ), width );
VariantClear( &v );
}
IWbemClassObject_Release( obj );
}
width += 2;
IEnumWbemClassObject_Reset( result );
for (;;)
{
IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count ); IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count );
if (!count) break; if (!count) break;
if (first) if (first)
{ {
if (!(prop = find_prop( obj, propname ))) output_header( prop, width );
{
output_error( STRING_INVALID_QUERY );
goto done;
}
output_header( prop );
first = FALSE; first = FALSE;
} }
if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR) if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR)
{ {
VariantChangeType( &v, &v, 0, VT_BSTR ); VariantChangeType( &v, &v, 0, VT_BSTR );
output_line( V_BSTR( &v ) ); output_line( V_BSTR( &v ), width );
VariantClear( &v ); VariantClear( &v );
} }
IWbemClassObject_Release( obj ); IWbemClassObject_Release( obj );
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment