enable auto scan in notification handler

src/ui/service/notification.go

@@ -16,6 +16,7 @@ package service
 
 import (
 	"encoding/json"
+	"os"
 	"regexp"
 	"strings"
 	"time"
@@ -25,6 +26,8 @@ import (
 	"github.com/vmware/harbor/src/common/utils"
 	"github.com/vmware/harbor/src/common/utils/log"
 	"github.com/vmware/harbor/src/ui/api"
+	"github.com/vmware/harbor/src/ui/config"
+	"github.com/vmware/harbor/src/ui/projectmanager/pms"
 
 	"github.com/astaxie/beego"
 )
@@ -103,6 +106,11 @@ func (n *NotificationHandler) Post() {
 				}
 			}()
 			go api.TriggerReplicationByRepository(repository, []string{tag}, models.RepOpTransfer)
+			if autoScanEnabled(project) {
+				if err := api.TriggerImageScan(repository, tag); err != nil {
+					log.Warningf("Failed to scan image, repository: %s, tag: %s, error: %v", repository, tag, err)
+				}
+			}
 		}
 		if action == "pull" {
 			go func() {
@@ -151,6 +159,27 @@ func filterEvents(notification *models.Notification) ([]*models.Event, error) {
 	return events, nil
 }
 
+func autoScanEnabled(projectName string) bool {
+	if !config.WithClair() {
+		log.Debugf("Auto Scan disabled because Harbor is not deployed with Clair")
+		return false
+	}
+	if config.WithAdmiral() {
+		//TODO get a project manager based on service account.
+		var pm *pms.ProjectManager = pms.NewProjectManager("", "")
+		p, err := pm.Get(projectName)
+		if err != nil {
+			log.Warningf("failed to get project, error: %v", err)
+			return false
+		} else if p == nil {
+			log.Warningf("project with name: %s not found.", projectName)
+			return false
+		}
+		return p.AutomaticallyScanImagesOnPush
+	}
+	return os.Getenv("ENABLE_HARBOR_SCAN_ON_PUSH") == "1"
+}
+
 // Render returns nil as it won't render any template.
 func (n *NotificationHandler) Render() error {
 	return nil