export class WidgetsQuery extends QueryEntity<WidgetsState, Widget> {
  ui: EntityUIQuery<WidgetsUIState, WidgetUI>;

  constructor( protected store: WidgetsStore ) {
    super(store);
    this.createUIQuery();
  }

  selectEntityWithUI(id: ID): Observable<(Widget & WidgetUI)[]> {
    const widgets = this.selectAll();
    const widgetsUI = this.selectAll({ asObject: true });

    return combineLatest(
      widgets,
      widgetsUI
    ).pipe(map(([widgets, widgetsUI]) => {
      return widgets.map(widget => {
        return {
          ...widget,
          ...widgetsUI[widget.id]
        }
      })
    }))
  }
}