continued development of the company enrich handler

Author: caffix

engine/plugins/api/aviato/company_enrich.go

@@ -48,19 +48,17 @@ func (ce *companyEnrich) check(e *et.Event) error {
 		return err
 	}
 
-	var o *dbt.Entity
-	var data *companyEnrichResult
+	var orgent *dbt.Entity
 	if support.AssetMonitoredWithinTTL(e.Session, e.Entity, ce.plugin.source, since) {
-		o = ce.lookup(e, e.Entity, since)
-	} else {
-		o, data = ce.query(e, e.Entity, keys)
-		if data != nil {
-			support.MarkAssetMonitored(e.Session, e.Entity, ce.plugin.source)
-		}
+		orgent = ce.lookup(e, e.Entity, since)
+	} else if o, data := ce.query(e, e.Entity, keys); data != nil {
+		orgent = o
+		ce.store(e, o, data)
+		support.MarkAssetMonitored(e.Session, e.Entity, ce.plugin.source)
 	}
 
-	if o != nil {
-		ce.process(e, e.Entity, o)
+	if orgent != nil {
+		ce.process(e, e.Entity, orgent)
 	}
 	return nil
 }
@@ -116,6 +114,10 @@ func (ce *companyEnrich) query(e *et.Event, ident *dbt.Entity, apikey []string)
 	return orgent, enrich
 }
 
+func (ce *companyEnrich) store(e *et.Event, orgent *dbt.Entity, data *companyEnrichResult) {
+
+}
+
 func (ce *companyEnrich) process(e *et.Event, ident, orgent *dbt.Entity) {
 	o := orgent.Asset.(*org.Organization)
 

engine/plugins/api/aviato/plugin.go

@@ -67,6 +67,22 @@ func (a *aviato) Start(r et.Registry) error {
 		return err
 	}
 
+	a.companyEnrich = &companyEnrich{
+		name:   a.name + "-Company-Enrich-Handler",
+		plugin: a,
+	}
+
+	if err := r.RegisterHandler(&et.Handler{
+		Plugin:     a,
+		Name:       a.companyEnrich.name,
+		Priority:   6,
+		Transforms: []string{string(oam.Organization)},
+		EventType:  oam.Identifier,
+		Callback:   a.companyEnrich.check,
+	}); err != nil {
+		return err
+	}
+
 	a.log.Info("Plugin started")
 	return nil
 }

engine/plugins/api/aviato/types.go

@@ -21,6 +21,7 @@ type aviato struct {
 	log           *slog.Logger
 	rlimit        *rate.Limiter
 	companySearch *companySearch
+	companyEnrich *companyEnrich
 	employees     *employees
 	source        *et.Source
 }