Compare commits
4 Commits
05a910ba80
...
943361453a
| Author | SHA1 | Date | |
|---|---|---|---|
| 943361453a | |||
| b5145147d7 | |||
| 91a48a4604 | |||
| 13e37816f8 |
@@ -42,6 +42,9 @@ import json
|
||||
# <field>: <value>
|
||||
# reset_data: # optional — latch reset filter
|
||||
# <field>: <value>
|
||||
# debounce_seconds: <float> # optional — ignore re-triggers within
|
||||
# # this delay (seconds). Useful for
|
||||
# # buttons that fire duplicate events.
|
||||
#
|
||||
# Usage:
|
||||
# handler = EventHandler(ad_api, self.args['events_to_listen'], my_callback)
|
||||
@@ -57,7 +60,7 @@ import json
|
||||
# =============================================================================
|
||||
|
||||
class EventDispatcher:
|
||||
def __init__(self,ad_api,event_name,callback,event_data,reset_data,event_context):
|
||||
def __init__(self,ad_api,event_name,callback,event_data,reset_data,event_context,debounce_seconds=None):
|
||||
self.ad_api = ad_api
|
||||
event_data = self._resolve_zha_device_name(event_data)
|
||||
reset_data = self._resolve_zha_device_name(reset_data)
|
||||
@@ -67,6 +70,8 @@ class EventDispatcher:
|
||||
self.reset_data = reset_data
|
||||
self.waiting_for_reset = False
|
||||
self.event_context = event_context
|
||||
self.debounce_seconds = debounce_seconds
|
||||
self._last_trigger_time = None
|
||||
if event_data == None:
|
||||
self.ad_api.listen_event(self.on_event,event_name)
|
||||
else:
|
||||
@@ -159,6 +164,13 @@ class EventDispatcher:
|
||||
|
||||
#special threatment for event
|
||||
if match_event_data and not self.waiting_for_reset:
|
||||
if self.debounce_seconds is not None:
|
||||
import time
|
||||
now = time.monotonic()
|
||||
if self._last_trigger_time is not None and (now - self._last_trigger_time) < self.debounce_seconds:
|
||||
return False
|
||||
self._last_trigger_time = now
|
||||
|
||||
if self.reset_data != None:
|
||||
self.waiting_for_reset = True
|
||||
|
||||
@@ -173,7 +185,7 @@ class EventDispatcher:
|
||||
class EventHandler:
|
||||
def __init__(self, ad_api,events_block,callback,event_context = None): #event_context is passed back to the CB has a context
|
||||
def register_event_with_params(event_block,callback,event_context):
|
||||
self.add_dispatcher(event_block['event_name'],callback,event_block['event_data'] if 'event_data' in event_block else None, event_block['reset_data'] if 'reset_data' in event_block else None,event_context)
|
||||
self.add_dispatcher(event_block['event_name'],callback,event_block['event_data'] if 'event_data' in event_block else None, event_block['reset_data'] if 'reset_data' in event_block else None,event_context, event_block.get('debounce_seconds'))
|
||||
|
||||
self.__ad_api = ad_api
|
||||
self.event_dispatchers = []
|
||||
@@ -185,7 +197,7 @@ class EventHandler:
|
||||
for event_block in events_block.values():
|
||||
register_event_with_params(event_block,callback,event_context)
|
||||
|
||||
def add_dispatcher(self,event_name,callback,event_data = None,reset_data = None ,event_context = None):
|
||||
def add_dispatcher(self,event_name,callback,event_data = None,reset_data = None ,event_context = None, debounce_seconds = None):
|
||||
self.__ad_api.log_info(f'Registering dispatcher {callback.__name__} for event "{event_name}" ({event_data})')
|
||||
dispatcher = EventDispatcher(self.__ad_api,event_name,callback,event_data,reset_data,event_context)
|
||||
dispatcher = EventDispatcher(self.__ad_api,event_name,callback,event_data,reset_data,event_context,debounce_seconds)
|
||||
self.event_dispatchers.append(dispatcher)
|
||||
@@ -25,7 +25,8 @@ class VirtualSensorBase:
|
||||
self.virtual_sensor_name = sensor_name.split('.')[1]
|
||||
self.sensor_domain = self.sensor_name.split(".")[0]
|
||||
|
||||
assert yaml_block ,f"{self.sensor_name} have a null Yaml block"
|
||||
if not yaml_block:
|
||||
self.log_error(f"{self.sensor_name} have a null Yaml block")
|
||||
self.yaml_block = yaml_block
|
||||
self.templates_library = templates_library
|
||||
|
||||
|
||||
@@ -96,10 +96,20 @@
|
||||
}
|
||||
],
|
||||
"keybindings": [
|
||||
{
|
||||
"command": "appdaemon.goToVirtualSensor",
|
||||
"key": "ctrl+shift+v",
|
||||
"mac": "cmd+shift+v"
|
||||
},
|
||||
{
|
||||
"command": "appdaemon.goToApp",
|
||||
"key": "ctrl+shift+a",
|
||||
"mac": "cmd+shift+a"
|
||||
},
|
||||
{
|
||||
"command": "appdaemon.goToVirtualSensor",
|
||||
"key": "ctrl+shift+v",
|
||||
"mac": "cmd+shift+v"
|
||||
}
|
||||
]
|
||||
},
|
||||
|
||||
@@ -75,6 +75,122 @@ async function findAllApps(): Promise<Array<{
|
||||
return results;
|
||||
}
|
||||
|
||||
// ── Virtual Sensor parsing ─────────────────────────────────────────────────
|
||||
|
||||
type VirtualSensorEntry = {
|
||||
sensorId: string;
|
||||
line: number;
|
||||
appName: string;
|
||||
uri: vscode.Uri;
|
||||
fileName: string;
|
||||
};
|
||||
|
||||
type VSSectionType = 'sensors' | 'continuous_conditions' | 'averagers' | 'binary_sensors' | 'value_selectors' | 'retain_conditions' | 'default_values' | '';
|
||||
|
||||
function mapVirtualSensorId(key: string, section: VSSectionType): string | null {
|
||||
if (section === 'default_values') {
|
||||
return /^(binary_sensor|sensor)\.[a-z0-9_]+$/.test(key) ? key : null;
|
||||
}
|
||||
if (section === 'sensors') {
|
||||
const dot = key.indexOf('.');
|
||||
if (dot < 0) { return null; }
|
||||
const prefix = key.slice(0, dot);
|
||||
const name = key.slice(dot + 1);
|
||||
switch (prefix) {
|
||||
case 'sensor': case 'value_selector': case 'averager': return `sensor.${name}`;
|
||||
case 'binary_sensor': case 'continuous_condition': case 'retain_condition': return `binary_sensor.${name}`;
|
||||
default: return null;
|
||||
}
|
||||
}
|
||||
if (section === 'continuous_conditions' || section === 'retain_conditions' || section === 'binary_sensors') {
|
||||
return key.includes('.') ? null : `binary_sensor.${key}`;
|
||||
}
|
||||
if (section === 'averagers' || section === 'value_selectors') {
|
||||
return key.includes('.') ? null : `sensor.${key}`;
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
function parseVSSectionName(raw: string, indent: number): VSSectionType {
|
||||
const re = new RegExp(`^\\s{${indent}}(sensors|continuous_conditions|averagers|binary_sensors|value_selectors|retain_conditions|default_values):\\s*$`);
|
||||
const m = raw.match(re);
|
||||
return m ? m[1] as VSSectionType : '';
|
||||
}
|
||||
|
||||
function parseVirtualSensorsFromText(text: string): Array<{ sensorId: string; line: number; appName: string }> {
|
||||
const lines = text.split('\n');
|
||||
const entries: Array<{ sensorId: string; line: number; appName: string }> = [];
|
||||
|
||||
let currentApp = '';
|
||||
let isVSApp = false; // module: virtualsensors
|
||||
let vsSection: VSSectionType = ''; // active sub-section in VirtualSensorsApp at indent=2
|
||||
let inVSKey = false; // in virtual_sensors: sub-block of another app
|
||||
let nestedSection: VSSectionType = ''; // active sub-section in virtual_sensors: at indent=4
|
||||
|
||||
for (let i = 0; i < lines.length; i++) {
|
||||
const raw = lines[i];
|
||||
const trimmed = raw.trimStart();
|
||||
if (!trimmed || trimmed.startsWith('#')) { continue; }
|
||||
const indent = raw.length - trimmed.length;
|
||||
|
||||
if (indent === 0) {
|
||||
const m = raw.match(/^([a-z][a-z0-9_]*):\s*$/);
|
||||
currentApp = m ? m[1] : '';
|
||||
isVSApp = false; vsSection = ''; inVSKey = false; nestedSection = '';
|
||||
continue;
|
||||
}
|
||||
|
||||
if (indent === 2) {
|
||||
vsSection = ''; inVSKey = false; nestedSection = '';
|
||||
if (/^\s+module:\s*virtualsensors/.test(raw)) {
|
||||
isVSApp = true;
|
||||
} else if (isVSApp) {
|
||||
vsSection = parseVSSectionName(raw, 2);
|
||||
} else {
|
||||
if (/^\s+virtual_sensors:\s*$/.test(raw)) { inVSKey = true; }
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if (indent === 4) {
|
||||
nestedSection = '';
|
||||
if (isVSApp && vsSection) {
|
||||
const km = raw.match(/^\s+([a-z][a-z0-9_.]+):/);
|
||||
if (km) {
|
||||
const sensorId = mapVirtualSensorId(km[1], vsSection);
|
||||
if (sensorId) { entries.push({ sensorId, line: i, appName: currentApp }); }
|
||||
}
|
||||
} else if (inVSKey) {
|
||||
nestedSection = parseVSSectionName(raw, 4);
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if (indent === 6 && inVSKey && nestedSection) {
|
||||
const km = raw.match(/^\s+([a-z][a-z0-9_.]+):/);
|
||||
if (km) {
|
||||
const sensorId = mapVirtualSensorId(km[1], nestedSection);
|
||||
if (sensorId) { entries.push({ sensorId, line: i, appName: currentApp }); }
|
||||
}
|
||||
}
|
||||
}
|
||||
return entries;
|
||||
}
|
||||
|
||||
async function findAllVirtualSensors(): Promise<VirtualSensorEntry[]> {
|
||||
const yamlFiles = await vscode.workspace.findFiles('**/*.yaml', '**/node_modules/**');
|
||||
const results: VirtualSensorEntry[] = [];
|
||||
for (const uri of yamlFiles) {
|
||||
let doc: vscode.TextDocument;
|
||||
try { doc = await vscode.workspace.openTextDocument(uri); } catch { continue; }
|
||||
const fileName = vscode.workspace.asRelativePath(uri);
|
||||
for (const e of parseVirtualSensorsFromText(doc.getText())) {
|
||||
results.push({ ...e, uri, fileName });
|
||||
}
|
||||
}
|
||||
return results;
|
||||
}
|
||||
|
||||
// ── Symbol Providers ─────────────────────────────────────────────────────────
|
||||
|
||||
class ADDocumentSymbolProvider implements vscode.DocumentSymbolProvider {
|
||||
@@ -97,9 +213,9 @@ class ADDocumentSymbolProvider implements vscode.DocumentSymbolProvider {
|
||||
|
||||
class ADWorkspaceSymbolProvider implements vscode.WorkspaceSymbolProvider {
|
||||
async provideWorkspaceSymbols(query: string): Promise<vscode.SymbolInformation[]> {
|
||||
const apps = await findAllApps();
|
||||
const lq = query.toLowerCase();
|
||||
return apps
|
||||
const [apps, sensors] = await Promise.all([findAllApps(), findAllVirtualSensors()]);
|
||||
const appSymbols = apps
|
||||
.filter(a => !lq || a.appName.toLowerCase().includes(lq))
|
||||
.map(a => new vscode.SymbolInformation(
|
||||
a.appName,
|
||||
@@ -107,6 +223,102 @@ class ADWorkspaceSymbolProvider implements vscode.WorkspaceSymbolProvider {
|
||||
`module: ${a.moduleName}`,
|
||||
new vscode.Location(a.uri, new vscode.Position(a.line, 0))
|
||||
));
|
||||
const sensorSymbols = sensors
|
||||
.filter(s => !lq || s.sensorId.toLowerCase().includes(lq))
|
||||
.map(s => new vscode.SymbolInformation(
|
||||
s.sensorId,
|
||||
vscode.SymbolKind.Variable,
|
||||
s.appName,
|
||||
new vscode.Location(s.uri, new vscode.Position(s.line, 0))
|
||||
));
|
||||
return [...appSymbols, ...sensorSymbols];
|
||||
}
|
||||
}
|
||||
|
||||
class ADVirtualSensorDefinitionProvider implements vscode.DefinitionProvider {
|
||||
// Simple cache: reset on save
|
||||
private cache: VirtualSensorEntry[] | undefined;
|
||||
private cacheExpiry = 0;
|
||||
|
||||
constructor(private haClient: HAClient) {}
|
||||
|
||||
invalidate() { this.cache = undefined; }
|
||||
|
||||
async getSensors(): Promise<VirtualSensorEntry[]> {
|
||||
const now = Date.now();
|
||||
if (this.cache && now < this.cacheExpiry) { return this.cache; }
|
||||
this.cache = await findAllVirtualSensors();
|
||||
this.cacheExpiry = now + 5000; // 5 s TTL
|
||||
return this.cache;
|
||||
}
|
||||
|
||||
async provideDefinition(
|
||||
document: vscode.TextDocument,
|
||||
position: vscode.Position
|
||||
): Promise<vscode.Definition | undefined> {
|
||||
// Match any entity ID pattern (same as hover provider)
|
||||
const wordRange = document.getWordRangeAtPosition(
|
||||
position,
|
||||
/[a-z][a-z_]*\.[a-z0-9][a-z0-9_]*/
|
||||
);
|
||||
if (!wordRange) { return undefined; }
|
||||
const entityId = document.getText(wordRange);
|
||||
|
||||
// 1. Check virtual sensor index (parsed from YAML)
|
||||
const sensors = await this.getSensors();
|
||||
const vsMatch = sensors.find(s => s.sensorId === entityId);
|
||||
if (vsMatch) {
|
||||
return new vscode.Location(vsMatch.uri, new vscode.Position(vsMatch.line, 0));
|
||||
}
|
||||
|
||||
// 2. Fallback: check HA entity's ad_app attribute → navigate to that app
|
||||
const entity = this.haClient.getEntity(entityId);
|
||||
const adApp = entity?.attributes?.ad_app as string | undefined;
|
||||
if (!adApp) { return undefined; }
|
||||
const apps = await findAllApps();
|
||||
const appMatch = apps.find(a => a.appName === adApp);
|
||||
if (!appMatch) { return undefined; }
|
||||
return new vscode.Location(appMatch.uri, new vscode.Position(appMatch.line, 0));
|
||||
}
|
||||
}
|
||||
|
||||
class ADVirtualSensorHoverProvider implements vscode.HoverProvider {
|
||||
constructor(
|
||||
private haClient: HAClient,
|
||||
private defProvider: ADVirtualSensorDefinitionProvider
|
||||
) {}
|
||||
|
||||
async provideHover(
|
||||
document: vscode.TextDocument,
|
||||
position: vscode.Position
|
||||
): Promise<vscode.Hover | undefined> {
|
||||
// Find a virtual sensor whose definition is on this exact line
|
||||
const sensors = await this.defProvider.getSensors();
|
||||
const match = sensors.find(
|
||||
s => s.uri.fsPath === document.uri.fsPath && s.line === position.line
|
||||
);
|
||||
if (!match) { return undefined; }
|
||||
|
||||
const entity = this.haClient.getEntity(match.sensorId);
|
||||
if (!entity) { return undefined; }
|
||||
|
||||
const md = new vscode.MarkdownString();
|
||||
md.isTrusted = true;
|
||||
const friendly = entity.attributes?.friendly_name as string | undefined;
|
||||
md.appendMarkdown(`### ${friendly || entity.entity_id}\n\n`);
|
||||
md.appendMarkdown(`| | |\n|---|---|\n`);
|
||||
md.appendMarkdown(`| **Entity** | \`${entity.entity_id}\` |\n`);
|
||||
md.appendMarkdown(`| **State** | **\`${entity.state}\`** |\n`);
|
||||
if (entity.attributes) {
|
||||
const skip = new Set(['friendly_name', 'icon', 'entity_picture', 'supported_features', 'supported_color_modes']);
|
||||
for (const [key, value] of Object.entries(entity.attributes).filter(([k]) => !skip.has(k)).slice(0, 12)) {
|
||||
const display = typeof value === 'object' ? JSON.stringify(value) : String(value);
|
||||
const truncated = display.length > 60 ? display.substring(0, 57) + '...' : display;
|
||||
md.appendMarkdown(`| ${key.replace(/\|/g, '\\|')} | \`${truncated.replace(/\|/g, '\\|')}\` |\n`);
|
||||
}
|
||||
}
|
||||
md.appendMarkdown(`\n*Last changed: ${entity.last_changed}*`);
|
||||
return new vscode.Hover(md);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -166,6 +378,15 @@ export async function activate(context: vscode.ExtensionContext) {
|
||||
{ scheme: 'file', language: 'python' }
|
||||
];
|
||||
|
||||
const virtualSensorDefProvider = new ADVirtualSensorDefinitionProvider(haClient);
|
||||
|
||||
// Invalidate definition cache on save so edits are picked up immediately
|
||||
context.subscriptions.push(
|
||||
vscode.workspace.onDidSaveTextDocument(doc => {
|
||||
if (doc.languageId === 'yaml') { virtualSensorDefProvider.invalidate(); }
|
||||
})
|
||||
);
|
||||
|
||||
context.subscriptions.push(
|
||||
vscode.languages.registerCompletionItemProvider(
|
||||
selector,
|
||||
@@ -182,12 +403,46 @@ export async function activate(context: vscode.ExtensionContext) {
|
||||
),
|
||||
vscode.languages.registerWorkspaceSymbolProvider(
|
||||
new ADWorkspaceSymbolProvider()
|
||||
),
|
||||
vscode.languages.registerDefinitionProvider(
|
||||
{ scheme: 'file', language: 'yaml' },
|
||||
virtualSensorDefProvider
|
||||
),
|
||||
vscode.languages.registerHoverProvider(
|
||||
{ scheme: 'file', language: 'yaml' },
|
||||
new ADVirtualSensorHoverProvider(haClient, virtualSensorDefProvider)
|
||||
)
|
||||
);
|
||||
|
||||
// ── Commands ─────────────────────────────────────────────────────────────
|
||||
|
||||
context.subscriptions.push(
|
||||
vscode.commands.registerCommand('appdaemon.goToVirtualSensor', async () => {
|
||||
const sensors = await findAllVirtualSensors();
|
||||
if (sensors.length === 0) {
|
||||
vscode.window.showWarningMessage('AppDaemon: No virtual sensors found in workspace');
|
||||
return;
|
||||
}
|
||||
type SensorItem = vscode.QuickPickItem & { sensor: VirtualSensorEntry };
|
||||
const items: SensorItem[] = sensors.map(s => ({
|
||||
label: `$(circuit-board) ${s.sensorId}`,
|
||||
description: s.fileName,
|
||||
detail: `in ${s.appName}`,
|
||||
sensor: s
|
||||
}));
|
||||
const picked = await vscode.window.showQuickPick(items, {
|
||||
placeHolder: 'Go to virtual sensor…',
|
||||
matchOnDescription: true,
|
||||
matchOnDetail: true
|
||||
});
|
||||
if (!picked) { return; }
|
||||
const doc = await vscode.workspace.openTextDocument(picked.sensor.uri);
|
||||
const editor = await vscode.window.showTextDocument(doc);
|
||||
const pos = new vscode.Position(picked.sensor.line, 0);
|
||||
editor.selection = new vscode.Selection(pos, pos);
|
||||
editor.revealRange(new vscode.Range(pos, pos), vscode.TextEditorRevealType.InCenter);
|
||||
}),
|
||||
|
||||
vscode.commands.registerCommand('appdaemon.goToApp', async () => {
|
||||
const apps = await findAllApps();
|
||||
if (apps.length === 0) {
|
||||
|
||||
Binary file not shown.
Reference in New Issue
Block a user