-- Version: 4.0
-- Date: 2009-10-29
-- Description:  Counts the number of instances of a process that are running
-- Arguments: process name, minimum count, maximum count
----------------------------------------------------------------------------------------------------------


-- This function is called by INM to retrieve a script configuration

function OnConfigure()

    -- The variable returned must be called "Config" so INM can find it.
    Config = LuaScriptConfigurator()

    -- Author.
    Config:SetAuthor("Kevin Prestage")

    -- Description.
    Config:SetDescription("Counts the number of instances of a process that are running")

    -- Minimum build version of INM, set to zero for if no specific build version is required.
    Config:SetMinBuildVersion(4.0)

    -- Script version (major/minor)
    Config:SetScriptVersion(1,0)

    -- A parameter configuration, add them in the order the script is extracting them.
    Config:AddArgument("Process Name","Enter the process name as it appears in task manager",LuaScriptConfigurator.CHECK_NOT_EMPTY)
    Config:AddArgument("Minimum Instance(s)","Monitor will go into alarm if the number of instance running falls below this number.",LuaScriptConfigurator.CHECK_NOT_EMPTY)
    Config:AddArgument("Minimum Instance(s)","Monitor will go into alarm if the number of instance running rises above this number.",LuaScriptConfigurator.CHECK_NOT_EMPTY)

    -- Set the entry point, this is the function called by INM
    Config:SetEntryPoint("main")

    -- Done with configuration, return the object
    return Config
end

-- This is the entry point

function main()

    processname = GetArgument(0)
    min = tonumber(GetArgument(1))
    max = tonumber(GetArgument(2))
    ok = true
      result = ""
    count = 0
    SetExitStatus("OK",true)


    --Query WMI for a list of all processes running on the machine.
    Query = TLuaWMIQuery();
    b  = Query:Execute("Select * from Win32_Process where Name = \"" .. processname .. "\"");

    --if the query failed, set the result.
      if (b == false) then
         result = "An error occurred querying WMI.  " .. Query:GetErrorDescription();
      else

        while (Query:NextInstance()) do
            count = count + 1
        end


        if (count < min) then
            result = "The number of running instances is " .. count .. " which is less than " .. min
            ok = false
        elseif (count > max) then
            result = "The number of running instances is " .. count .. " which is more than " .. max
            ok = false
        else
            result = "The number of running instances is " .. count .. " which is between " .. min .. " and " .. max
            ok = true
        end
    end
    SetExitStatus(result,ok);
end

-----------------------------------------------------------------
-- Name: ServiceRunning.lua
-- Author: Kevin Prestage
-- Required INM version: 3.3
-- Version: 1.0
-- Date: 2008-03-21
-- Description: This script will check the
-- running state of a service.
--
-- Arguments:
-- 1) The name of the service to check
-- 2) "true" or "false".  Use true to check if the service IS running.  
--      flase will verify that the service is NOT running.
----------------------------------------------------------------------------------------------------------

--Start a timer so we can monitor the performance of this script.
Timer = TLuaTimer();
Timer:Start();


result = ""; --this will store our result text
b = true; --this is the status of the script.



--set the name of the file containing safe processes.
sn = GetArgument(0);
state = GetArgument(1);
foundit = false;


--Query WMI for a list of all processes running on the machine.
Query = TLuaWMIQuery();
b = Query:Execute("Select * from Win32_Service");

--if the query failed, set the result.
if (b == false) then
    result = "An error occurred querying WMI.  " .. Query:GetErrorDescription();
else

    while (Query:NextInstance()) do
    
    

            --get the process name     
            bOk,exeName = Query:GetProperty("Name",exeName);
            exeName = string.lower(exeName);
        --print (exeName);
    
            --if the service is a match and it SHOULD be running, set foundit = true
                if exeName == string.lower(sn)  then
                foundit = true;
            end
    end
    
    if foundit == true then
        
        if state == "true" then
            b = true;
            result = "The service is running!";
        else
            b = false;
            result = "The service is running and it SHOULD NOT BE";
        end
    else
        if state == "false" then
            b = true;
            reuslt = "The service is NOT running";
        else
            b = false;
            result = "The service is NOT running and it SHOULD BE.";
        end
    end

    

end



--Store the execution time of the script.
mil = Timer:Stop();
StoreStatisticalData(0,mil,0,"Milliseconds");


SetExitStatus(result,b);

-----------------------------------------------------------------
-- Name: SQLReplicationError.lua
-- Author: Kevin Prestage
-- Required INM version: 3.3
-- Version: 1.0
-- Date: 2007-12-13
-- Description: This script will monitor Distribution,
-- Merge, Snapshot and LogReader agents on a
-- server and alert you of any problems
--
-- Arguments:
-- 1) Server Name
-----------------------------------------------------------------


function getSQL(s)
sql =

"select distinct " ..
"    case when a.runstatus = 1 then 'Start' " ..
"         when a.runstatus = 2 then 'Succeed' " ..
"         when a.runstatus = 3 then 'In progress' " ..
"         when a.runstatus = 4 then 'Idle' " ..
"         when a.runstatus = 5 then 'Retry' " ..
"         when a.runstatus = 6 then 'Fail' " ..
"         else 'Unknown' end as current_status, " ..
"    a.runstatus, " ..
"    a.start_time, " ..
"    a.time, " ..
"    a.comments, " ..
"    a.error_id, " ..
"    x.name, " ..
"    '" .. s .. "' " ..
" from dbo.MS" .. s .. "_history a  " ..
" join dbo.MS" .. s .. "_history b on b.agent_id = a.agent_id " ..
" join  dbo.MS" .. s .. "_agents x on x.id = a.agent_id " ..
" where a.time = (select max(time) from MS" .. s .. "_history where agent_id = a.agent_id) ";

return sql;

end










servername = GetArgument(0);

db = TLuaDB();
con = servername .. "@distribution";
print(con);

sql = getSQL("Distribution");
sql = sql .. "union all ";
sql = sql .. getSQL("LogReader");
sql = sql .. "union all ";
sql = sql .. getSQL("Snapshot");
sql = sql .. "union all ";
sql = sql .. getSQL("Merge");



x = db:Connect(con,TLuaDB.CLIENT_SQLSERVER);
result = "";
time = "";
status = "";
statuscode = "";
comments = "";
errorid = 0;
type = "";
name = "";

if (x == true) then
    x = db:Execute(sql);
    if (db:ResultAvilable() == true) then
        while (db:NextRow() == true) do
            time = db:GetCol(4);
            status = db:GetCol(1);
            statuscode  = db:GetCol(2);
            comments = db:GetCol(5);
            errorid = db:GetCol(6);
            name = db:GetCol(7);
            type = db:GetCol(8);
        
            result = result .. name .. "\t" .. type .. "\t" .. status.."\t" .. time .. "\t" .. comments .. "\t" .. errorid .. "\n";
            if (status == "Fail") then
                x = false;
            end
            
        end


    else
        x = false;
        result = "No Agents Found.";
    end
    


else
    result = db:GetErrorDescription();
end





SetExitStatus(result, x);

----------------------------------------------------------------------------------------------------------
-- Name: BackupFile.lua
-- Version: 1.0
-- Date: 2007-11-28
-- Description: Creates a backup copy of a file in the specified folder.  Files are tagged with
-- the script execution time in YYYYMMDDhhmmss format.
-- Arguments:
-- 1) Path of file to backup
-- 2) Name of file to backup
-- 2) Destination Folder to copy to
----------------------------------------------------------------------------------------------------------

backuppath = GetArgument(0);
backupfile = GetArgument(1);
destination = GetArgument(2);

--get the current date
d = TLuaDateTime();

--build the path to the file to backup.
fn = backuppath .. "\\" .. backupfile;

--build the path to the destination file name.
fn2 = destination .. "\\" .. backupfile .. "." .. d:GetDate("%Y%m%d") .. "_" .. d:GetTime("%H%M%S");

--copy the file to the destination.
file = TLuaFile();
result = file:CopyFile(fn, fn2);

if (result == 0) then
    error = GetLastError();
    errdesc = FormatErrorString(error);
    SetExitStatus(error .. " - " .. errdesc,false);

else
    SetExitStatus("File copied to " .. fn2,true);
end

-----------------------------------------------------------------
-- Name: BEJobStatus.lua
-- Author: Kevin Prestage
-- Required INM version: 3.3
-- Version: 1.0
-- Date: 2007-11-22
-- Description: Checks the status of the last occurance of the
-- specified Backup exec job.  Will go into alarm when the
-- status is anything other than 19 (Success) or 3
-- (Complete but with Exceptions).  It will also
-- store statistics including duration in minutes, size of backup
-- set in MB, Backup rate in MB/Minute and the total number of
-- files backed up.
--
-- Arguments:
-- 1) Server Name
-- 2) Name of the backup job.
-----------------------------------------------------------------

servername = GetArgument(0);
jobname = GetArgument(1);


db = TLuaDB();
con = servername .. "@BEDB";
print(con);
sql = "select top 1 isjobactive, finaljobstatus,actualstarttime, endtime, elapsedtimeseconds,totaldatasizebytes, totalratembmin,    totalnumberoffiles, totalnumberofdirectories ";
sql = sql .. "from vwJobHistorySummary ";
sql = sql .. "where JobName = '" .. jobname .. "' and isjobactive = 0 ";
sql = sql .. "order by actualStartTime desc";


x = db:Connect(con,TLuaDB.CLIENT_SQLSERVER);
result = "";
minutes = 0;
size = 0;
rate = 0;
files = 0;

if (x == true) then
    x = db:Execute(sql);
    if (db:ResultAvilable() == true) then
        db:NextRow();
        status = db:GetCol(2);

        if (status == "19") then
            result = "SUCCESS";
        end

        if (status == "3") then
        
            result  = "COMPLETE W/Exceptions";
        end

        if (result == "") then
            result = "The last execution of this job was not successful.";
            x = false;
        else
            minutes = db:GetCol(5) / 60;
            size = db:GetCol(6) / 1048576;

            rate =  db:GetCol(7);
            files = db:GetCol(8);

            result = result.. " " .. minutes.. " Minutes, " .. size .. " MB, " .. rate .. " MB/Min, " .. files .. " Files Backed Up.";
        
            --print (result);

            StoreStatisticalData(0,minutes,0,"Minutes")
            StoreStatisticalData(1,size,0,"MB")
            StoreStatisticalData(2,rate,0,"MB/Min")
            StoreStatisticalData(3,files,0,"Files")
        end
    end
    


else
    result = db:GetErrorDescription();
end





SetExitStatus(result, x);