data-viewer.component.ts

import {Component, EventEmitter, Input, OnChanges, OnInit, Output, SimpleChanges} from '@angular/core';
import {AddValueDialogComponent, ValueMode} from '../add-value-dialog/add-value-dialog.component';
import {MatDialog, MatSnackBar} from '@angular/material';
import {ConfirmDialogComponent} from '../confirm-dialog/confirm-dialog.component';
import {RedisService} from '../../services/redis.service';
import _ from 'lodash';
import {UtilService} from '../../services/util.service';
import {Observable} from 'rxjs';
import {Store} from '@ngrx/store';
import {ReqFetchTree} from '../../ngrx/actions/redis-actions';

/**
 * the backend type to frontend type map
 */
const TYPE_MAP = {
  string: 'String',
  list: 'List',
  set: 'Set',
  zset: 'Ordered Set',
  hash: 'Hash Map',
};

/**
 * redis data type viewer component
 */
@Component({
  selector: 'app-data-viewer',
  templateUrl: './data-viewer.component.html',
  styleUrls: ['./data-viewer.component.scss']
})
export class DataViewerComponent implements OnInit, OnChanges {
  @Input() pageData = null;
  @Output() onNewValue = new EventEmitter();
  @Output() onDeleteValue = new EventEmitter();
  loadingPageData = false;

  public page = {
    pageIndex: 0,
    pageSize: 20,
    totalSize: 0
  };
  cli$: Observable<any> = null;
  public data = [];
  public setCachedData = null;
  public hashCachedData = null;
  public selectedMap = {};
  public showPagination = false;
  // when item type is string
  public stringValue: string;

  constructor(
    public dialogService: MatDialog,
    private snackBar: MatSnackBar,
    private redisService: RedisService,
    private util: UtilService,
    private _store: Store<any>,
  ) {
    this.cli$ = this._store.select('cli');
  }

  ngOnInit() {

  }

  /*
  * check if object has keys to use in view
  */
  hasKeys() {
    return _.filter(this.selectedMap, (v) => v).length > 0;
  }


  /**
   * is need show table,string type not need table?
   * @return {boolean}
   */
  isNeedShowTable() {
    return ['list', 'set', 'zset', 'hash'].findIndex(v => v === this.pageData.item.type) >= 0;
  }

  /**
   * remove element
   * @param element the element
   * @param cb delete callback
   */
  removeElement(element, cb = null) {
    const t = this.pageData.item.type;
    const pk = this.pageData.item.key;
    let values = [];
    if (element) {
      values = [this.key(element)];
    } else {
      _.each(this.selectedMap, (v, k) => {
        if (v) {
          values.push(k);
        }
      });
    }
    this.dialogService.open(ConfirmDialogComponent, {
      width: '250px', data: {
        title: 'Delete Confirmation',
        message: `Are you sure you want to delete ${element ? 'this' : 'selected'} value${values.length > 1 ? 's' : ''} ?`
      }
    }).afterClosed().subscribe(ret => {
      if (ret) {
        let command = [];
        if (t === 'set') {
          command = ['SREM', pk];
        } else if (t === 'zset') {
          command = ['ZREM', pk];
        } else if (t === 'hash') {
          command = ['HDEL', pk];
        }
        command = command.concat(values);
        this.redisService.call(this.pageData.id, [command]).subscribe(() => {

          _.each(values, v => {
            delete this.selectedMap[v];
          });

          this._store.dispatch(new ReqFetchTree({id: this.pageData.id}));
          this.hashCachedData = null;
          this.setCachedData = null;
          this.fetchData();
          this.util.showMessage('Deleted successfully.');
          if (cb) { cb(); }
        }, () => this.util.showMessage('Delete is failed.'));
      }
    });
  }

  /**
   * get columns by type
   */
  getColumns() {
    const t = this.pageData.item.type;
    if (t === 'list' || t === 'set') {
      let c = ['index', 'value'];
      if (t === 'set') {
        c = ['checkbox'].concat(c);
        c.push('actions');
      }
      return c;
    } else if (t === 'zset') {
      return ['checkbox', 'index', 'value', 'score', 'actions'];
    } else {
      return ['checkbox', 'key', 'value', 'actions'];
    }
  }

  /**
   * get unique key for item
   * @param item
   */
  key(item) {
    const t = this.pageData.item.type;
    if (t === 'hash') {
      return item.key;
    }
    return item.value;
  }


  /**
   * fetch data from redis according to the paging and type
   */
  fetchData() {
    const start = this.page.pageIndex * this.page.pageSize;
    const end = start + this.page.pageSize - 1;
    const type = this.pageData.item.type;
    const key = this.pageData.item.key;
    const instanceId = this.pageData.id;
    const injectValuesToArray = (values) => (_.map(values, (v, index) => ({
      index: this.page.pageIndex * this.page.pageSize + index,
      value: v
    })));

    this.showPagination = false;
    if (type === 'list') {
      this.loadingPageData = true;
      this.redisService.call(instanceId, [['LRANGE', key, start, end], ['LLEN', key]]).subscribe(ret => {
          this.page.totalSize = ret[1];
          this.data = injectValuesToArray(ret[0]);
          this.showPagination = true;
          this.loadingPageData = false;
        }
      );
    } else if (type === 'zset') {
      this.loadingPageData = true;
      this.redisService.call(instanceId, [['ZRANGE', key, start, end, 'withscores'], ['ZCARD', key]]).subscribe(ret => {
          this.page.totalSize = ret[1];
          this.data = [];
          for (let i = 0; i < ret[0].length;) {
            this.data.push({
              index: this.page.pageIndex * this.page.pageSize + (i / 2),
              score: parseFloat(ret[0][i + 1]),
              value: ret[0][i],
            });
            i += 2;
          }
          this.showPagination = true;
          this.loadingPageData = false;
        }
      );
    } else if (type === 'set') {
      if (!this.setCachedData) {
        this.loadingPageData = true;
        this.redisService.call(instanceId, [['SMEMBERS', key]]).subscribe(ret => {
          this.setCachedData = injectValuesToArray(ret[0]);
          this.page.totalSize = this.setCachedData.length;
          this.data = this.setCachedData.slice(start, end);
          this.loadingPageData = false;
          this.showPagination = true;
        });
      } else {
        this.showPagination = true;
        this.data = this.setCachedData.slice(start, end);
      }
    } else if (type === 'hash') {
      if (!this.hashCachedData) {
        this.loadingPageData = true;
        this.redisService.call(instanceId, [['HGETALL', key]]).subscribe(ret => {
            this.hashCachedData = [];
            for (let i = 0; i < ret[0].length;) {
              this.hashCachedData.push({
                key: ret[0][i],
                value: ret[0][i + 1],
              });
              i += 2;
            }
            this.page.totalSize = this.hashCachedData.length;
            this.data = this.hashCachedData.slice(start, end);
            this.loadingPageData = false;
            this.showPagination = true;
          }
        );
      } else {
        this.showPagination = true;
        this.data = this.hashCachedData.slice(start, end);
      }
    } else if (type === 'string') {
      this.loadingPageData = true;
      this.redisService.call(instanceId, [['GET', key]]).subscribe(ret => {
        this.loadingPageData = false;
        this.stringValue = ret[0];
      });
    }
  }

  /**
   * on add new record
   * @param values the record base values
   * @param edit is edit mode
   */
  onAddNewRecords(values, edit = false) {
    const viewMode = new ValueMode();
    viewMode.type = TYPE_MAP[this.pageData.item.type];
    viewMode.hideType = true;

    if (this.pageData.item.type === 'folder') {
      viewMode.type = 'String';
      viewMode.hideType = false;
    }
    viewMode.id = this.pageData.id;
    viewMode.key = this.pageData.item['key'];
    if (values) {
      viewMode.values = values;
      viewMode.isEditMode = true;
    }
    if (values && values.hashMapValues.length > 0) {
      viewMode.onValueDelete = (element, cb) => {
        this.removeElement(element, cb);
        return 0;
      };
    }
    this.dialogService.open(AddValueDialogComponent, {
      minWidth: Math.min(1000, Math.max(480, (viewMode.key.length / 50) * 480)) + 'px',
      minHeight: '400px',
      data: viewMode,
    }).afterClosed().subscribe(ret => {
      if (ret) {
        ret.onSuccess = () => {
          this._store.dispatch(new ReqFetchTree({id: this.pageData.id}));
          if (this.pageData.item.type !== 'folder') {
            this.hashCachedData = null;
            this.setCachedData = null;
            this.fetchData();
          }
        };
        ret.edit = edit;
        this.onNewValue.emit(ret);
      }
    });
  }

  /**
   * on save string event
   */
  onSaveString() {
    if (this.stringValue.trim() === '') {
      this.util.showMessage('The value cannot be empty.');
    } else {
      this.redisService.call(this.pageData.id,
        [['set', this.pageData.item.key, this.stringValue.trim()]]).subscribe(() => {
        this.util.showMessage('Updated successfully.');
      });
    }
  }


  /**
   * get all keys and sub keys from folder item
   * @return {any[]}
   */
  getAllFolderKeys() {
    const result = [];
    const getAllKeys = (arr, children) => {
      children.forEach(i => {
        if (i.type === 'folder') {
          getAllKeys(arr, i.children);
        } else {
          arr.push(i.key);
        }
      });
    };
    getAllKeys(result, this.pageData.item.children);
    return result;
  }


  /**
   * on delete redis value
   */
  onDelete() {
    if (_.some(this.selectedMap, v => v)) {
      this.removeElement(null);
      return;
    }
    this.dialogService.open(ConfirmDialogComponent, {
      width: '320px', data: {
        title: 'Delete Confirmation',
        message: `Are you sure you want to delete
        ${this.pageData.item.type === 'folder' ?  'all the keys inside this branch' : 'the key'}: "${this.pageData.item.key}"?`,
      }
    }).afterClosed().subscribe(ret => {
      if (ret) {
        let keys = [];
        if (this.pageData.item.type === 'folder') {
          keys = this.getAllFolderKeys();
        } else {
          keys = [this.pageData.item.key];
        }
        this.redisService.call(this.pageData.id, [['DEL'].concat(keys)]).subscribe(() => {
          this.util.showMessage('Deleted successfully.');
          this.pageData.item.deleted = true;
          this.onDeleteValue.emit();
          this._store.dispatch(new ReqFetchTree({id: this.pageData.id}));
        }, e => {
          this.util.showMessage('Delete is failed.');
          console.error(e);
        });
      }
    });
  }

  /**
   * on edit map elements
   * @param elements the element arr
   */
  onEditMapElements(elements) {
    this.onAddNewRecords({hashMapValues: JSON.parse(JSON.stringify(elements))}, true);
  }

  /**
   * on edit select values
   */
  onEdit() {
    const keys = [];
    _.each(this.selectedMap, (v, k) => {
      if (v) {
        keys.push(k);
      }
    });

    if (keys.length <= 0) {
      return this.util.showMessage('You need to select a row first.');
    }

    if (this.pageData.item.type === 'hash') {
      const items = [];
      _.each(keys, k => {
        items.push(this.hashCachedData.find(i => i.key === k));
      });
      this.onEditMapElements(items);
    }
  }

  /**
   * when page changed
   * @param page the page item
   */
  onPageEvent(page) {
    this.page = page;
    this.fetchData();
  }

  /**
   * master checkbox changed
   * @param v the value
   */
  masterCheckboxToggle({checked}) {
    if (!checked) {
      this.selectedMap = {};
    } else {
      _.each(this.data, (v) => this.selectedMap[this.key(v)] = true);
    }
  }

  /**
   * check all item is selected or not
   */
  isAllSelected() {
    return _.filter(this.selectedMap, (v) => v).length === this.data.length;
  }

  /**
   * when type changed from parent
   */
  ngOnChanges(changes: SimpleChanges): void {

    if (changes.pageData.previousValue && changes.pageData.currentValue &&
      changes.pageData.currentValue.item.key !== changes.pageData.previousValue.item.key
    ) {
      this.page.pageIndex = 0;
    }

    this.data = [];
    this.selectedMap = {};
    this.setCachedData = null;
    this.hashCachedData = null;
    this.fetchData();
  }
}