mirror of
https://github.com/overcuriousity/trace.git
synced 2025-12-21 05:22:21 +00:00
Compare commits
4 Commits
v0.0.1-alp
...
v0.0.2-alp
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
aa0f67f1fc | ||
|
|
dc8bd777ef | ||
|
|
89e7b20694 | ||
|
|
ba1fff36f2 |
8
.github/workflows/release.yml
vendored
8
.github/workflows/release.yml
vendored
@@ -27,14 +27,14 @@ jobs:
|
|||||||
|
|
||||||
- name: Build Linux binary
|
- name: Build Linux binary
|
||||||
run: |
|
run: |
|
||||||
pyinstaller --onefile --name trace-linux main.py
|
pyinstaller --onefile --name trace main.py
|
||||||
|
|
||||||
- name: Upload Linux binary to release
|
- name: Upload Linux binary to release
|
||||||
uses: softprops/action-gh-release@v2
|
uses: softprops/action-gh-release@v2
|
||||||
env:
|
env:
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
with:
|
with:
|
||||||
files: ./dist/trace-linux
|
files: ./dist/trace
|
||||||
|
|
||||||
build-windows:
|
build-windows:
|
||||||
runs-on: windows-latest
|
runs-on: windows-latest
|
||||||
@@ -55,11 +55,11 @@ jobs:
|
|||||||
|
|
||||||
- name: Build Windows executable
|
- name: Build Windows executable
|
||||||
run: |
|
run: |
|
||||||
pyinstaller --onefile --name trace-windows main.py
|
pyinstaller --onefile --name trace main.py
|
||||||
|
|
||||||
- name: Upload Windows executable to release
|
- name: Upload Windows executable to release
|
||||||
uses: softprops/action-gh-release@v2
|
uses: softprops/action-gh-release@v2
|
||||||
env:
|
env:
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
with:
|
with:
|
||||||
files: ./dist/trace-windows.exe
|
files: ./dist/trace.exe
|
||||||
|
|||||||
@@ -122,7 +122,13 @@ Run `trace` without arguments to open the interface.
|
|||||||
* IOC counts are displayed in red in case and evidence views.
|
* IOC counts are displayed in red in case and evidence views.
|
||||||
* `a`: **Set Active**. Sets the currently selected Case or Evidence as the global "Active" context.
|
* `a`: **Set Active**. Sets the currently selected Case or Evidence as the global "Active" context.
|
||||||
* `d`: Delete the selected Case or Evidence (with confirmation).
|
* `d`: Delete the selected Case or Evidence (with confirmation).
|
||||||
* `v`: View all notes for the current Case (in Case Detail view).
|
* `v`: **View All Notes**. View all notes for the current Case or Evidence in a scrollable full-screen view.
|
||||||
|
* **IOC Highlighting**: All IOCs in notes are automatically highlighted in red for immediate visibility.
|
||||||
|
* **Tag Highlighting**: Hashtags are highlighted in cyan.
|
||||||
|
* Press `Enter` on any note in case/evidence detail view to jump directly to that note in the full view.
|
||||||
|
* The selected note will be centered and highlighted.
|
||||||
|
* Navigate with arrow keys, Page Up/Down, Home/End.
|
||||||
|
* Press `n` to add a new note without leaving the view.
|
||||||
* `/`: Filter list (type to search, `Esc` or `Enter` to exit filter mode).
|
* `/`: Filter list (type to search, `Esc` or `Enter` to exit filter mode).
|
||||||
* `s`: Settings menu (in Case List view).
|
* `s`: Settings menu (in Case List view).
|
||||||
* `Esc`: Cancel during input dialogs.
|
* `Esc`: Cancel during input dialogs.
|
||||||
|
|||||||
11
trace/cli.py
11
trace/cli.py
@@ -42,11 +42,14 @@ def quick_add_note(content: str):
|
|||||||
signature = None
|
signature = None
|
||||||
if settings.get("pgp_enabled", True):
|
if settings.get("pgp_enabled", True):
|
||||||
gpg_key_id = settings.get("gpg_key_id", None)
|
gpg_key_id = settings.get("gpg_key_id", None)
|
||||||
signature = Crypto.sign_content(f"Hash: {note.content_hash}\nContent: {note.content}", key_id=gpg_key_id)
|
if gpg_key_id:
|
||||||
if signature:
|
signature = Crypto.sign_content(f"Hash: {note.content_hash}\nContent: {note.content}", key_id=gpg_key_id)
|
||||||
note.signature = signature
|
if signature:
|
||||||
|
note.signature = signature
|
||||||
|
else:
|
||||||
|
print("Warning: GPG signature failed (GPG not found or no key). Note saved without signature.")
|
||||||
else:
|
else:
|
||||||
print("Warning: GPG signature failed (GPG not found or no key). Note saved without signature.")
|
print("Warning: No GPG key ID configured. Note saved without signature.")
|
||||||
|
|
||||||
# Attach to evidence or case
|
# Attach to evidence or case
|
||||||
if target_evidence:
|
if target_evidence:
|
||||||
|
|||||||
@@ -96,6 +96,71 @@ class Note:
|
|||||||
data = f"{self.timestamp}:{self.content}".encode('utf-8')
|
data = f"{self.timestamp}:{self.content}".encode('utf-8')
|
||||||
self.content_hash = hashlib.sha256(data).hexdigest()
|
self.content_hash = hashlib.sha256(data).hexdigest()
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def extract_iocs_from_text(text):
|
||||||
|
"""Extract IOCs from text and return as list of (ioc, type) tuples"""
|
||||||
|
iocs = []
|
||||||
|
seen = set()
|
||||||
|
|
||||||
|
# IPv4 addresses
|
||||||
|
ipv4_pattern = r'\b(?:[0-9]{1,3}\.){3}[0-9]{1,3}\b'
|
||||||
|
for match in re.findall(ipv4_pattern, text):
|
||||||
|
if match not in seen:
|
||||||
|
seen.add(match)
|
||||||
|
iocs.append((match, 'ipv4'))
|
||||||
|
|
||||||
|
# IPv6 addresses (simplified)
|
||||||
|
ipv6_pattern = r'\b(?:[0-9a-fA-F]{1,4}:){7}[0-9a-fA-F]{1,4}\b'
|
||||||
|
for match in re.findall(ipv6_pattern, text):
|
||||||
|
if match not in seen:
|
||||||
|
seen.add(match)
|
||||||
|
iocs.append((match, 'ipv6'))
|
||||||
|
|
||||||
|
# URLs (check before domains to avoid double-matching)
|
||||||
|
url_pattern = r'https?://[^\s]+'
|
||||||
|
for match in re.findall(url_pattern, text):
|
||||||
|
if match not in seen:
|
||||||
|
seen.add(match)
|
||||||
|
iocs.append((match, 'url'))
|
||||||
|
|
||||||
|
# Domain names (basic pattern)
|
||||||
|
domain_pattern = r'\b(?:[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?\.)+[a-zA-Z]{2,}\b'
|
||||||
|
for match in re.findall(domain_pattern, text):
|
||||||
|
# Filter out common false positives and already seen URLs
|
||||||
|
if match not in seen and not match.startswith('example.'):
|
||||||
|
seen.add(match)
|
||||||
|
iocs.append((match, 'domain'))
|
||||||
|
|
||||||
|
# SHA256 hashes (64 hex chars) - check before SHA1 and MD5
|
||||||
|
sha256_pattern = r'\b[a-fA-F0-9]{64}\b'
|
||||||
|
for match in re.findall(sha256_pattern, text):
|
||||||
|
if match not in seen:
|
||||||
|
seen.add(match)
|
||||||
|
iocs.append((match, 'sha256'))
|
||||||
|
|
||||||
|
# SHA1 hashes (40 hex chars) - check before MD5
|
||||||
|
sha1_pattern = r'\b[a-fA-F0-9]{40}\b'
|
||||||
|
for match in re.findall(sha1_pattern, text):
|
||||||
|
if match not in seen:
|
||||||
|
seen.add(match)
|
||||||
|
iocs.append((match, 'sha1'))
|
||||||
|
|
||||||
|
# MD5 hashes (32 hex chars)
|
||||||
|
md5_pattern = r'\b[a-fA-F0-9]{32}\b'
|
||||||
|
for match in re.findall(md5_pattern, text):
|
||||||
|
if match not in seen:
|
||||||
|
seen.add(match)
|
||||||
|
iocs.append((match, 'md5'))
|
||||||
|
|
||||||
|
# Email addresses
|
||||||
|
email_pattern = r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b'
|
||||||
|
for match in re.findall(email_pattern, text):
|
||||||
|
if match not in seen:
|
||||||
|
seen.add(match)
|
||||||
|
iocs.append((match, 'email'))
|
||||||
|
|
||||||
|
return iocs
|
||||||
|
|
||||||
def to_dict(self):
|
def to_dict(self):
|
||||||
return {
|
return {
|
||||||
"note_id": self.note_id,
|
"note_id": self.note_id,
|
||||||
|
|||||||
@@ -23,7 +23,6 @@ class Storage:
|
|||||||
|
|
||||||
def _create_demo_case(self):
|
def _create_demo_case(self):
|
||||||
"""Create a demo case with evidence showcasing all features"""
|
"""Create a demo case with evidence showcasing all features"""
|
||||||
# Create demo case
|
|
||||||
demo_case = Case(
|
demo_case = Case(
|
||||||
case_number="DEMO-2024-001",
|
case_number="DEMO-2024-001",
|
||||||
name="Sample Investigation",
|
name="Sample Investigation",
|
||||||
|
|||||||
619
trace/tui.py
619
trace/tui.py
@@ -490,67 +490,21 @@ class TUI:
|
|||||||
self.stdscr.attroff(note_color)
|
self.stdscr.attroff(note_color)
|
||||||
y_pos += 1
|
y_pos += 1
|
||||||
|
|
||||||
# Split screen between case notes and evidence
|
# Split screen between evidence and case notes
|
||||||
# Allocate space: half for case notes, half for evidence (if both exist)
|
# Allocate space: half for evidence, half for case notes (if both exist)
|
||||||
available_space = self.content_h - 5
|
available_space = self.content_h - 5
|
||||||
case_notes = self.active_case.notes
|
case_notes = self.active_case.notes
|
||||||
evidence_list = self._get_filtered_list(self.active_case.evidence, "name", "description")
|
evidence_list = self._get_filtered_list(self.active_case.evidence, "name", "description")
|
||||||
|
|
||||||
# Determine context: are we selecting notes or evidence?
|
# Determine context: are we selecting evidence or notes?
|
||||||
# If there are case notes, treat indices 0 to len(notes)-1 as notes
|
# Evidence items are indices 0 to len(evidence)-1
|
||||||
# If there is evidence, treat indices len(notes) to len(notes)+len(evidence)-1 as evidence
|
# Case notes are indices len(evidence) to len(evidence)+len(notes)-1
|
||||||
total_items = len(case_notes) + len(evidence_list)
|
total_items = len(evidence_list) + len(case_notes)
|
||||||
|
|
||||||
# Determine what's selected
|
# Determine what's selected
|
||||||
selecting_note = self.selected_index < len(case_notes)
|
selecting_evidence = self.selected_index < len(evidence_list)
|
||||||
|
|
||||||
# Case Notes section
|
|
||||||
if case_notes:
|
|
||||||
if y_pos < self.height - 3:
|
|
||||||
self.stdscr.attron(curses.color_pair(5) | curses.A_BOLD)
|
|
||||||
self.stdscr.addstr(y_pos, 2, "▪ Case Notes")
|
|
||||||
self.stdscr.attroff(curses.color_pair(5) | curses.A_BOLD)
|
|
||||||
self.stdscr.attron(curses.color_pair(6) | curses.A_DIM)
|
|
||||||
self.stdscr.addstr(y_pos, 16, f"({len(case_notes)} notes)")
|
|
||||||
self.stdscr.attroff(curses.color_pair(6) | curses.A_DIM)
|
|
||||||
y_pos += 1
|
|
||||||
|
|
||||||
# Calculate space for case notes
|
|
||||||
notes_space = min(len(case_notes), available_space // 2) if evidence_list else available_space
|
|
||||||
|
|
||||||
# Update scroll position if needed
|
|
||||||
self._update_scroll(total_items)
|
|
||||||
|
|
||||||
# Display notes
|
|
||||||
for i in range(notes_space):
|
|
||||||
note_idx = self.scroll_offset + i
|
|
||||||
if note_idx >= len(case_notes):
|
|
||||||
break
|
|
||||||
|
|
||||||
note = case_notes[note_idx]
|
|
||||||
y = y_pos + 1 + i
|
|
||||||
|
|
||||||
# Check if we're out of bounds
|
|
||||||
if y >= self.height - 3:
|
|
||||||
break
|
|
||||||
|
|
||||||
# Format note content
|
|
||||||
note_content = note.content.replace('\n', ' ').replace('\r', ' ')
|
|
||||||
display_str = f"- {note_content}"
|
|
||||||
display_str = self._safe_truncate(display_str, self.width - 6)
|
|
||||||
|
|
||||||
# Highlight if selected
|
|
||||||
if note_idx == self.selected_index:
|
|
||||||
self.stdscr.attron(curses.color_pair(1))
|
|
||||||
self.stdscr.addstr(y, 4, display_str)
|
|
||||||
self.stdscr.attroff(curses.color_pair(1))
|
|
||||||
else:
|
|
||||||
self.stdscr.addstr(y, 4, display_str)
|
|
||||||
|
|
||||||
y_pos += notes_space + 1
|
|
||||||
|
|
||||||
# Evidence section header
|
# Evidence section header
|
||||||
y_pos += 1
|
|
||||||
if y_pos < self.height - 3:
|
if y_pos < self.height - 3:
|
||||||
self.stdscr.attron(curses.color_pair(5) | curses.A_BOLD)
|
self.stdscr.attron(curses.color_pair(5) | curses.A_BOLD)
|
||||||
self.stdscr.addstr(y_pos, 2, "▪ Evidence")
|
self.stdscr.addstr(y_pos, 2, "▪ Evidence")
|
||||||
@@ -578,14 +532,26 @@ class TUI:
|
|||||||
|
|
||||||
self._update_scroll(total_items)
|
self._update_scroll(total_items)
|
||||||
|
|
||||||
for i in range(list_h):
|
# Calculate space for evidence
|
||||||
# Evidence indices start after case notes
|
evidence_space = min(len(evidence_list), available_space // 2) if case_notes else available_space
|
||||||
evidence_idx = self.scroll_offset + i - len(case_notes)
|
|
||||||
|
self._update_scroll(total_items)
|
||||||
|
|
||||||
|
# Calculate which evidence items to display
|
||||||
|
# If selecting a case note, show evidence from the beginning
|
||||||
|
# If selecting evidence, scroll to show the selected evidence
|
||||||
|
if selecting_evidence:
|
||||||
|
evidence_scroll_offset = max(0, self.selected_index - evidence_space // 2)
|
||||||
|
else:
|
||||||
|
evidence_scroll_offset = 0
|
||||||
|
|
||||||
|
for i in range(evidence_space):
|
||||||
|
evidence_idx = evidence_scroll_offset + i
|
||||||
if evidence_idx < 0 or evidence_idx >= len(evidence_list):
|
if evidence_idx < 0 or evidence_idx >= len(evidence_list):
|
||||||
continue
|
continue
|
||||||
|
|
||||||
ev = evidence_list[evidence_idx]
|
ev = evidence_list[evidence_idx]
|
||||||
y = y_pos + 2 + i
|
y = y_pos + i
|
||||||
if y >= self.height - 3: # Don't overflow into status bar
|
if y >= self.height - 3: # Don't overflow into status bar
|
||||||
break
|
break
|
||||||
|
|
||||||
@@ -628,8 +594,7 @@ class TUI:
|
|||||||
base_display = self._safe_truncate(display_str, self.width - 6)
|
base_display = self._safe_truncate(display_str, self.width - 6)
|
||||||
|
|
||||||
# Check if this evidence item is selected
|
# Check if this evidence item is selected
|
||||||
item_idx = len(case_notes) + evidence_idx
|
if evidence_idx == self.selected_index:
|
||||||
if item_idx == self.selected_index:
|
|
||||||
# Highlighted selection
|
# Highlighted selection
|
||||||
self.stdscr.attron(curses.color_pair(1))
|
self.stdscr.attron(curses.color_pair(1))
|
||||||
self.stdscr.addstr(y, 4, base_display)
|
self.stdscr.addstr(y, 4, base_display)
|
||||||
@@ -662,6 +627,56 @@ class TUI:
|
|||||||
else:
|
else:
|
||||||
self.stdscr.addstr(y, 4, base_display)
|
self.stdscr.addstr(y, 4, base_display)
|
||||||
|
|
||||||
|
y_pos += evidence_space
|
||||||
|
|
||||||
|
# Case Notes section
|
||||||
|
if case_notes:
|
||||||
|
y_pos += 2
|
||||||
|
if y_pos < self.height - 3:
|
||||||
|
self.stdscr.attron(curses.color_pair(5) | curses.A_BOLD)
|
||||||
|
self.stdscr.addstr(y_pos, 2, "▪ Case Notes")
|
||||||
|
self.stdscr.attroff(curses.color_pair(5) | curses.A_BOLD)
|
||||||
|
self.stdscr.attron(curses.color_pair(6) | curses.A_DIM)
|
||||||
|
self.stdscr.addstr(y_pos, 16, f"({len(case_notes)} notes)")
|
||||||
|
self.stdscr.attroff(curses.color_pair(6) | curses.A_DIM)
|
||||||
|
y_pos += 1
|
||||||
|
|
||||||
|
# Calculate remaining space for case notes
|
||||||
|
remaining_space = self.content_h - (y_pos - 2)
|
||||||
|
notes_space = max(1, remaining_space)
|
||||||
|
|
||||||
|
# Calculate which notes to display
|
||||||
|
if selecting_evidence:
|
||||||
|
notes_scroll_offset = 0
|
||||||
|
else:
|
||||||
|
notes_scroll_offset = max(0, (self.selected_index - len(evidence_list)) - notes_space // 2)
|
||||||
|
|
||||||
|
for i in range(notes_space):
|
||||||
|
note_idx = notes_scroll_offset + i
|
||||||
|
if note_idx >= len(case_notes):
|
||||||
|
break
|
||||||
|
|
||||||
|
note = case_notes[note_idx]
|
||||||
|
y = y_pos + i
|
||||||
|
|
||||||
|
# Check if we're out of bounds
|
||||||
|
if y >= self.height - 3:
|
||||||
|
break
|
||||||
|
|
||||||
|
# Format note content
|
||||||
|
note_content = note.content.replace('\n', ' ').replace('\r', ' ')
|
||||||
|
display_str = f"- {note_content}"
|
||||||
|
display_str = self._safe_truncate(display_str, self.width - 6)
|
||||||
|
|
||||||
|
# Highlight if selected
|
||||||
|
item_idx = len(evidence_list) + note_idx
|
||||||
|
if item_idx == self.selected_index:
|
||||||
|
self.stdscr.attron(curses.color_pair(1))
|
||||||
|
self.stdscr.addstr(y, 4, display_str)
|
||||||
|
self.stdscr.attroff(curses.color_pair(1))
|
||||||
|
else:
|
||||||
|
self.stdscr.addstr(y, 4, display_str)
|
||||||
|
|
||||||
self.stdscr.addstr(self.height - 3, 2, "[N] New Evidence [n] Add Note [t] Tags [i] IOCs [v] View Notes [a] Active [d] Delete [?] Help", curses.color_pair(3))
|
self.stdscr.addstr(self.height - 3, 2, "[N] New Evidence [n] Add Note [t] Tags [i] IOCs [v] View Notes [a] Active [d] Delete [?] Help", curses.color_pair(3))
|
||||||
|
|
||||||
def draw_evidence_detail(self):
|
def draw_evidence_detail(self):
|
||||||
@@ -899,11 +914,11 @@ class TUI:
|
|||||||
|
|
||||||
current_y += 1
|
current_y += 1
|
||||||
|
|
||||||
# Content with tag highlighting
|
# Content with tag and IOC highlighting
|
||||||
self.stdscr.addstr(current_y, 2, "Content:", curses.A_BOLD)
|
self.stdscr.addstr(current_y, 2, "Content:", curses.A_BOLD)
|
||||||
current_y += 1
|
current_y += 1
|
||||||
|
|
||||||
# Display content with highlighted tags
|
# Display content with highlighted tags and IOCs
|
||||||
content_lines = self.current_note.content.split('\n')
|
content_lines = self.current_note.content.split('\n')
|
||||||
max_content_lines = self.content_h - (current_y - 2) - 6 # Reserve space for hash/sig
|
max_content_lines = self.content_h - (current_y - 2) - 6 # Reserve space for hash/sig
|
||||||
|
|
||||||
@@ -911,27 +926,72 @@ class TUI:
|
|||||||
if current_y >= self.height - 6:
|
if current_y >= self.height - 6:
|
||||||
break
|
break
|
||||||
|
|
||||||
# Highlight tags in the content
|
# Highlight both tags and IOCs in the content
|
||||||
display_line = self._safe_truncate(line, self.width - 6)
|
display_line = self._safe_truncate(line, self.width - 6)
|
||||||
x_pos = 4
|
x_pos = 4
|
||||||
|
|
||||||
# Simple tag highlighting - split by words and color tags
|
# Extract IOCs and tags from the line
|
||||||
|
from .models import Note
|
||||||
import re
|
import re
|
||||||
parts = re.split(r'(#\w+)', display_line)
|
|
||||||
for part in parts:
|
iocs_found = Note.extract_iocs_from_text(display_line)
|
||||||
if part.startswith('#'):
|
tags_pattern = r'#\w+'
|
||||||
|
tags_found = [(match.group(), match.start()) for match in re.finditer(tags_pattern, display_line)]
|
||||||
|
|
||||||
|
# Combine IOCs and tags into a list of (text, start_pos, type)
|
||||||
|
highlights = []
|
||||||
|
for ioc, _ in iocs_found:
|
||||||
|
pos = display_line.find(ioc)
|
||||||
|
if pos != -1:
|
||||||
|
highlights.append((ioc, pos, 'ioc'))
|
||||||
|
for tag, pos in tags_found:
|
||||||
|
highlights.append((tag, pos, 'tag'))
|
||||||
|
|
||||||
|
# Sort by position
|
||||||
|
highlights.sort(key=lambda x: x[1])
|
||||||
|
|
||||||
|
if highlights:
|
||||||
|
# Display with highlighting
|
||||||
|
remaining = display_line
|
||||||
|
for i, (text, orig_pos, htype) in enumerate(highlights):
|
||||||
|
# Find position in remaining text
|
||||||
|
pos = remaining.find(text)
|
||||||
|
if pos == -1:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Print text before highlight
|
||||||
|
if pos > 0:
|
||||||
|
try:
|
||||||
|
self.stdscr.addstr(current_y, x_pos, remaining[:pos])
|
||||||
|
x_pos += pos
|
||||||
|
except curses.error:
|
||||||
|
break
|
||||||
|
|
||||||
|
# Print highlighted text
|
||||||
try:
|
try:
|
||||||
self.stdscr.addstr(current_y, x_pos, part, curses.color_pair(3))
|
if htype == 'ioc':
|
||||||
|
self.stdscr.addstr(current_y, x_pos, text, curses.color_pair(4))
|
||||||
|
else: # tag
|
||||||
|
self.stdscr.addstr(current_y, x_pos, text, curses.color_pair(3))
|
||||||
|
x_pos += len(text)
|
||||||
|
except curses.error:
|
||||||
|
break
|
||||||
|
|
||||||
|
# Update remaining text
|
||||||
|
remaining = remaining[pos + len(text):]
|
||||||
|
|
||||||
|
# Print any remaining text
|
||||||
|
if remaining and x_pos < self.width - 2:
|
||||||
|
try:
|
||||||
|
self.stdscr.addstr(current_y, x_pos, remaining[:self.width - x_pos - 2])
|
||||||
except curses.error:
|
except curses.error:
|
||||||
pass
|
pass
|
||||||
x_pos += len(part)
|
else:
|
||||||
else:
|
# No highlights, display normally
|
||||||
if x_pos < self.width - 2:
|
try:
|
||||||
try:
|
self.stdscr.addstr(current_y, x_pos, display_line)
|
||||||
self.stdscr.addstr(current_y, x_pos, part[:self.width - x_pos - 2])
|
except curses.error:
|
||||||
except curses.error:
|
pass
|
||||||
pass
|
|
||||||
x_pos += len(part)
|
|
||||||
|
|
||||||
current_y += 1
|
current_y += 1
|
||||||
|
|
||||||
@@ -984,11 +1044,11 @@ class TUI:
|
|||||||
help_lines.append((" n Add note to case", curses.A_NORMAL))
|
help_lines.append((" n Add note to case", curses.A_NORMAL))
|
||||||
help_lines.append((" t View tags across case and all evidence", curses.A_NORMAL))
|
help_lines.append((" t View tags across case and all evidence", curses.A_NORMAL))
|
||||||
help_lines.append((" i View IOCs across case and all evidence", curses.A_NORMAL))
|
help_lines.append((" i View IOCs across case and all evidence", curses.A_NORMAL))
|
||||||
help_lines.append((" v View all case notes", curses.A_NORMAL))
|
help_lines.append((" v View all case notes with IOC highlighting", curses.A_NORMAL))
|
||||||
help_lines.append((" a Set case (or selected evidence) as active", curses.A_NORMAL))
|
help_lines.append((" a Set case (or selected evidence) as active", curses.A_NORMAL))
|
||||||
help_lines.append((" d Delete selected evidence item", curses.A_NORMAL))
|
help_lines.append((" d Delete selected evidence item or note", curses.A_NORMAL))
|
||||||
help_lines.append((" / Filter evidence by name or description", curses.A_NORMAL))
|
help_lines.append((" / Filter evidence by name or description", curses.A_NORMAL))
|
||||||
help_lines.append((" Enter Open evidence details", curses.A_NORMAL))
|
help_lines.append((" Enter Open evidence details or jump to note", curses.A_NORMAL))
|
||||||
help_lines.append(("", curses.A_NORMAL))
|
help_lines.append(("", curses.A_NORMAL))
|
||||||
|
|
||||||
# Evidence Detail View
|
# Evidence Detail View
|
||||||
@@ -996,9 +1056,10 @@ class TUI:
|
|||||||
help_lines.append((" n Add note to evidence", curses.A_NORMAL))
|
help_lines.append((" n Add note to evidence", curses.A_NORMAL))
|
||||||
help_lines.append((" t View tags for this evidence", curses.A_NORMAL))
|
help_lines.append((" t View tags for this evidence", curses.A_NORMAL))
|
||||||
help_lines.append((" i View IOCs for this evidence", curses.A_NORMAL))
|
help_lines.append((" i View IOCs for this evidence", curses.A_NORMAL))
|
||||||
help_lines.append((" v View all evidence notes", curses.A_NORMAL))
|
help_lines.append((" v View all evidence notes with IOC highlighting", curses.A_NORMAL))
|
||||||
help_lines.append((" a Set evidence as active context", curses.A_NORMAL))
|
help_lines.append((" a Set evidence as active context", curses.A_NORMAL))
|
||||||
help_lines.append((" d Delete selected note", curses.A_NORMAL))
|
help_lines.append((" d Delete selected note", curses.A_NORMAL))
|
||||||
|
help_lines.append((" Enter Jump to selected note in full view", curses.A_NORMAL))
|
||||||
help_lines.append(("", curses.A_NORMAL))
|
help_lines.append(("", curses.A_NORMAL))
|
||||||
|
|
||||||
# Tags View
|
# Tags View
|
||||||
@@ -1028,7 +1089,11 @@ class TUI:
|
|||||||
help_lines.append((" Active Context Set with 'a' key - enables CLI quick notes", curses.A_NORMAL))
|
help_lines.append((" Active Context Set with 'a' key - enables CLI quick notes", curses.A_NORMAL))
|
||||||
help_lines.append((" Run: trace \"your note text\"", curses.A_DIM))
|
help_lines.append((" Run: trace \"your note text\"", curses.A_DIM))
|
||||||
help_lines.append((" Tags Use #hashtag in notes for auto-tagging", curses.A_NORMAL))
|
help_lines.append((" Tags Use #hashtag in notes for auto-tagging", curses.A_NORMAL))
|
||||||
|
help_lines.append((" Highlighted in cyan throughout the interface", curses.A_DIM))
|
||||||
help_lines.append((" IOCs Auto-extracts IPs, domains, URLs, hashes, emails", curses.A_NORMAL))
|
help_lines.append((" IOCs Auto-extracts IPs, domains, URLs, hashes, emails", curses.A_NORMAL))
|
||||||
|
help_lines.append((" Highlighted in red in full note views", curses.A_DIM))
|
||||||
|
help_lines.append((" Note Navigation Press Enter on any note to view with highlighting", curses.A_NORMAL))
|
||||||
|
help_lines.append((" Selected note auto-centered and highlighted", curses.A_DIM))
|
||||||
help_lines.append((" Integrity All notes SHA256 hashed + optional GPG signing", curses.A_NORMAL))
|
help_lines.append((" Integrity All notes SHA256 hashed + optional GPG signing", curses.A_NORMAL))
|
||||||
help_lines.append((" GPG Settings Press 's' to toggle signing & select GPG key", curses.A_NORMAL))
|
help_lines.append((" GPG Settings Press 's' to toggle signing & select GPG key", curses.A_NORMAL))
|
||||||
help_lines.append((" Source Hash Store evidence file hashes for chain of custody", curses.A_NORMAL))
|
help_lines.append((" Source Hash Store evidence file hashes for chain of custody", curses.A_NORMAL))
|
||||||
@@ -1148,10 +1213,10 @@ class TUI:
|
|||||||
filtered = self._get_filtered_list(self.cases, "case_number", "name")
|
filtered = self._get_filtered_list(self.cases, "case_number", "name")
|
||||||
max_idx = len(filtered) - 1
|
max_idx = len(filtered) - 1
|
||||||
elif self.current_view == "case_detail" and self.active_case:
|
elif self.current_view == "case_detail" and self.active_case:
|
||||||
# Total items = case notes + evidence
|
# Total items = evidence + case notes
|
||||||
case_notes = self.active_case.notes
|
case_notes = self.active_case.notes
|
||||||
filtered = self._get_filtered_list(self.active_case.evidence, "name", "description")
|
filtered = self._get_filtered_list(self.active_case.evidence, "name", "description")
|
||||||
max_idx = len(case_notes) + len(filtered) - 1
|
max_idx = len(filtered) + len(case_notes) - 1
|
||||||
elif self.current_view == "evidence_detail" and self.active_evidence:
|
elif self.current_view == "evidence_detail" and self.active_evidence:
|
||||||
# Navigate through notes in evidence detail view
|
# Navigate through notes in evidence detail view
|
||||||
max_idx = len(self.active_evidence.notes) - 1
|
max_idx = len(self.active_evidence.notes) - 1
|
||||||
@@ -1185,25 +1250,55 @@ class TUI:
|
|||||||
self.current_view = "case_detail"
|
self.current_view = "case_detail"
|
||||||
self.selected_index = 0
|
self.selected_index = 0
|
||||||
self.scroll_offset = 0
|
self.scroll_offset = 0
|
||||||
self.filter_query = "" # Reset filter on view change
|
self.filter_query = ""
|
||||||
|
elif self.current_view == "evidence_detail" and self.active_evidence:
|
||||||
|
# Check if a note is selected
|
||||||
|
notes = self.active_evidence.notes
|
||||||
|
list_h = self.content_h - 5
|
||||||
|
display_notes = notes[-list_h:] if len(notes) > list_h else notes
|
||||||
|
|
||||||
|
if display_notes and self.selected_index < len(display_notes):
|
||||||
|
# Calculate the actual note index in the full list
|
||||||
|
note_offset = len(notes) - len(display_notes)
|
||||||
|
actual_note_index = note_offset + self.selected_index
|
||||||
|
# Open notes view and jump to selected note
|
||||||
|
self._highlight_note_idx = actual_note_index
|
||||||
|
self.view_evidence_notes(highlight_note_index=actual_note_index)
|
||||||
|
delattr(self, '_highlight_note_idx') # Reset filter on view change
|
||||||
elif self.current_view == "case_detail":
|
elif self.current_view == "case_detail":
|
||||||
if self.active_case:
|
if self.active_case:
|
||||||
case_notes = self.active_case.notes
|
case_notes = self.active_case.notes
|
||||||
filtered = self._get_filtered_list(self.active_case.evidence, "name", "description")
|
filtered = self._get_filtered_list(self.active_case.evidence, "name", "description")
|
||||||
|
|
||||||
# Check if selecting a note or evidence
|
# Check if a note is selected
|
||||||
if self.selected_index < len(case_notes):
|
if self.selected_index < len(case_notes):
|
||||||
# Selected a note - show note detail view
|
# Open notes view and jump to selected note
|
||||||
self.current_note = case_notes[self.selected_index]
|
self._highlight_note_idx = self.selected_index
|
||||||
self.previous_view = "case_detail"
|
self.view_case_notes(highlight_note_index=self.selected_index)
|
||||||
self.current_view = "note_detail"
|
delattr(self, '_highlight_note_idx')
|
||||||
self.filter_query = ""
|
elif self.selected_index - len(case_notes) < len(filtered):
|
||||||
elif filtered and self.selected_index - len(case_notes) < len(filtered):
|
# Evidence selected - open it
|
||||||
# Selected evidence - navigate to evidence detail
|
|
||||||
evidence_idx = self.selected_index - len(case_notes)
|
evidence_idx = self.selected_index - len(case_notes)
|
||||||
self.active_evidence = filtered[evidence_idx]
|
self.active_evidence = filtered[evidence_idx]
|
||||||
self.current_view = "evidence_detail"
|
self.current_view = "evidence_detail"
|
||||||
self.selected_index = 0
|
self.selected_index = 0
|
||||||
|
self.scroll_offset = 0
|
||||||
|
case_notes = self.active_case.notes
|
||||||
|
filtered = self._get_filtered_list(self.active_case.evidence, "name", "description")
|
||||||
|
|
||||||
|
# Check if selecting evidence or note
|
||||||
|
if self.selected_index < len(filtered):
|
||||||
|
# Selected evidence - navigate to evidence detail
|
||||||
|
self.active_evidence = filtered[self.selected_index]
|
||||||
|
self.current_view = "evidence_detail"
|
||||||
|
self.selected_index = 0
|
||||||
|
self.filter_query = ""
|
||||||
|
elif case_notes and self.selected_index - len(filtered) < len(case_notes):
|
||||||
|
# Selected a note - show note detail view
|
||||||
|
note_idx = self.selected_index - len(filtered)
|
||||||
|
self.current_note = case_notes[note_idx]
|
||||||
|
self.previous_view = "case_detail"
|
||||||
|
self.current_view = "note_detail"
|
||||||
self.filter_query = ""
|
self.filter_query = ""
|
||||||
elif self.current_view == "tags_list":
|
elif self.current_view == "tags_list":
|
||||||
# Enter tag -> show notes with that tag
|
# Enter tag -> show notes with that tag
|
||||||
@@ -1376,21 +1471,21 @@ class TUI:
|
|||||||
case_notes = self.active_case.notes
|
case_notes = self.active_case.notes
|
||||||
filtered = self._get_filtered_list(self.active_case.evidence, "name", "description")
|
filtered = self._get_filtered_list(self.active_case.evidence, "name", "description")
|
||||||
|
|
||||||
# Only allow setting active for evidence, not notes
|
# Evidence is displayed first (indices 0 to len(evidence)-1)
|
||||||
if self.selected_index < len(case_notes):
|
# Case notes are displayed second (indices len(evidence) to len(evidence)+len(notes)-1)
|
||||||
|
if self.selected_index < len(filtered):
|
||||||
|
# Selected evidence - set it as active
|
||||||
|
ev = filtered[self.selected_index]
|
||||||
|
self.state_manager.set_active(case_id=self.active_case.case_id, evidence_id=ev.evidence_id)
|
||||||
|
self.global_active_case_id = self.active_case.case_id
|
||||||
|
self.global_active_evidence_id = ev.evidence_id
|
||||||
|
self.show_message(f"Active: {ev.name}")
|
||||||
|
elif case_notes and self.selected_index - len(filtered) < len(case_notes):
|
||||||
# Selected a note - set case as active (not evidence)
|
# Selected a note - set case as active (not evidence)
|
||||||
self.state_manager.set_active(case_id=self.active_case.case_id, evidence_id=None)
|
self.state_manager.set_active(case_id=self.active_case.case_id, evidence_id=None)
|
||||||
self.global_active_case_id = self.active_case.case_id
|
self.global_active_case_id = self.active_case.case_id
|
||||||
self.global_active_evidence_id = None
|
self.global_active_evidence_id = None
|
||||||
self.show_message(f"Active: Case {self.active_case.case_number}")
|
self.show_message(f"Active: Case {self.active_case.case_number}")
|
||||||
elif filtered and self.selected_index - len(case_notes) < len(filtered):
|
|
||||||
# Selected evidence - set it as active
|
|
||||||
evidence_idx = self.selected_index - len(case_notes)
|
|
||||||
ev = filtered[evidence_idx]
|
|
||||||
self.state_manager.set_active(case_id=self.active_case.case_id, evidence_id=ev.evidence_id)
|
|
||||||
self.global_active_case_id = self.active_case.case_id
|
|
||||||
self.global_active_evidence_id = ev.evidence_id
|
|
||||||
self.show_message(f"Active: {ev.name}")
|
|
||||||
else:
|
else:
|
||||||
# Nothing selected - set case as active
|
# Nothing selected - set case as active
|
||||||
self.state_manager.set_active(case_id=self.active_case.case_id, evidence_id=None)
|
self.state_manager.set_active(case_id=self.active_case.case_id, evidence_id=None)
|
||||||
@@ -1653,7 +1748,7 @@ class TUI:
|
|||||||
win.addstr(y, 2, " " * input_width)
|
win.addstr(y, 2, " " * input_width)
|
||||||
|
|
||||||
if line_idx < len(lines):
|
if line_idx < len(lines):
|
||||||
# Show line content
|
# Show line content (truncated if too long)
|
||||||
display_text = lines[line_idx][:input_width]
|
display_text = lines[line_idx][:input_width]
|
||||||
win.addstr(y, 2, display_text)
|
win.addstr(y, 2, display_text)
|
||||||
|
|
||||||
@@ -1775,6 +1870,18 @@ class TUI:
|
|||||||
line = lines[cursor_line]
|
line = lines[cursor_line]
|
||||||
lines[cursor_line] = line[:cursor_col] + chr(ch) + line[cursor_col:]
|
lines[cursor_line] = line[:cursor_col] + chr(ch) + line[cursor_col:]
|
||||||
cursor_col += 1
|
cursor_col += 1
|
||||||
|
|
||||||
|
# Auto-wrap to next line if cursor exceeds visible width
|
||||||
|
if cursor_col >= input_width:
|
||||||
|
# Always ensure there's a next line to move to
|
||||||
|
if cursor_line >= len(lines) - 1:
|
||||||
|
# We're on the last line, add a new line
|
||||||
|
lines.append("")
|
||||||
|
cursor_line += 1
|
||||||
|
cursor_col = 0
|
||||||
|
# Adjust scroll if needed
|
||||||
|
if cursor_line >= scroll_offset + input_height:
|
||||||
|
scroll_offset = cursor_line - input_height + 1
|
||||||
|
|
||||||
def dialog_confirm(self, message):
|
def dialog_confirm(self, message):
|
||||||
curses.curs_set(0)
|
curses.curs_set(0)
|
||||||
@@ -1818,7 +1925,7 @@ class TUI:
|
|||||||
x = (self.width - w) // 2
|
x = (self.width - w) // 2
|
||||||
|
|
||||||
win = curses.newwin(h, w, y, x)
|
win = curses.newwin(h, w, y, x)
|
||||||
win.keypad(1) # Enable keypad mode for arrow keys
|
win.keypad(True) # Enable keypad mode for arrow keys
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
win.clear()
|
win.clear()
|
||||||
@@ -1915,7 +2022,7 @@ class TUI:
|
|||||||
x = (self.width - w) // 2
|
x = (self.width - w) // 2
|
||||||
|
|
||||||
win = curses.newwin(h, w, y, x)
|
win = curses.newwin(h, w, y, x)
|
||||||
win.keypad(1) # Enable keypad mode for arrow keys
|
win.keypad(True) # Enable keypad mode for arrow keys
|
||||||
scroll_offset = 0
|
scroll_offset = 0
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
@@ -2036,14 +2143,10 @@ class TUI:
|
|||||||
return
|
return
|
||||||
|
|
||||||
desc = self._input_dialog("New Evidence - Step 2/3", "Enter description (optional):")
|
desc = self._input_dialog("New Evidence - Step 2/3", "Enter description (optional):")
|
||||||
if desc is None:
|
# For optional fields, treat None as empty string (user pressed Enter on empty field)
|
||||||
self.show_message("Evidence creation cancelled.")
|
|
||||||
return
|
|
||||||
|
|
||||||
source_hash = self._input_dialog("New Evidence - Step 3/3", "Enter source hash (optional, e.g. SHA256):")
|
source_hash = self._input_dialog("New Evidence - Step 3/3", "Enter source hash (optional, e.g. SHA256):")
|
||||||
if source_hash is None:
|
# For optional fields, treat None as empty string (user pressed Enter on empty field)
|
||||||
self.show_message("Evidence creation cancelled.")
|
|
||||||
return
|
|
||||||
|
|
||||||
ev = Evidence(name=name, description=desc or "")
|
ev = Evidence(name=name, description=desc or "")
|
||||||
if source_hash:
|
if source_hash:
|
||||||
@@ -2062,12 +2165,12 @@ class TUI:
|
|||||||
|
|
||||||
if self.current_view == "evidence_detail" and self.active_evidence:
|
if self.current_view == "evidence_detail" and self.active_evidence:
|
||||||
context_title = f"Add Note → Evidence: {self.active_evidence.name}"
|
context_title = f"Add Note → Evidence: {self.active_evidence.name}"
|
||||||
context_prompt = f"Case: {self.active_case.case_number if self.active_case else '?'}\nEvidence: {self.active_evidence.name}\n\nNote will be added to this evidence."
|
context_prompt = f"Case: {self.active_case.case_number if self.active_case else '?'}\nEvidence: {self.active_evidence.name}\n"
|
||||||
recent_notes = self.active_evidence.notes[-5:] if len(self.active_evidence.notes) > 0 else []
|
recent_notes = self.active_evidence.notes[-5:] if len(self.active_evidence.notes) > 0 else []
|
||||||
target_evidence = self.active_evidence
|
target_evidence = self.active_evidence
|
||||||
elif self.current_view == "case_detail" and self.active_case:
|
elif self.current_view == "case_detail" and self.active_case:
|
||||||
context_title = f"Add Note → Case: {self.active_case.case_number}"
|
context_title = f"Add Note → Case: {self.active_case.case_number}"
|
||||||
context_prompt = f"Case: {self.active_case.case_number}\n{self.active_case.name if self.active_case.name else ''}\n\nNote will be added to case notes."
|
context_prompt = f"Case: {self.active_case.case_number}\n{self.active_case.name if self.active_case.name else ''}\nNote will be added to case notes."
|
||||||
recent_notes = self.active_case.notes[-5:] if len(self.active_case.notes) > 0 else []
|
recent_notes = self.active_case.notes[-5:] if len(self.active_case.notes) > 0 else []
|
||||||
target_case = self.active_case
|
target_case = self.active_case
|
||||||
elif self.current_view == "case_list":
|
elif self.current_view == "case_list":
|
||||||
@@ -2080,14 +2183,14 @@ class TUI:
|
|||||||
for ev in active_case.evidence:
|
for ev in active_case.evidence:
|
||||||
if ev.evidence_id == self.global_active_evidence_id:
|
if ev.evidence_id == self.global_active_evidence_id:
|
||||||
context_title = f"Add Note → Evidence: {ev.name}"
|
context_title = f"Add Note → Evidence: {ev.name}"
|
||||||
context_prompt = f"Case: {active_case.case_number}\nEvidence: {ev.name}\n\nNote will be added to this evidence."
|
context_prompt = f"Case: {active_case.case_number}\nEvidence: {ev.name}\n"
|
||||||
recent_notes = ev.notes[-5:] if len(ev.notes) > 0 else []
|
recent_notes = ev.notes[-5:] if len(ev.notes) > 0 else []
|
||||||
target_case = active_case
|
target_case = active_case
|
||||||
target_evidence = ev
|
target_evidence = ev
|
||||||
break
|
break
|
||||||
else:
|
else:
|
||||||
context_title = f"Add Note → Case: {active_case.case_number}"
|
context_title = f"Add Note → Case: {active_case.case_number}"
|
||||||
context_prompt = f"Case: {active_case.case_number}\n\nNote will be added to case notes."
|
context_prompt = f"Case: {active_case.case_number}\nNote will be added to case notes."
|
||||||
recent_notes = active_case.notes[-5:] if len(active_case.notes) > 0 else []
|
recent_notes = active_case.notes[-5:] if len(active_case.notes) > 0 else []
|
||||||
target_case = active_case
|
target_case = active_case
|
||||||
|
|
||||||
@@ -2120,7 +2223,7 @@ class TUI:
|
|||||||
|
|
||||||
signed = False
|
signed = False
|
||||||
if pgp_enabled:
|
if pgp_enabled:
|
||||||
sig = Crypto.sign_content(f"Hash: {note.content_hash}\nContent: {note.content}", key_id=gpg_key_id)
|
sig = Crypto.sign_content(f"Hash: {note.content_hash}\nContent: {note.content}", key_id=gpg_key_id or "")
|
||||||
if sig:
|
if sig:
|
||||||
note.signature = sig
|
note.signature = sig
|
||||||
signed = True
|
signed = True
|
||||||
@@ -2219,7 +2322,7 @@ class TUI:
|
|||||||
self.scroll_offset = 0
|
self.scroll_offset = 0
|
||||||
self.show_message("Note deleted.")
|
self.show_message("Note deleted.")
|
||||||
|
|
||||||
def view_case_notes(self):
|
def view_case_notes(self, highlight_note_index=None):
|
||||||
if not self.active_case: return
|
if not self.active_case: return
|
||||||
|
|
||||||
h = int(self.height * 0.8)
|
h = int(self.height * 0.8)
|
||||||
@@ -2227,45 +2330,151 @@ class TUI:
|
|||||||
y = int(self.height * 0.1)
|
y = int(self.height * 0.1)
|
||||||
x = int(self.width * 0.1)
|
x = int(self.width * 0.1)
|
||||||
|
|
||||||
|
scroll_offset = 0
|
||||||
|
highlight_idx = highlight_note_index # Store for persistent highlighting
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
win = curses.newwin(h, w, y, x)
|
win = curses.newwin(h, w, y, x)
|
||||||
|
win.keypad(True)
|
||||||
win.box()
|
win.box()
|
||||||
win.addstr(1, 2, f"Notes: {self.active_case.case_number}", curses.A_BOLD)
|
win.addstr(1, 2, f"Notes: {self.active_case.case_number} ({len(self.active_case.notes)} total)", curses.A_BOLD)
|
||||||
|
|
||||||
notes = self.active_case.notes
|
notes = self.active_case.notes
|
||||||
max_lines = h - 4
|
content_lines = []
|
||||||
|
note_line_ranges = [] # Track which lines belong to which note
|
||||||
|
|
||||||
|
# Build all content lines with separators between notes
|
||||||
|
for note_idx, note in enumerate(notes):
|
||||||
|
start_line = len(content_lines)
|
||||||
|
timestamp_str = time.ctime(note.timestamp)
|
||||||
|
content_lines.append(f"[{timestamp_str}]")
|
||||||
|
# Split multi-line notes and wrap long lines
|
||||||
|
for line in note.content.split('\n'):
|
||||||
|
# Wrap long lines
|
||||||
|
while len(line) > w - 6:
|
||||||
|
content_lines.append(" " + line[:w-6])
|
||||||
|
line = line[w-6:]
|
||||||
|
content_lines.append(" " + line)
|
||||||
|
content_lines.append("") # Blank line between notes
|
||||||
|
end_line = len(content_lines) - 1
|
||||||
|
note_line_ranges.append((start_line, end_line, note_idx))
|
||||||
|
|
||||||
# Scroll last notes
|
max_display_lines = h - 5
|
||||||
display_notes = notes[-max_lines:] if len(notes) > max_lines else notes
|
total_lines = len(content_lines)
|
||||||
|
|
||||||
for i, note in enumerate(display_notes):
|
# Jump to highlighted note on first render
|
||||||
# Replace newlines with spaces for single-line display
|
if highlight_note_index is not None and note_line_ranges:
|
||||||
note_content = note.content.replace('\n', ' ').replace('\r', ' ')
|
for start, end, idx in note_line_ranges:
|
||||||
display_str = f"- [{time.ctime(note.timestamp)}] {note_content}"
|
if idx == highlight_note_index:
|
||||||
# Truncate safely for Unicode
|
# Center the note in the view
|
||||||
display_str = self._safe_truncate(display_str, w - 4)
|
note_middle = (start + end) // 2
|
||||||
win.addstr(3 + i, 2, display_str)
|
scroll_offset = max(0, note_middle - max_display_lines // 2)
|
||||||
|
highlight_note_index = None # Only jump once
|
||||||
|
break
|
||||||
|
|
||||||
win.addstr(h-2, 2, "[n] Add Note [b/q/Esc] Close", curses.color_pair(3))
|
# Adjust scroll bounds
|
||||||
|
max_scroll = max(0, total_lines - max_display_lines)
|
||||||
|
scroll_offset = max(0, min(scroll_offset, max_scroll))
|
||||||
|
|
||||||
|
# Display lines with highlighting
|
||||||
|
for i in range(max_display_lines):
|
||||||
|
line_idx = scroll_offset + i
|
||||||
|
if line_idx >= total_lines:
|
||||||
|
break
|
||||||
|
display_line = self._safe_truncate(content_lines[line_idx], w - 4)
|
||||||
|
|
||||||
|
# Check if this line belongs to the highlighted note
|
||||||
|
is_highlighted = False
|
||||||
|
if highlight_idx is not None:
|
||||||
|
for start, end, idx in note_line_ranges:
|
||||||
|
if start <= line_idx <= end and idx == highlight_idx:
|
||||||
|
is_highlighted = True
|
||||||
|
break
|
||||||
|
|
||||||
|
try:
|
||||||
|
y_pos = 3 + i
|
||||||
|
if is_highlighted:
|
||||||
|
# Highlight entire line for selected note
|
||||||
|
win.addstr(y_pos, 2, display_line, curses.color_pair(1))
|
||||||
|
else:
|
||||||
|
# Check for IOCs in the line and highlight them
|
||||||
|
from .models import Note
|
||||||
|
iocs_found = Note.extract_iocs_from_text(display_line)
|
||||||
|
|
||||||
|
if iocs_found:
|
||||||
|
# Display with IOC highlighting
|
||||||
|
x_pos = 2
|
||||||
|
remaining = display_line
|
||||||
|
while iocs_found and remaining:
|
||||||
|
# Find the earliest IOC in the remaining text
|
||||||
|
earliest_ioc = None
|
||||||
|
earliest_pos = len(remaining)
|
||||||
|
for ioc, _ in iocs_found:
|
||||||
|
pos = remaining.find(ioc)
|
||||||
|
if pos != -1 and pos < earliest_pos:
|
||||||
|
earliest_pos = pos
|
||||||
|
earliest_ioc = ioc
|
||||||
|
|
||||||
|
if earliest_ioc:
|
||||||
|
# Print text before IOC
|
||||||
|
if earliest_pos > 0:
|
||||||
|
win.addstr(y_pos, x_pos, remaining[:earliest_pos])
|
||||||
|
x_pos += earliest_pos
|
||||||
|
# Print IOC in color
|
||||||
|
win.addstr(y_pos, x_pos, earliest_ioc, curses.color_pair(4))
|
||||||
|
x_pos += len(earliest_ioc)
|
||||||
|
# Update remaining text
|
||||||
|
remaining = remaining[earliest_pos + len(earliest_ioc):]
|
||||||
|
# Remove found IOC from list
|
||||||
|
iocs_found = [(ioc, t) for ioc, t in iocs_found if ioc != earliest_ioc]
|
||||||
|
else:
|
||||||
|
break
|
||||||
|
# Print any remaining text
|
||||||
|
if remaining:
|
||||||
|
win.addstr(y_pos, x_pos, remaining)
|
||||||
|
else:
|
||||||
|
# No IOCs, display normally
|
||||||
|
win.addstr(y_pos, 2, display_line)
|
||||||
|
except curses.error:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Show scroll indicator
|
||||||
|
if total_lines > max_display_lines:
|
||||||
|
scroll_info = f"[{scroll_offset + 1}-{min(scroll_offset + max_display_lines, total_lines)}/{total_lines}]"
|
||||||
|
try:
|
||||||
|
win.addstr(2, w - len(scroll_info) - 3, scroll_info, curses.A_DIM)
|
||||||
|
except curses.error:
|
||||||
|
pass
|
||||||
|
|
||||||
|
win.addstr(h-2, 2, "[↑↓] Scroll [n] Add Note [b/q/Esc] Close", curses.color_pair(3))
|
||||||
win.refresh()
|
win.refresh()
|
||||||
key = win.getch()
|
key = win.getch()
|
||||||
del win
|
del win
|
||||||
|
|
||||||
# Handle key presses
|
# Handle key presses
|
||||||
if key == ord('n') or key == ord('N'):
|
if key == curses.KEY_UP:
|
||||||
|
scroll_offset = max(0, scroll_offset - 1)
|
||||||
|
elif key == curses.KEY_DOWN:
|
||||||
|
scroll_offset = min(max_scroll, scroll_offset + 1)
|
||||||
|
elif key == curses.KEY_PPAGE: # Page Up
|
||||||
|
scroll_offset = max(0, scroll_offset - max_display_lines)
|
||||||
|
elif key == curses.KEY_NPAGE: # Page Down
|
||||||
|
scroll_offset = min(max_scroll, scroll_offset + max_display_lines)
|
||||||
|
elif key == curses.KEY_HOME:
|
||||||
|
scroll_offset = 0
|
||||||
|
elif key == curses.KEY_END:
|
||||||
|
scroll_offset = max_scroll
|
||||||
|
elif key == ord('n') or key == ord('N'):
|
||||||
# Save current view and switch to case_detail temporarily for context
|
# Save current view and switch to case_detail temporarily for context
|
||||||
saved_view = self.current_view
|
saved_view = self.current_view
|
||||||
self.current_view = "case_detail"
|
self.current_view = "case_detail"
|
||||||
self.dialog_add_note()
|
self.dialog_add_note()
|
||||||
self.current_view = saved_view
|
self.current_view = saved_view
|
||||||
# Continue loop to refresh with new note
|
scroll_offset = max_scroll # Jump to bottom to show new note
|
||||||
elif key == ord('b') or key == ord('B') or key == ord('q') or key == ord('Q') or key == 27: # 27 is Esc
|
elif key == ord('b') or key == ord('B') or key == ord('q') or key == ord('Q') or key == 27: # 27 is Esc
|
||||||
break
|
break
|
||||||
else:
|
|
||||||
# Any other key also closes (backwards compatibility)
|
|
||||||
break
|
|
||||||
|
|
||||||
def view_evidence_notes(self):
|
def view_evidence_notes(self, highlight_note_index=None):
|
||||||
if not self.active_evidence: return
|
if not self.active_evidence: return
|
||||||
|
|
||||||
h = int(self.height * 0.8)
|
h = int(self.height * 0.8)
|
||||||
@@ -2273,43 +2482,149 @@ class TUI:
|
|||||||
y = int(self.height * 0.1)
|
y = int(self.height * 0.1)
|
||||||
x = int(self.width * 0.1)
|
x = int(self.width * 0.1)
|
||||||
|
|
||||||
|
scroll_offset = 0
|
||||||
|
highlight_idx = highlight_note_index # Store for persistent highlighting
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
win = curses.newwin(h, w, y, x)
|
win = curses.newwin(h, w, y, x)
|
||||||
|
win.keypad(True)
|
||||||
win.box()
|
win.box()
|
||||||
win.addstr(1, 2, f"Notes: {self.active_evidence.name}", curses.A_BOLD)
|
win.addstr(1, 2, f"Notes: {self.active_evidence.name} ({len(self.active_evidence.notes)} total)", curses.A_BOLD)
|
||||||
|
|
||||||
notes = self.active_evidence.notes
|
notes = self.active_evidence.notes
|
||||||
max_lines = h - 4
|
content_lines = []
|
||||||
|
note_line_ranges = [] # Track which lines belong to which note
|
||||||
|
|
||||||
|
# Build all content lines with separators between notes
|
||||||
|
for note_idx, note in enumerate(notes):
|
||||||
|
start_line = len(content_lines)
|
||||||
|
timestamp_str = time.ctime(note.timestamp)
|
||||||
|
content_lines.append(f"[{timestamp_str}]")
|
||||||
|
# Split multi-line notes and wrap long lines
|
||||||
|
for line in note.content.split('\n'):
|
||||||
|
# Wrap long lines
|
||||||
|
while len(line) > w - 6:
|
||||||
|
content_lines.append(" " + line[:w-6])
|
||||||
|
line = line[w-6:]
|
||||||
|
content_lines.append(" " + line)
|
||||||
|
content_lines.append("") # Blank line between notes
|
||||||
|
end_line = len(content_lines) - 1
|
||||||
|
note_line_ranges.append((start_line, end_line, note_idx))
|
||||||
|
|
||||||
# Scroll last notes
|
max_display_lines = h - 5
|
||||||
display_notes = notes[-max_lines:] if len(notes) > max_lines else notes
|
total_lines = len(content_lines)
|
||||||
|
|
||||||
for i, note in enumerate(display_notes):
|
# Jump to highlighted note on first render
|
||||||
# Replace newlines with spaces for single-line display
|
if highlight_note_index is not None and note_line_ranges:
|
||||||
note_content = note.content.replace('\n', ' ').replace('\r', ' ')
|
for start, end, idx in note_line_ranges:
|
||||||
display_str = f"- [{time.ctime(note.timestamp)}] {note_content}"
|
if idx == highlight_note_index:
|
||||||
# Truncate safely for Unicode
|
# Center the note in the view
|
||||||
display_str = self._safe_truncate(display_str, w - 4)
|
note_middle = (start + end) // 2
|
||||||
win.addstr(3 + i, 2, display_str)
|
scroll_offset = max(0, note_middle - max_display_lines // 2)
|
||||||
|
highlight_note_index = None # Only jump once
|
||||||
|
break
|
||||||
|
|
||||||
win.addstr(h-2, 2, "[n] Add Note [b/q/Esc] Close", curses.color_pair(3))
|
# Adjust scroll bounds
|
||||||
|
max_scroll = max(0, total_lines - max_display_lines)
|
||||||
|
scroll_offset = max(0, min(scroll_offset, max_scroll))
|
||||||
|
|
||||||
|
# Display lines with highlighting
|
||||||
|
for i in range(max_display_lines):
|
||||||
|
line_idx = scroll_offset + i
|
||||||
|
if line_idx >= total_lines:
|
||||||
|
break
|
||||||
|
display_line = self._safe_truncate(content_lines[line_idx], w - 4)
|
||||||
|
|
||||||
|
# Check if this line belongs to the highlighted note
|
||||||
|
is_highlighted = False
|
||||||
|
if highlight_idx is not None:
|
||||||
|
for start, end, idx in note_line_ranges:
|
||||||
|
if start <= line_idx <= end and idx == highlight_idx:
|
||||||
|
is_highlighted = True
|
||||||
|
break
|
||||||
|
|
||||||
|
try:
|
||||||
|
y_pos = 3 + i
|
||||||
|
if is_highlighted:
|
||||||
|
# Highlight entire line for selected note
|
||||||
|
win.addstr(y_pos, 2, display_line, curses.color_pair(1))
|
||||||
|
else:
|
||||||
|
# Check for IOCs in the line and highlight them
|
||||||
|
from .models import Note
|
||||||
|
iocs_found = Note.extract_iocs_from_text(display_line)
|
||||||
|
|
||||||
|
if iocs_found:
|
||||||
|
# Display with IOC highlighting
|
||||||
|
x_pos = 2
|
||||||
|
remaining = display_line
|
||||||
|
while iocs_found and remaining:
|
||||||
|
# Find the earliest IOC in the remaining text
|
||||||
|
earliest_ioc = None
|
||||||
|
earliest_pos = len(remaining)
|
||||||
|
for ioc, _ in iocs_found:
|
||||||
|
pos = remaining.find(ioc)
|
||||||
|
if pos != -1 and pos < earliest_pos:
|
||||||
|
earliest_pos = pos
|
||||||
|
earliest_ioc = ioc
|
||||||
|
|
||||||
|
if earliest_ioc:
|
||||||
|
# Print text before IOC
|
||||||
|
if earliest_pos > 0:
|
||||||
|
win.addstr(y_pos, x_pos, remaining[:earliest_pos])
|
||||||
|
x_pos += earliest_pos
|
||||||
|
# Print IOC in color
|
||||||
|
win.addstr(y_pos, x_pos, earliest_ioc, curses.color_pair(4))
|
||||||
|
x_pos += len(earliest_ioc)
|
||||||
|
# Update remaining text
|
||||||
|
remaining = remaining[earliest_pos + len(earliest_ioc):]
|
||||||
|
# Remove found IOC from list
|
||||||
|
iocs_found = [(ioc, t) for ioc, t in iocs_found if ioc != earliest_ioc]
|
||||||
|
else:
|
||||||
|
break
|
||||||
|
# Print any remaining text
|
||||||
|
if remaining:
|
||||||
|
win.addstr(y_pos, x_pos, remaining)
|
||||||
|
else:
|
||||||
|
# No IOCs, display normally
|
||||||
|
win.addstr(y_pos, 2, display_line)
|
||||||
|
except curses.error:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Show scroll indicator
|
||||||
|
if total_lines > max_display_lines:
|
||||||
|
scroll_info = f"[{scroll_offset + 1}-{min(scroll_offset + max_display_lines, total_lines)}/{total_lines}]"
|
||||||
|
try:
|
||||||
|
win.addstr(2, w - len(scroll_info) - 3, scroll_info, curses.A_DIM)
|
||||||
|
except curses.error:
|
||||||
|
pass
|
||||||
|
|
||||||
|
win.addstr(h-2, 2, "[↑↓] Scroll [n] Add Note [b/q/Esc] Close", curses.color_pair(3))
|
||||||
win.refresh()
|
win.refresh()
|
||||||
key = win.getch()
|
key = win.getch()
|
||||||
del win
|
del win
|
||||||
|
|
||||||
# Handle key presses
|
# Handle key presses
|
||||||
if key == ord('n') or key == ord('N'):
|
if key == curses.KEY_UP:
|
||||||
|
scroll_offset = max(0, scroll_offset - 1)
|
||||||
|
elif key == curses.KEY_DOWN:
|
||||||
|
scroll_offset = min(max_scroll, scroll_offset + 1)
|
||||||
|
elif key == curses.KEY_PPAGE: # Page Up
|
||||||
|
scroll_offset = max(0, scroll_offset - max_display_lines)
|
||||||
|
elif key == curses.KEY_NPAGE: # Page Down
|
||||||
|
scroll_offset = min(max_scroll, scroll_offset + max_display_lines)
|
||||||
|
elif key == curses.KEY_HOME:
|
||||||
|
scroll_offset = 0
|
||||||
|
elif key == curses.KEY_END:
|
||||||
|
scroll_offset = max_scroll
|
||||||
|
elif key == ord('n') or key == ord('N'):
|
||||||
# Save current view and switch to evidence_detail temporarily for context
|
# Save current view and switch to evidence_detail temporarily for context
|
||||||
saved_view = self.current_view
|
saved_view = self.current_view
|
||||||
self.current_view = "evidence_detail"
|
self.current_view = "evidence_detail"
|
||||||
self.dialog_add_note()
|
self.dialog_add_note()
|
||||||
self.current_view = saved_view
|
self.current_view = saved_view
|
||||||
# Continue loop to refresh with new note
|
scroll_offset = max_scroll # Jump to bottom to show new note
|
||||||
elif key == ord('b') or key == ord('B') or key == ord('q') or key == ord('Q') or key == 27: # 27 is Esc
|
elif key == ord('b') or key == ord('B') or key == ord('q') or key == ord('Q') or key == 27: # 27 is Esc
|
||||||
break
|
break
|
||||||
else:
|
|
||||||
# Any other key also closes (backwards compatibility)
|
|
||||||
break
|
|
||||||
|
|
||||||
def export_iocs(self):
|
def export_iocs(self):
|
||||||
"""Export IOCs from current context to a text file"""
|
"""Export IOCs from current context to a text file"""
|
||||||
|
|||||||
Reference in New Issue
Block a user