highperfocused/P4wnP1_aloa
1
0
Fork 0
mirror of https://github.com/RoganDawes/P4wnP1_aloa.git synced 2025-03-17 21:31:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

More work on TriggerActions

Browse Source
This commit is contained in:
MaMe82 2018-09-26 01:08:51 +02:00
parent 84320719c0
commit 416b6413bd
1 changed files with 59 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
service/triggerAction.go
View File

@ -29,6 +29,7 @@ func (tam *TriggerActionManager) processing_loop() {
break Outer // abort loop on "nil" event, as this indicates the EventQueue channel has been closed
}
fmt.Println("TriggerActionManager received unfiltered event", evt)
tam.dispatchTriggerEvent(evt)
// check if relevant and dispatch to triggers
case <- tam.evtRcv.Ctx.Done():
// evvent Receiver cancelled or unregistered
@ -38,7 +39,56 @@ func (tam *TriggerActionManager) processing_loop() {
fmt.Println("TAM processing loop finished")
}
func (tam *TriggerActionManager) dispatchTriggerEvent(evt *pb.Event) {
if evt.Type != common_web.EVT_TRIGGER { return }
triggerTypeRcv := common_web.EvtTriggerType(evt.Values[0].GetTint64())
tam.registeredTriggerActionMutex.Lock()
defer tam.registeredTriggerActionMutex.Unlock()
for _,ta := range tam.registeredTriggerAction {
switch triggerType := ta.Trigger.(type) {
case *pb.TriggerAction_ServiceStarted:
if triggerTypeRcv == common_web.EVT_TRIGGER_TYPE_SERVICE_STARTED {
// trigger action of ta
tam.fireAction(ta.Trigger, ta.Action)
}
case *pb.TriggerAction_UsbGadgetConnected:
if triggerTypeRcv == common_web.EVT_TRIGGER_TYPE_USB_GADGET_CONNECTED {
// trigger action of ta
tam.fireAction(triggerType.UsbGadgetConnected, ta.Action)
}
case *pb.TriggerAction_UsbGadgetDisconnected:
if triggerTypeRcv == common_web.EVT_TRIGGER_TYPE_USB_GADGET_DISCONNECTED {
// trigger action of ta
tam.fireAction(triggerType.UsbGadgetDisconnected, ta.Action)
}
case *pb.TriggerAction_DhcpLeaseGranted:
if triggerTypeRcv == common_web.EVT_TRIGGER_TYPE_USB_GADGET_DISCONNECTED {
// trigger action of ta
tam.fireAction(triggerType.DhcpLeaseGranted, ta.Action)
}
}
}
}
func (tam *TriggerActionManager) fireAction(trigger interface{}, action interface{}) (err error ) {
switch actionType := action.(type) {
case *pb.TriggerAction_BashScript:
bs := actionType.BashScript
fmt.Printf("Fire bash script '%s'\n", bs.ScriptPath)
case *pb.TriggerAction_HidScript:
hs := actionType.HidScript
fmt.Printf("Starting HID script '%s'\n", hs.ScriptName)
case *pb.TriggerAction_DeploySettingsTemplate:
st := actionType.DeploySettingsTemplate
strType := pb.ActionDeploySettingsTemplate_TemplateType_name[int32(st.Type)]
fmt.Printf("Deploy settings template of type [%s] with name '%s'\n", strType, st.TemplateName)
case *pb.TriggerAction_Log:
fmt.Printf("Logging trigger '%+v'\n", trigger)
}
return nil
}
func (tam *TriggerActionManager) AddTriggerAction(ta *pb.TriggerAction) (err error) {
tam.registeredTriggerActionMutex.Lock()
@ -64,7 +114,16 @@ func (tam *TriggerActionManager) Start() {
},
},
}
serviceUpLog := &pb.TriggerAction{
Trigger: &pb.TriggerAction_ServiceStarted{
ServiceStarted: &pb.TriggerServiceStarted{},
},
Action: &pb.TriggerAction_Log{
Log: &pb.ActionLog{},
},
}
tam.AddTriggerAction(serviceUpRunScript)
tam.AddTriggerAction(serviceUpLog)
fmt.Printf("TEST TRIGGER ACTION: Service up %+v\n", serviceUpRunScript)
go tam.processing_loop()