Skip to content

Create uWmiVBNetCode.pas #21

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
304 changes: 304 additions & 0 deletions Units/uWmiVBNetCode.pas
Original file line number Diff line number Diff line change
@@ -0,0 +1,304 @@
unit uWmiVBNetCode;

interface

uses
uWmiGenCode,
Classes;


type
TCVBNetWmiClassCodeGenerator=class(TWmiClassCodeGenerator)
public
procedure GenerateCode(Props: TStrings);override;
end;

TCVBNetWmiEventCodeGenerator=class(TWmiEventCodeGenerator)
public
procedure GenerateCode(ParamsIn, Values, Conds, PropsOut: TStrings);override;
end;

TCVBNetWmiMethodCodeGenerator=class(TWmiMethodCodeGenerator)
private
function GetWmiClassDescription: string;
public
procedure GenerateCode(ParamsIn, Values: TStrings);override;
end;

implementation

uses
uSelectCompilerVersion,
StrUtils,
IOUtils,
ComObj,
uWmi_Metadata,
SysUtils;


const
sTagVBNetCode = '[VBNETCODE]';
sTagVBNetEventsWql = '[VBNETEVENTSWQL]';
sTagVBNetEventsOut = '[VBNETEVENTSOUT]';
sTagVBNetCodeParamsIn = '[VBNETCODEINPARAMS]';
sTagVBNetCodeParamsOut = '[VBNETCODEOUTPARAMS]';

function EscapeVBNetStr(const Value:string) : string;
const
ArrChr : Array [0..1] of Char = ('\','"');
ArrChrEscape : Array [0..1] of Char = ('\','\');
Var
i : integer;
begin
Result:=Value;
for i:= Low(ArrChr) to High(ArrChr) do
Result:=StringReplace(Result,ArrChr[i],ArrChrEscape[i]+ArrChr[i], [rfReplaceAll]);
end;


{ TCVBNetWmiClassCodeGenerator }

procedure TCVBNetWmiClassCodeGenerator.GenerateCode(Props: TStrings);
var
StrCode: string;
Descr: string;
DynCode: TStrings;
i: integer;
TemplateCode : string;
Padding : string;
CimType:Integer;

begin
Descr := GetWmiClassDescription;

OutPutCode.BeginUpdate;
DynCode := TStringList.Create;
try
OutPutCode.Clear;
StrCode := TFile.ReadAllText(GetTemplateLocation(Lng_VBNet, ModeCodeGeneration, TWmiGenCode.WmiClasses));

StrCode := StringReplace(StrCode, sTagVersionApp, FileVersionStr, [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagWmiClassName, WmiClass, [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagWmiNameSpace, EscapeVBNetStr(WmiNamespace), [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagHelperTemplate, TemplateCode, [rfReplaceAll]);
Padding:=StringOfChar(' ',20);
if Props.Count > 0 then
for i := 0 to Props.Count - 1 do
begin
CimType:=Integer(Props.Objects[i]);
case CimType of
wbemCimtypeDatetime :
DynCode.Add(Format(
Padding+'Console.WriteLine("{0,-35} {1,-40}","%s",ManagementDateTimeConverter.ToDateTime((string)WmiObject("%s")))'' %s',
[Props.Names[i], Props.Names[i], Props.ValueFromIndex[i]]))
else
DynCode.Add(Format(
Padding+'Console.WriteLine("{0,-35} {1,-40}","%s",WmiObject("%s"))'' %s',
[Props.Names[i], Props.Names[i], Props.ValueFromIndex[i]]));
end;
end;

StrCode := StringReplace(StrCode, sTagVBNetCode, DynCode.Text, [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagWmiClassDescr, Descr, [rfReplaceAll]);
OutPutCode.Text := StrCode;
finally
OutPutCode.EndUpdate;
DynCode.Free;
end;
end;


{ TCVBNetWmiEventCodeGenerator }

procedure TCVBNetWmiEventCodeGenerator.GenerateCode(ParamsIn, Values, Conds,
PropsOut: TStrings);
var
StrCode: string;
sValue: string;
Wql: string;
i, Len: integer;
Props: TStrings;
Padding : string;
CimType:Integer;
begin
Padding := StringOfChar(' ',10);
StrCode := TFile.ReadAllText(GetTemplateLocation(Lng_VBNet, ModeCodeGeneration, TWmiGenCode.WmiEvents));

WQL := Format('Select * From %s Within %d ', [WmiClass, PollSeconds]);
WQL := Format(Padding+StringOfChar(' ',6)+'WmiQuery ="%s"+%s', [WQL, sLineBreak]);

if WmiTargetInstance <> '' then
begin
sValue := Format('"Where TargetInstance ISA %s "', [QuotedStr(WmiTargetInstance)]);
WQL := WQL + Padding+StringOfChar(' ',6) + sValue + '+' + sLineBreak;
end;

for i := 0 to Conds.Count - 1 do
begin
sValue := '';
if (i > 0) or ((i = 0) and (WmiTargetInstance <> '')) then
sValue := 'AND ';
sValue := sValue + ' ' + ParamsIn.Names[i] + Conds[i] + Values[i] + ' ';
WQL := WQL + StringOfChar(' ', 8) + QuotedStr(sValue) + '+' + sLineBreak;
end;

i := LastDelimiter('+', Wql);
if i > 0 then
Wql[i] := ';';


Len := GetMaxLengthItem(PropsOut) + 6;
Props := TStringList.Create;
try
for i := 0 to PropsOut.Count - 1 do
begin
CimType:=Integer(PropsOut.Objects[i]);
case CimType of
wbemCimtypeDatetime :
if StartsText(wbemTargetInstance,PropsOut[i]) then
Props.Add(Format(Padding+' Console.WriteLine("%-'+IntToStr(Len)+'s" & ManagementDateTimeConverter.ToDateTime((string)((ManagementBaseObject)e.NewEvent.Properties("%s").Value)("%s"))',
[PropsOut[i]+' : ',wbemTargetInstance, StringReplace(PropsOut[i],wbemTargetInstance+'.','',[rfReplaceAll])]))
else
Props.Add(Format(Padding+' Console.WriteLine("%-'+IntToStr(Len)+'s" & ManagementDateTimeConverter.ToDateTime(e.NewEvent.Properties("%s").Value.ToString()))', [PropsOut[i]+' : ', PropsOut[i]]));
else
begin
if StartsText(wbemTargetInstance,PropsOut[i]) then
Props.Add(Format(Padding+' Console.WriteLine("%-'+IntToStr(Len)+'s" & ((ManagementBaseObject)e.NewEvent.Properties("%s").Value)("%s")',
[PropsOut[i]+' : ',wbemTargetInstance, StringReplace(PropsOut[i],wbemTargetInstance+'.','',[rfReplaceAll])]))
else
Props.Add(Format(Padding+' Console.WriteLine("%-'+IntToStr(Len)+'s" & e.NewEvent.Properties("%s").Value.ToString())', [PropsOut[i]+' : ', PropsOut[i]]));
end;

end;
end;

StrCode := StringReplace(StrCode, sTagVBNetEventsOut, Props.Text, [rfReplaceAll]);
finally
props.Free;
end;

StrCode := StringReplace(StrCode, sTagVersionApp, FileVersionStr, [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagVBNetEventsWql, WQL, [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagWmiClassName, WmiClass, [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagWmiNameSpace, EscapeVBNetStr(WmiNamespace), [rfReplaceAll]);
OutPutCode.Text := StrCode;
end;


{ TCVBNetWmiMethodCodeGenerator }

procedure TCVBNetWmiMethodCodeGenerator.GenerateCode(ParamsIn,
Values: TStrings);
var
StrCode: string;
Descr: string;
DynCodeInParams: TStrings;
DynCodeOutParams: TStrings;
i: integer;

IsStatic: boolean;
TemplateCode : string;
Padding : string;
begin
Padding := stringofchar(' ', 14);
Descr := GetWmiClassDescription;
OutPutCode.BeginUpdate;
DynCodeInParams := TStringList.Create;
DynCodeOutParams := TStringList.Create;
try
IsStatic := WMiClassMetaData.MethodByName[WmiMethod].IsStatic;

OutPutCode.Clear;
if IsStatic then
begin
{
if UseHelperFunctions then
TemplateCode := TFile.ReadAllText(GetTemplateLocation(sTemplateTemplateFuncts))
else
TemplateCode:='';
}
StrCode := TFile.ReadAllText(GetTemplateLocation(Lng_VBNet, ModeCodeGeneration, TWmiGenCode.WmiMethodStatic));
end
else
begin
{
if UseHelperFunctions then
TemplateCode := TFile.ReadAllText(GetTemplateLocation(sTemplateTemplateFuncts))
else
TemplateCode:='';
}
StrCode := TFile.ReadAllText(GetTemplateLocation(Lng_VBNet, ModeCodeGeneration, TWmiGenCode.WmiMethodNonStatic));
end;

StrCode := StringReplace(StrCode, sTagVersionApp, FileVersionStr, [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagWmiClassName, WmiClass, [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagWmiNameSpace, EscapeVBNetStr(WmiNamespace), [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagWmiMethodName, WmiMethod, [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagWmiPath, EscapeVBNetStr(WmiPath), [rfReplaceAll]);
StrCode := StringReplace(StrCode, sTagHelperTemplate, TemplateCode, [rfReplaceAll]);


//In Params
if ParamsIn.Count > 0 then
for i := 0 to ParamsIn.Count - 1 do
if Values[i] <> WbemEmptyParam then
if ParamsIn.ValueFromIndex[i] = wbemtypeString then
DynCodeInParams.Add(
Format(Padding+' inParams("%s")="%s"', [ParamsIn.Names[i], Values[i]]))
else
DynCodeInParams.Add(
Format(Padding+' inParams("%s")=%s', [ParamsIn.Names[i], Values[i]]));
StrCode := StringReplace(StrCode, sTagVBNetCodeParamsIn, DynCodeInParams.Text, [rfReplaceAll]);

//Out Params
if WMiClassMetaData.MethodByName[WmiMethod].OutParameters.Count > 1 then
begin
for i := 0 to WMiClassMetaData.MethodByName[WmiMethod].OutParameters.Count - 1 do
DynCodeOutParams.Add(
Format(Padding+' Console.WriteLine("{0,-35} {1,-40}","%s",outParams("%s")',
[WMiClassMetaData.MethodByName[WmiMethod].OutParameters[i].Name, WMiClassMetaData.MethodByName[WmiMethod].OutParameters[i].Name]));
end
else
if WMiClassMetaData.MethodByName[WmiMethod].OutParameters.Count = 1 then
begin
DynCodeOutParams.Add(
Format(Padding+' Console.WriteLine("{0,-35} {1,-40}","Return Value",%s)',['outParams("ReturnValue")']));
end;

StrCode := StringReplace(StrCode, sTagVBNetCodeParamsOut,
DynCodeOutParams.Text, [rfReplaceAll]);

StrCode := StringReplace(StrCode, sTagWmiMethodDescr, Descr, [rfReplaceAll]);
OutPutCode.Text := StrCode;
finally
OutPutCode.EndUpdate;
DynCodeInParams.Free;
DynCodeOutParams.Free;
end;
end;


function TCVBNetWmiMethodCodeGenerator.GetWmiClassDescription: string;
var
ClassDescr : TStringList;
Index : Integer;
begin
ClassDescr:=TStringList.Create;
try
Result := WMiClassMetaData.MethodByName[WmiMethod].Description;

if Pos(#10, Result) = 0 then //check if the description has format
ClassDescr.Text := WrapText(Result, 80)
else
ClassDescr.Text := Result;//WrapText(Summary,sLineBreak,[#10],80);

for Index := 0 to ClassDescr.Count - 1 do
ClassDescr[Index] := Format(''' %s', [ClassDescr[Index]]);

Result:=ClassDescr.Text;
finally
ClassDescr.Free;
end;
end;

end.