1 files changed, 138 insertions, 0 deletions
diff --git a/gae/frontend/src/app/menu/device/device.component.ts b/gae/frontend/src/app/menu/device/device.component.ts
new file mode 100644
index 0000000..6258aed
--- /dev/null
+++ b/gae/frontend/src/app/menu/device/device.component.ts
@@ -0,0 +1,138 @@
+/**
+ * Copyright (C) 2018 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { Component, OnInit, ViewChild } from '@angular/core';
+import { MatSnackBar, MatTableDataSource, PageEvent } from '@angular/material';
+
+import { AppService } from '../../appservice';
+import { Device } from '../../model/device';
+import { DeviceService } from './device.service';
+import { DeviceStatus, SchedulingStatus } from '../../shared/vtslab_status';
+import { FilterComponent } from '../../shared/filter/filter.component';
+import { FilterItem } from '../../model/filter_item';
+import { MenuBaseClass } from '../menu_base';
+
+
+/** Component that handles device menu. */
+@Component({
+  selector: 'app-device',
+  templateUrl: './device.component.html',
+  providers: [ DeviceService ],
+  styleUrls: ['./device.component.scss'],
+})
+export class DeviceComponent extends MenuBaseClass implements OnInit {
+  columnTitles = [
+    '_index',
+    'hostname',
+    'product',
+    'serial',
+    'status',
+    'scheduling_status',
+    'device_equipment',
+    'timestamp',
+  ];
+  dataSource = new MatTableDataSource<Device>();
+  pageEvent: PageEvent;
+  deviceStatusEnum = DeviceStatus;
+  schedulingStatusEnum = SchedulingStatus;
+  appliedFilters: FilterItem[];
+
+  sort = '';
+  sortDirection = '';
+
+  @ViewChild(FilterComponent) filterComponent: FilterComponent;
+
+  constructor(private deviceService: DeviceService,
+              appService: AppService,
+              snackBar: MatSnackBar) {
+    super(appService, snackBar);
+  }
+
+  ngOnInit(): void {
+    this.sort = 'hostname';
+    this.sortDirection = 'asc';
+
+    this.filterComponent.setSelectorList(Device);
+    this.getCount();
+    this.getDevices(this.pageSize, this.pageSize * this.pageIndex);
+  }
+
+  /** Gets a total count of devices. */
+  getCount(observer = this.getDefaultCountObservable()) {
+    const filterJSON = (this.appliedFilters) ? JSON.stringify(this.appliedFilters) : '';
+    this.deviceService.getCount(filterJSON).subscribe(observer);
+  }
+
+  /** Gets devices.
+   * @param size A number, at most this many results will be returned.
+   * @param offset A Number of results to skip.
+   */
+  getDevices(size = 0, offset = 0) {
+    this.loading = true;
+    const filterJSON = (this.appliedFilters) ? JSON.stringify(this.appliedFilters) : '';
+    this.deviceService.getDevices(size, offset, filterJSON, this.sort, this.sortDirection)
+      .subscribe(
+        (response) => {
+          this.loading = false;
+          if (this.count >= 0) {
+            let length = 0;
+            if (response.devices) {
+              length = response.devices.length;
+            }
+            const total = length + offset;
+            if (response.has_next) {
+              if (length !== this.pageSize) {
+                this.showSnackbar('Received unexpected number of entities.');
+              } else if (this.count <= total) {
+                this.getCount();
+              }
+            } else {
+              if (this.count !== total) {
+                if (length !== this.count) {
+                  this.getCount();
+                } else if (this.count > total) {
+                  const countObservable = this.getDefaultCountObservable([
+                    () => {
+                      this.pageIndex = Math.floor(this.count / this.pageSize);
+                      this.getDevices(this.pageSize, this.pageSize * this.pageIndex);
+                    }
+                  ]);
+                  this.getCount(countObservable);
+                }
+              }
+            }
+          }
+          this.dataSource.data = response.devices;
+        },
+        (error) => this.showSnackbar(`[${error.status}] ${error.name}`)
+      );
+  }
+
+  /** Hooks a page event and handles properly. */
+  onPageEvent(event: PageEvent) {
+    this.pageSize = event.pageSize;
+    this.pageIndex = event.pageIndex;
+    this.getDevices(this.pageSize, this.pageSize * this.pageIndex);
+    return event;
+  }
+
+  /** Applies a filter and get entities with it. */
+  applyFilters(filters) {
+    this.pageIndex = 0;
+    this.appliedFilters = filters;
+    this.getCount();
+    this.getDevices(this.pageSize, this.pageSize * this.pageIndex);
+  }
+}