safesight/edge/agent/internal/httpapi/capabilities.go

package httpapi

import (
    "net/http"
)

// Capability represents a detection capability the device supports.
type Capability struct {
    Key         string `json:"key"`         // "face", "shoe", "intrusion", ...
    Label       string `json:"label"`       // human-readable name
    Available   bool   `json:"available"`   // whether the device has the required plugins
    Description string `json:"description"` // brief description
}

// handleCapabilities returns the list of detection capabilities supported by this device,
// derived from the available media-server node types (plugins).
func (s *Server) handleCapabilities(w http.ResponseWriter, r *http.Request) {
    if r.Method != http.MethodGet {
        errorJSON(w, http.StatusMethodNotAllowed, "method not allowed")
        return
    }
    if !s.authorize(r, false) {
        errorJSON(w, http.StatusUnauthorized, "unauthorized")
        return
    }

    // Query media-server for installed node types.
    installed := s.installedNodeTypes()

    caps := []Capability{
        {
            Key:         "face",
            Label:       "人脸识别",
            Available:   hasAll(installed, "ai_face_recog"),
            Description: "人脸检测与识别",
        },
        {
            Key:         "shoe",
            Label:       "劳保鞋检测",
            Available:   hasAll(installed, "ai_shoe_det"),
            Description: "工鞋/劳保鞋检测",
        },
        {
            Key:         "helmet",
            Label:       "安全帽检测",
            Available:   hasAll(installed, "ai_yolo"),
            Description: "安全帽佩戴检测（需安全帽模型）",
        },
        {
            Key:         "intrusion",
            Label:       "区域入侵",
            Available:   hasAll(installed, "region_event"),
            Description: "区域入侵/越线检测",
        },
        {
            Key:         "action",
            Label:       "行为识别",
            Available:   hasAll(installed, "action_recog"),
            Description: "摔倒、打架等行为识别",
        },
        {
            Key:         "tracking",
            Label:       "人员跟踪",
            Available:   hasAll(installed, "tracker"),
            Description: "多目标跟踪",
        },
    }

    writeJSON(w, http.StatusOK, map[string]any{
        "capabilities": caps,
    })
}

// installedNodeTypes returns all node types known to this agent build.
// The media-server binary carries the same plugin set, so all catalogued
// types are considered available.
func (s *Server) installedNodeTypes() map[string]bool {
    set := map[string]bool{}
    for _, nt := range graphNodeTypesCatalog() {
        set[nt.Type] = true
    }
    return set
}

func hasAll(installed map[string]bool, keys ...string) bool {
    for _, k := range keys {
        if !installed[k] {
            return false
        }
    }
    return true
}