cloudron-box/dashboard/src/components/DiskUsageItem.vue

<script setup>

import { ref, onUnmounted } from 'vue';
import { Button, ProgressBar } from '@cloudron/pankow';
import { prettyDecimalSize } from '@cloudron/pankow/utils';
import AppsModel from '../models/AppsModel.js';
import VolumesModel from '../models/VolumesModel.js';
import SystemModel from '../models/SystemModel.js';

const appsModel = AppsModel.create();
const volumesModel = VolumesModel.create();
const systemModel = SystemModel.create();

const props = defineProps({
  filesystem: Object
});

function hue(numOfSteps, step) {
  const deg = 360/numOfSteps;
  return `hsl(${deg*step} 70% 50%)`;
}

let colorIndex = 0;
let colors = [];
function resetColors(n) {
  colorIndex = 7;
  colors = [];
  for (let i = 0; i < n; i++) colors.push(hue(n, i));
}

function getNextColor() {
  return colors[colorIndex++];
}

const isExpanded = ref(false);
const percent = ref(0);
const contents = ref([]);
const speed = ref(-1);
const highlight = ref(null);

let eventSource = null;

async function refresh() {
  let [error, result] = await appsModel.list();
  if (error) return console.error(error);

  const appsById = {};
  result.forEach(a => { appsById[a.id] = a; });

  [error, result] = await volumesModel.list();
  if (error) return console.error(error);

  const volumesById = {};
  result.forEach(v => { volumesById[v.id] = v; });

  [error, result] = await systemModel.filesystemUsage(props.filesystem.filesystem);
  if (error) return console.error(error);

  contents.value = [];

  eventSource = result;

  eventSource.addEventListener('message', function (message) {
      const payload = JSON.parse(message.data);

      if (payload.type === 'done') {
        percent.value = 100;

        // we first 8 colors are reserved for known system contents
        resetColors(contents.value.length + 8);
        contents.value.forEach(content => {
          // assign fixed colors for known entries
          if (content.id === 'platformdata') content.color = colors[0];
          else if (content.id === 'boxdata') content.color = colors[1];
          else if (content.id === 'maildata') content.color = colors[2];
          else if (content.id === 'cloudron-backup-default') content.color = colors[3];
          else if (content.id === 'docker') content.color = colors[4];
          else if (content.id === 'docker-volumes') content.color = colors[5];
          else if (content.id === '/apps.swap') content.color = colors[6];
          else if (content.id === 'os') content.color = colors[7];
          else content.color = getNextColor();
        });
        contents.value.sort((a, b) => b.usage - a.usage);

        eventSource.close();
      } else if (payload.type === 'progress') {
        percent.value = payload.percent;
      } else {
        if (payload.speed) {
          speed.value = payload.speed;
        } else if (payload.content) {
          if (payload.content.type === 'app') {
            payload.content.app = appsById[payload.content.id];
            if (!payload.content.app) payload.content.uninstalled = true;
            else payload.content.label = payload.content.app.label || payload.content.app.fqdn;
          } else if (payload.content.type === 'volume') {
            payload.content.volume = volumesById[payload.content.id];
            payload.content.label = payload.content.volume ? `Volume ${payload.content.volume.name}` : 'Removed volume';
          } else {
            payload.content.label = payload.content.id;
          }
          contents.value.push(payload.content);
        } else {
          console.error('Unkown data', payload);
        }
      }
  });

  eventSource.addEventListener('error', function (error) {
    console.log('error: errored', error);
    eventSource.close();
  });
}

async function onExpand() {
  if (isExpanded.value) return;

  isExpanded.value = true;

  refresh();
}

onUnmounted(() => {
  if (eventSource) eventSource.close();
});

</script>

<template>
  <div class="disk-item">
    <div class="disk-item-title">
      <div>{{ filesystem.mountpoint }} <small class="text-muted">{{ filesystem.type }}</small></div>
      <Button v-if="isExpanded" small tool plain icon="fa-solid fa-rotate" :disabled="percent < 100" :loading="percent < 100" @click="refresh()"/>
    </div>
    <div class="disk-item-size-and-speed">
      <div>{{ prettyDecimalSize(filesystem.available) }} free of {{ prettyDecimalSize(filesystem.size) }} total</div>
      <div v-if="speed !== -1">I/O Rate: {{ prettyDecimalSize(speed * 1000 * 1000) }}/sec</div>
    </div>
    <div v-if="isExpanded" @mouseout="highlight = null">
      <ProgressBar v-if="percent < 100" mode="indeterminate" :show-label="false"/>
      <div v-else class="disk-size" style="overflow: visible;">
        <div class="disk-used" v-for="content in contents" :key="content.id" v-tooltip="content.id" @mouseover="highlight = content.id" :style="{ 'background-color': content.color, width: 100*content.usage/filesystem.size + '%' }" :class="{ highlight: highlight === content.id }"></div>
      </div>

      <div v-if="percent < 100" style="text-align: center; margin-top: 10px;">Calculating speed and disk usage ... {{ percent }}%</div>
      <div v-else>
        <table style="width: 100%">
          <tr v-for="content in contents" :key="content.id" @mouseover="highlight = content.id" :class="{ highlight: highlight === content.id }">
            <td style="width: 20px"><div class="content-color-indicator" :style="{ backgroundColor: content.color }"></div></td>
            <td>{{ content.label }}</td>
            <td style="text-align: right">{{ prettyDecimalSize(content.usage) }}</td>
          </tr>
        </table>
      </div>
    </div>
    <div v-else>
      <ProgressBar :value="parseInt(filesystem.capacity*100)" :show-label="false"/>
      <div style="text-align: center; margin-top: 10px;">
        <Button plain @click="onExpand()">Details</Button>
      </div>
    </div>
  </div>
</template>

<style scoped>

.disk-item {
  position: relative;
  display: flex;
  flex-direction: column;
  flex-grow: 1;
  margin: 10px;
  max-width: 620px;
  padding: 10px 16px;
  overflow: hidden;
  border-radius: 10px;
  background-color: var(--card-background);
}

.disk-item:focus,
.disk-item:hover {
  box-shadow: 0px 2px 5px rgba(0, 0, 0, 0.1);
  background-color: var(--pankow-color-background-hover) !important;
  text-decoration: none;
}

.disk-item > div {
  margin: 6px 0;
}

.disk-item-title {
  display: flex;
  justify-content: space-between;
  font-weight: bold;
  font-size: 18px;
}

.disk-item-size-and-speed {
  display: flex;
  justify-content: space-between;
}

.disk-size {
  display: flex;
  position: relative;
  background-color: white;
  border-radius: var(--pankow-border-radius);
  height: 12px;
  overflow: hidden;
  margin-bottom: 12px;
}

.disk-used {
  display: inline-block;
  background-color: var(--pankow-color-primary);
  white-space: nowrap;
  height: 12px;
  min-width: 2px;
  transition: all 250ms;
}

.disk-used.highlight {
  transform: scaleY(2);
}

.content-legend-item {
  display: flex;
  align-items: center;
}

.content-color-indicator {
  height: 16px;
  width: 16px;
  border-radius: 4px;
}

.td {
  vertical-align: middle;
}

tr.highlight {
  font-weight: bold;
}

</style>