mirror of
https://github.com/RuoJi6/ARTEX.git
synced 2026-11-06 04:03:50 +08:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3f2c44be91 | ||
|
|
57424249ce |
@@ -680,6 +680,11 @@ func (d *DB) GetFinding(id int64) (*DBFinding, error) {
|
|||||||
if reportConv.Valid {
|
if reportConv.Valid {
|
||||||
f.ReportConversationID = &reportConv.Int64
|
f.ReportConversationID = &reportConv.Int64
|
||||||
}
|
}
|
||||||
|
if f.ReportConversationID == nil && strings.TrimSpace(f.Report) != "" {
|
||||||
|
if resolved, resolveErr := d.ResolveFindingReportConversation(f.ID); resolveErr == nil {
|
||||||
|
f.ReportConversationID = resolved
|
||||||
|
}
|
||||||
|
}
|
||||||
return f, nil
|
return f, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -688,6 +693,31 @@ func (d *DB) SetFindingReportConversationID(findingID, conversationID int64) err
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ResolveFindingReportConversation backfills links for reports generated before
|
||||||
|
// report_conversation_id was persisted.
|
||||||
|
func (d *DB) ResolveFindingReportConversation(findingID int64) (*int64, error) {
|
||||||
|
var id int64
|
||||||
|
err := d.QueryRow(`SELECT c.id FROM conversations c
|
||||||
|
JOIN conversation_activities a ON a.conversation_id=c.id
|
||||||
|
WHERE c.agent_key='reporter' AND (
|
||||||
|
c.title = '漏洞报告 · finding#' || $1::text OR
|
||||||
|
c.title ILIKE '%finding#' || $1::text || '%' OR
|
||||||
|
a.detail ILIKE '%finding_id=' || $1::text || '%' OR
|
||||||
|
a.detail ILIKE '%"finding_id":' || $1::text || '%'
|
||||||
|
)
|
||||||
|
ORDER BY c.updated_at DESC, c.id DESC LIMIT 1`, findingID).Scan(&id)
|
||||||
|
if err == sql.ErrNoRows {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := d.SetFindingReportConversationID(findingID, id); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &id, nil
|
||||||
|
}
|
||||||
|
|
||||||
// DeleteFinding removes a finding entirely: the standalone findings row and its
|
// DeleteFinding removes a finding entirely: the standalone findings row and its
|
||||||
// originating exploration node (kind='finding'), so it disappears from the findings
|
// originating exploration node (kind='finding'), so it disappears from the findings
|
||||||
// list, the per-task 发现 Tab, and the exploration graph alike. Deleting the node
|
// list, the per-task 发现 Tab, and the exploration graph alike. Deleting the node
|
||||||
|
|||||||
Reference in new issue
Block a user