Refactor Frigate specific details out of ViewMediaBase.

This commit is contained in:
Dermot Duffy
2023-01-24 19:36:54 -08:00
parent 36f14e153c
commit f073f06789
9 changed files with 243 additions and 234 deletions
+9 -7
View File
@@ -6,8 +6,7 @@ import { CAMERA_BIRDSEYE } from '../../const';
import { CameraConfig, RecordingSegment } from '../../types'; import { CameraConfig, RecordingSegment } from '../../types';
import { MediaQueriesResults } from '../../view/media-queries-results'; import { MediaQueriesResults } from '../../view/media-queries-results';
import { MediaQueriesClassifier } from '../../view/media-queries-classifier'; import { MediaQueriesClassifier } from '../../view/media-queries-classifier';
import { ViewMedia, ViewMediaFactory } from '../../view/media'; import { ViewMedia } from '../../view/media';
import { ViewMediaClassifier } from '../../view/media-classifier';
import { RecordingSegmentsCache } from '../cache'; import { RecordingSegmentsCache } from '../cache';
import { import {
CameraManagerEngine, CameraManagerEngine,
@@ -45,6 +44,9 @@ import throttle from 'lodash-es/throttle';
import { runWhenIdleIfSupported } from '../../utils/basic'; import { runWhenIdleIfSupported } from '../../utils/basic';
import { fromUnixTime } from 'date-fns'; import { fromUnixTime } from 'date-fns';
import { sum } from 'lodash-es'; import { sum } from 'lodash-es';
import { FrigateViewMediaClassifier } from './media-classifier';
import { ViewMediaClassifier } from '../../view/media-classifier';
import { FrigateViewMediaFactory } from './media';
const EVENT_REQUEST_CACHE_MAX_AGE_SECONDS = 60; const EVENT_REQUEST_CACHE_MAX_AGE_SECONDS = 60;
const RECORDING_SUMMARY_REQUEST_CACHE_MAX_AGE_SECONDS = 60; const RECORDING_SUMMARY_REQUEST_CACHE_MAX_AGE_SECONDS = 60;
@@ -93,13 +95,13 @@ export class FrigateCameraManagerEngine implements CameraManagerEngine {
media: ViewMedia, media: ViewMedia,
): string | null { ): string | null {
let path: string | null = null; let path: string | null = null;
if (ViewMediaClassifier.isFrigateEvent(media)) { if (FrigateViewMediaClassifier.isFrigateEvent(media)) {
path = path =
`/api/frigate/${cameraConfig.frigate.client_id}` + `/api/frigate/${cameraConfig.frigate.client_id}` +
`/notifications/${media.getID()}/` + `/notifications/${media.getID()}/` +
`${ViewMediaClassifier.isClip(media) ? 'clip.mp4' : 'snapshot.jpg'}` + `${ViewMediaClassifier.isClip(media) ? 'clip.mp4' : 'snapshot.jpg'}` +
`?download=true`; `?download=true`;
} else if (ViewMediaClassifier.isFrigateRecording(media)) { } else if (FrigateViewMediaClassifier.isFrigateRecording(media)) {
path = path =
`/api/frigate/${cameraConfig.frigate.client_id}` + `/api/frigate/${cameraConfig.frigate.client_id}` +
`/recording/${cameraConfig.frigate.camera_name}` + `/recording/${cameraConfig.frigate.camera_name}` +
@@ -160,7 +162,7 @@ export class FrigateCameraManagerEngine implements CameraManagerEngine {
favorite: boolean, favorite: boolean,
): Promise<void> { ): Promise<void> {
const clientID = cameraConfig.frigate.client_id; const clientID = cameraConfig.frigate.client_id;
if (!ViewMediaClassifier.isFrigateEvent(media)) { if (!FrigateViewMediaClassifier.isFrigateEvent(media)) {
return; return;
} }
@@ -331,7 +333,7 @@ export class FrigateCameraManagerEngine implements CameraManagerEngine {
if (!mediaType) { if (!mediaType) {
continue; continue;
} }
const media = ViewMediaFactory.createViewMediaFromFrigateEvent( const media = FrigateViewMediaFactory.createEventViewMedia(
mediaType, mediaType,
query.cameraID, query.cameraID,
event, event,
@@ -353,7 +355,7 @@ export class FrigateCameraManagerEngine implements CameraManagerEngine {
const output: ViewMedia[] = []; const output: ViewMedia[] = [];
for (const recording of results.recordings) { for (const recording of results.recordings) {
const media = ViewMediaFactory.createViewMediaFromFrigateRecording( const media = FrigateViewMediaFactory.createRecordingViewMedia(
query.cameraID, query.cameraID,
recording, recording,
); );
+18
View File
@@ -0,0 +1,18 @@
import { ViewMedia } from '../../view/media';
import { FrigateEventViewMedia, FrigateRecordingViewMedia } from './media';
export class FrigateViewMediaClassifier {
public static isFrigateMedia(
media: ViewMedia,
): media is FrigateEventViewMedia | FrigateRecordingViewMedia {
return this.isFrigateEvent(media) || this.isFrigateRecording(media);
}
public static isFrigateEvent(media: ViewMedia): media is FrigateEventViewMedia {
return media instanceof FrigateEventViewMedia;
}
public static isFrigateRecording(
media: ViewMedia,
): media is FrigateRecordingViewMedia {
return media instanceof FrigateRecordingViewMedia;
}
}
+164
View File
@@ -0,0 +1,164 @@
import fromUnixTime from 'date-fns/fromUnixTime';
import isEqual from 'lodash-es/isEqual';
import { CameraConfig } from '../../types';
import { ViewMediaBase, EventViewMedia, RecordingViewMedia } from '../../view/media';
import { FrigateEvent, FrigateRecording } from './types';
import {
getEventMediaContentID,
getEventThumbnailURL,
getEventTitle,
getRecordingMediaContentID,
getRecordingTitle,
} from './util';
export class FrigateEventViewMedia
extends ViewMediaBase<FrigateEvent>
implements EventViewMedia<FrigateEvent>
{
public hasClip(): boolean {
return !!this._source.has_clip;
}
public getStartTime(): Date {
return fromUnixTime(this._source.start_time);
}
public getEndTime(): Date | null {
return this._source.end_time ? fromUnixTime(this._source.end_time) : null;
}
// eslint-disable-next-line @typescript-eslint/no-unused-vars
public getID(_cameraConfig?: CameraConfig): string {
return this._source.id;
}
public getContentID(cameraConfig?: CameraConfig): string | null {
if (
!cameraConfig ||
!cameraConfig.frigate.client_id ||
!cameraConfig.frigate.camera_name
) {
return null;
}
return getEventMediaContentID(
cameraConfig.frigate.client_id,
cameraConfig.frigate.camera_name,
this._source,
this._mediaType === 'clip' ? 'clips' : 'snapshots',
);
}
// eslint-disable-next-line @typescript-eslint/no-unused-vars
public getTitle(_cameraConfig?: CameraConfig): string | null {
return getEventTitle(this._source);
}
public getThumbnail(cameraConfig?: CameraConfig): string | null {
if (cameraConfig?.frigate.client_id) {
return getEventThumbnailURL(cameraConfig.frigate.client_id, this._source);
}
return null;
}
public isFavorite(): boolean | null {
return this._source.retain_indefinitely ?? null;
}
public setFavorite(favorite: boolean): void {
this._source.retain_indefinitely = favorite;
}
public getWhat(): string[] | null {
return [this._source.label];
}
public getWhere(): string[] | null {
const zones = this._source.zones;
return zones.length ? zones : null;
}
public getScore(): number | null {
return this._source.top_score;
}
// eslint-disable-next-line @typescript-eslint/no-unused-vars
public isGroupableWith(that: EventViewMedia<unknown>): boolean {
return (
this.getMediaType() === that.getMediaType() &&
isEqual(this.getWhere(), that.getWhere()) &&
isEqual(this.getWhat(), that.getWhat())
);
}
public getClipEquivalent(): EventViewMedia<FrigateEvent> | null {
if (!this.hasClip()) {
return null;
}
return FrigateViewMediaFactory.createEventViewMedia(
'clip',
this._cameraID,
this._source,
);
}
}
export class FrigateRecordingViewMedia
extends ViewMediaBase<FrigateRecording>
implements RecordingViewMedia<FrigateRecording>
{
public getID(cameraConfig?: CameraConfig): string | null {
// ID name is derived from the real camera name (not CameraID) since the
// recordings for the same camera across multiple zones will be the same and
// can be dedup'd from this id.
if (cameraConfig) {
return `${cameraConfig.frigate?.client_id ?? ''}/${
cameraConfig.frigate.camera_name ?? ''
}/${this._source.startTime.getTime()}/${this._source.endTime.getTime()}}`;
}
return null;
}
public getStartTime(): Date {
return this._source.startTime;
}
public getEndTime(): Date {
return this._source.endTime;
}
public getContentID(cameraConfig?: CameraConfig): string | null {
if (
!cameraConfig ||
!cameraConfig.frigate.client_id ||
!cameraConfig.frigate.camera_name
) {
return null;
}
return getRecordingMediaContentID(
cameraConfig.frigate.client_id,
cameraConfig.frigate.camera_name,
this._source,
);
}
// eslint-disable-next-line @typescript-eslint/no-unused-vars
public getTitle(cameraConfig?: CameraConfig): string | null {
if (!cameraConfig) {
return null;
}
return getRecordingTitle(cameraConfig, this._source);
}
public getEventCount(): number {
return this._source.events;
}
}
export class FrigateViewMediaFactory {
static createEventViewMedia(
type: 'clip' | 'snapshot',
cameraID: string,
event: FrigateEvent,
): FrigateEventViewMedia | null {
if (
(type === 'clip' && event.has_clip) ||
(type === 'snapshot' && event.has_snapshot)
) {
return new FrigateEventViewMedia(type, cameraID, event);
}
return null;
}
static createRecordingViewMedia(
cameraID: string,
recording: FrigateRecording,
): FrigateRecordingViewMedia | null {
return new FrigateRecordingViewMedia('recording', cameraID, recording);
}
}
+12 -9
View File
@@ -1,6 +1,5 @@
import utcToZonedTime from 'date-fns-tz/utcToZonedTime'; import utcToZonedTime from 'date-fns-tz/utcToZonedTime';
import fromUnixTime from 'date-fns/fromUnixTime'; import { CameraConfig, ClipsOrSnapshots } from '../../types';
import { ClipsOrSnapshots } from '../../types';
import { formatDateAndTime, prettifyTitle } from '../../utils/basic'; import { formatDateAndTime, prettifyTitle } from '../../utils/basic';
import { FrigateEvent, FrigateRecording } from './types'; import { FrigateEvent, FrigateRecording } from './types';
@@ -22,9 +21,12 @@ export const getEventTitle = (event: FrigateEvent): string => {
)}%]`; )}%]`;
}; };
export const getRecordingTitle = (recording: FrigateRecording): string => { export const getRecordingTitle = (
return `${prettifyTitle(recording.camera)} ${formatDateAndTime( cameraConfig: CameraConfig,
fromUnixTime(recording.start_time), recording: FrigateRecording,
): string => {
return `${prettifyTitle(cameraConfig.frigate.camera_name)} ${formatDateAndTime(
recording.startTime,
)}`; )}`;
}; };
@@ -67,15 +69,16 @@ export const getRecordingMediaContentID = (
cameraName: string, cameraName: string,
recording: FrigateRecording, recording: FrigateRecording,
): string => { ): string => {
const date = fromUnixTime(recording.start_time);
return [ return [
'media-source://frigate', 'media-source://frigate',
clientId, clientId,
'recordings', 'recordings',
cameraName, cameraName,
`${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, '0')}-${String( `${recording.startTime.getFullYear()}-${String(
String(date.getDate()).padStart(2, '0'), recording.startTime.getMonth() + 1,
).padStart(2, '0')}-${String(
String(recording.startTime.getDate()).padStart(2, '0'),
)}`, )}`,
String(date.getHours()).padStart(2, '0'), String(recording.startTime.getHours()).padStart(2, '0'),
].join('/'); ].join('/');
}; };
+2 -2
View File
@@ -128,7 +128,7 @@ export class FrigateCardThumbnailFeatureRecording extends LitElement {
@customElement('frigate-card-thumbnail-details-event') @customElement('frigate-card-thumbnail-details-event')
export class FrigateCardThumbnailDetailsEvent extends LitElement { export class FrigateCardThumbnailDetailsEvent extends LitElement {
@property({ attribute: false }) @property({ attribute: false })
public media?: EventViewMedia; public media?: EventViewMedia<unknown>;
@property({ attribute: false }) @property({ attribute: false })
public mediaSeek?: MediaSeek; public mediaSeek?: MediaSeek;
@@ -180,7 +180,7 @@ export class FrigateCardThumbnailDetailsEvent extends LitElement {
@customElement('frigate-card-thumbnail-details-recording') @customElement('frigate-card-thumbnail-details-recording')
export class FrigateCardThumbnailDetailsRecording extends LitElement { export class FrigateCardThumbnailDetailsRecording extends LitElement {
@property({ attribute: false }) @property({ attribute: false })
public media?: RecordingViewMedia; public media?: RecordingViewMedia<unknown>;
@property({ attribute: false }) @property({ attribute: false })
public mediaSeek?: MediaSeek; public mediaSeek?: MediaSeek;
+5 -2
View File
@@ -374,8 +374,11 @@ export class FrigateCardViewerCarousel extends LitElement {
!media || !media ||
// If this specific media item has no clip, then do nothing (even if all // If this specific media item has no clip, then do nothing (even if all
// the other media items do). // the other media items do).
!ViewMediaClassifier.isFrigateEvent(media) || !ViewMediaClassifier.isEvent(media) ||
!media.hasClip() || // If the event certainly has no clip, don't bother going further. If
// we're not sure for this camera type (i.e. hasClip() === null) the query
// will proceed anyway.
media.hasClip() === false ||
!MediaQueriesClassifier.areEventQueries(this.view.query) !MediaQueriesClassifier.areEventQueries(this.view.query)
) { ) {
return; return;
+23 -21
View File
@@ -179,7 +179,8 @@ export const mergeFrigateBrowseMediaSources = async (
return createEventParentForChildren( return createEventParentForChildren(
'Merged events', 'Merged events',
children.sort(sortYoungestToOldest), children
// TODO: Restore sort or remove: children.sort(sortYoungestToOldest),
); );
}; };
@@ -350,25 +351,26 @@ export const createChild = (
return result; return result;
}; };
/** // TODO Remove or fix this if not needed post gallery.
* Sort the timeline items most recent to least recent. // /**
* @param a The first item. // * Sort the timeline items most recent to least recent.
* @param b The second item. // * @param a The first item.
* @returns -1, 0, 1 (standard array sort function configuration). // * @param b The second item.
*/ // * @returns -1, 0, 1 (standard array sort function configuration).
export const sortYoungestToOldest = ( // */
a: FrigateBrowseMediaSource, // export const sortYoungestToOldest = (
b: FrigateBrowseMediaSource, // a: FrigateBrowseMediaSource,
): number => { // b: FrigateBrowseMediaSource,
const a_source = a.frigate?.event ?? a.frigate?.recording; // ): number => {
const b_source = b.frigate?.event ?? b.frigate?.recording; // const a_source = a.frigate?.event ?? a.frigate?.recording;
// const b_source = b.frigate?.event ?? b.frigate?.recording;
if (!a_source || (b_source && b_source.start_time > a_source.start_time)) { // if (!a_source || (b_source && b_source.start_time > a_source.start_time)) {
return 1; // return 1;
} // }
if (!b_source || (a_source && b_source.start_time < a_source.start_time)) { // if (!b_source || (a_source && b_source.start_time < a_source.start_time)) {
return -1; // return -1;
} // }
return 0; // return 0;
}; // };
+3 -22
View File
@@ -1,29 +1,10 @@
import { import { ViewMedia, RecordingViewMedia, EventViewMedia } from './media';
ViewMedia,
FrigateEventViewMedia,
FrigateRecordingViewMedia,
RecordingViewMedia,
EventViewMedia,
} from './media';
export class ViewMediaClassifier { export class ViewMediaClassifier {
public static isFrigateMedia( public static isEvent(media: ViewMedia): media is EventViewMedia<unknown> {
media: ViewMedia,
): media is FrigateEventViewMedia | FrigateRecordingViewMedia {
return this.isFrigateEvent(media) || this.isFrigateRecording(media);
}
public static isFrigateEvent(media: ViewMedia): media is FrigateEventViewMedia {
return media instanceof FrigateEventViewMedia;
}
public static isFrigateRecording(
media: ViewMedia,
): media is FrigateRecordingViewMedia {
return media instanceof FrigateRecordingViewMedia;
}
public static isEvent(media: ViewMedia): media is EventViewMedia {
return this.isClip(media) || this.isSnapshot(media); return this.isClip(media) || this.isSnapshot(media);
} }
public static isRecording(media: ViewMedia): media is RecordingViewMedia { public static isRecording(media: ViewMedia): media is RecordingViewMedia<unknown> {
return media.getMediaType() === 'recording'; return media.getMediaType() === 'recording';
} }
public static isClip(media: ViewMedia): boolean { public static isClip(media: ViewMedia): boolean {
+6 -170
View File
@@ -1,19 +1,7 @@
import fromUnixTime from 'date-fns/fromUnixTime';
import isEqual from 'lodash-es/isEqual';
import { CameraConfig } from '../types.js'; import { CameraConfig } from '../types.js';
import {
getEventMediaContentID,
getEventThumbnailURL,
getEventTitle,
getRecordingMediaContentID,
getRecordingTitle,
} from '../camera/frigate/util.js';
import { FrigateEvent, FrigateRecording } from '../camera/frigate/types.js';
export type ViewMediaType = 'clip' | 'snapshot' | 'recording'; export type ViewMediaType = 'clip' | 'snapshot' | 'recording';
export type ViewMediaSourceType = FrigateEvent | FrigateRecording;
class ViewMediaBase<T extends ViewMediaSourceType> { export class ViewMediaBase<T> {
protected _mediaType: ViewMediaType; protected _mediaType: ViewMediaType;
protected _cameraID: string; protected _cameraID: string;
protected _source: T; protected _source: T;
@@ -72,167 +60,15 @@ class ViewMediaBase<T extends ViewMediaSourceType> {
} }
} }
export interface EventViewMedia extends ViewMedia { export interface EventViewMedia<T> extends ViewMediaBase<T> {
getScore(): number | null; getScore(): number | null;
getWhat(): string[] | null; getWhat(): string[] | null;
isGroupableWith(that: EventViewMedia): boolean; isGroupableWith(that: EventViewMedia<T>): boolean;
hasClip(): boolean | null;
} }
export interface RecordingViewMedia extends ViewMedia { export interface RecordingViewMedia<T> extends ViewMediaBase<T> {
getEventCount(): number | null; getEventCount(): number | null;
} }
// Creates a 'public interface only' version of ViewMediaBase for use elsewhere export type ViewMedia = ViewMediaBase<unknown>;
// (typescript struggles with the ViewMediaClassifier classification functions
// used above if the object has data elements).
export type ViewMedia = {
[P in keyof ViewMediaBase<ViewMediaSourceType>]: ViewMediaBase<ViewMediaSourceType>[P];
};
export class FrigateEventViewMedia
extends ViewMediaBase<FrigateEvent>
implements EventViewMedia
{
public hasClip(): boolean {
return !!this._source.has_clip;
}
public getClipEquivalent(): ViewMedia | null {
if (!this.hasClip()) {
return null;
}
return ViewMediaFactory.createViewMediaFromFrigateEvent(
'clip',
this._cameraID,
this._source,
);
}
public getStartTime(): Date {
return fromUnixTime(this._source.start_time);
}
public getEndTime(): Date | null {
return this._source.end_time ? fromUnixTime(this._source.end_time) : null;
}
// eslint-disable-next-line @typescript-eslint/no-unused-vars
public getID(_cameraConfig?: CameraConfig): string {
return this._source.id;
}
public getContentID(cameraConfig?: CameraConfig): string | null {
if (
!cameraConfig ||
!cameraConfig.frigate.client_id ||
!cameraConfig.frigate.camera_name
) {
return null;
}
return getEventMediaContentID(
cameraConfig.frigate.client_id,
cameraConfig.frigate.camera_name,
this._source,
this._mediaType === 'clip' ? 'clips' : 'snapshots',
);
}
// eslint-disable-next-line @typescript-eslint/no-unused-vars
public getTitle(_cameraConfig?: CameraConfig): string | null {
return getEventTitle(this._source);
}
public getThumbnail(cameraConfig?: CameraConfig): string | null {
if (cameraConfig?.frigate.client_id) {
return getEventThumbnailURL(cameraConfig.frigate.client_id, this._source);
}
return null;
}
public isFavorite(): boolean | null {
return this._source.retain_indefinitely ?? null;
}
public setFavorite(favorite: boolean): void {
this._source.retain_indefinitely = favorite;
}
public getWhat(): string[] | null {
return [this._source.label];
}
public getWhere(): string[] | null {
const zones = this._source.zones;
return zones.length ? zones : null;
}
public getScore(): number | null {
return this._source.top_score;
}
// eslint-disable-next-line @typescript-eslint/no-unused-vars
public isGroupableWith(that: EventViewMedia): boolean {
return (
this.getMediaType() === that.getMediaType() &&
isEqual(this.getWhere(), that.getWhere()) &&
isEqual(this.getWhat(), that.getWhat())
);
}
}
export class FrigateRecordingViewMedia
extends ViewMediaBase<FrigateRecording>
implements RecordingViewMedia
{
public getID(cameraConfig?: CameraConfig): string | null {
// ID name is derived from the real camera name (not CameraID) since the
// recordings for the same camera across multiple zones will be the same and
// can be dedup'd from this id.
if (cameraConfig) {
return `${cameraConfig.frigate?.client_id ?? ''}/${
cameraConfig.frigate.camera_name ?? ''
}/${this._source.start_time}/${this._source.end_time}}`;
}
return null;
}
public getStartTime(): Date {
return fromUnixTime(this._source.start_time);
}
public getEndTime(): Date {
return fromUnixTime(this._source.end_time);
}
public getContentID(cameraConfig?: CameraConfig): string | null {
if (
!cameraConfig ||
!cameraConfig.frigate.client_id ||
!cameraConfig.frigate.camera_name
) {
return null;
}
return getRecordingMediaContentID(
cameraConfig.frigate.client_id,
cameraConfig.frigate.camera_name,
this._source,
);
}
// eslint-disable-next-line @typescript-eslint/no-unused-vars
public getTitle(_cameraConfig?: CameraConfig): string | null {
return getRecordingTitle(this._source);
}
public getEventCount(): number {
return this._source.events;
}
}
export class ViewMediaFactory {
static createViewMediaFromFrigateEvent(
type: 'clip' | 'snapshot',
cameraID: string,
event: FrigateEvent,
): ViewMedia | null {
if (
(type === 'clip' && event.has_clip) ||
(type === 'snapshot' && event.has_snapshot)
) {
return new FrigateEventViewMedia(type, cameraID, event);
}
return null;
}
static createViewMediaFromFrigateRecording(
cameraID: string,
recording: FrigateRecording,
): ViewMedia | null {
return new FrigateRecordingViewMedia('recording', cameraID, recording);
}
}