extension.js

/* jshint esversion: 6 */
/* jshint proto: true */

/*
 * Space Disk Utility
 */

const Clutter = imports.gi.Clutter;
const Lang = imports.lang;

const St = imports.gi.St;
const GObject = imports.gi.GObject;
const GLib = imports.gi.GLib;
const Gtk = imports.gi.Gtk;
const Gio = imports.gi.Gio;
const not = imports.gi.Notify;
const GTop = imports.gi.GTop;

const Main = imports.ui.main;
const Panel = imports.ui.panel;
const PanelMenu = imports.ui.panelMenu;
const PopupMenu = imports.ui.popupMenu;
const MessageTray = imports.ui.messageTray;

const ExtensionUtils = imports.misc.extensionUtils;
const Me = ExtensionUtils.getCurrentExtension();
// const Convenience = Me.imports.convenience;

const Gettext = imports.gettext.domain('gnome-shell-extensions');
const _ = Gettext.gettext;

/*
 * Initialize application
 */
function init(metadata) {
  // Convenience.initTranslations();
}

/*
 * List disk
 */
function ListDisks() {
  this._init.apply(this, arguments);
}

ListDisks.prototype = {
  __proto__: PopupMenu.PopupBaseMenuItem.prototype,

  _init: function(volume_name, volume_size, volume_free, params) {
    PopupMenu.PopupBaseMenuItem.prototype._init.call(this, params);

    this.liste = new St.BoxLayout({ vertical: false });
    this.text_items = this.create_text_items(volume_name, volume_size, volume_free);

    for(let item in this.text_items) {
      this.liste.add_actor(this.text_items[item]);
    }
    this.actor.add_actor(this.liste);
  },
  create_text_items: function(disk_name, disk_free, disk_size) {
    return [
      new St.Label({ text: disk_name, style_class: 'disk-name' }),
      new St.Label({ text: disk_free, style_class: 'disk-free' }),
      new St.Label({ text: disk_size, style_class: 'disk-size' }),
    ];
  },
};

/*
 * Application window
 */
const SpaceIndicator = new Lang.Class({
  Name: 'SpaceIndicator',
  Extends: PanelMenu.Button,

  _TimeoutUd: null,
  _FirstTimeoutId: null,
  _updateProcess_sourceId: null,
  _updateProcess_stream: null,

  _init: function() {
    this.parent(0.0, "SpaceIndicator");
    Gtk.IconTheme.get_default().append_search_path(Me.dir.get_child('icons').get_path());

    this.updateIcon = new St.Icon({icon_name: "hard-disk", style_class: 'system-status-icon'});

    let box = new St.BoxLayout({ vertical: false, style_class: 'panel-status-menu-box' });
    this.label = new St.Label({ text: '', y_expand: true, y_align: Clutter.ActorAlign.CENTER });

    box.add_child(this.updateIcon);
    box.add_child(this.label);
    this.actor.add_child(box);

    // Prepare menu
    this.menuExpander = new PopupMenu.PopupSubMenuMenuItem('');
    this.updatesListMenuLabel = new St.Label();
    this.menuExpander.menu.box.add(this.updatesListMenuLabel);
    this.menuExpander.menu.box.style_class = 'space-list';

    // Configure settings popup
    // let settingsMenuItem = new PopupMenu.PopupMenuItem(_('Settings'));

    //Add menu elements
    // this.menu.addMenuItem(this.menuExpander);
    // this.menu.addMenuItem(settingsMenuItem);

    // Apply action to menu
    // settingsMenuItem.connect('activate', Lang.bind(this, this._openSettings));

    // Add list disk
    this._setupDiskViews();
  },

  _setupDiskViews: function() {
    let arrayMount = [];
    arrayMount.push("/");
    arrayMount.push("/home/jeremy/.wine");
    arrayMount.push("/home/jeremy/Games");

    for(let i = 0; i < arrayMount.length; i++) {
      this._createDefaultApps(arrayMount[i], i, arrayMount);
    }
  },

  _createDefaultApps: function(element, index, array) {
    let d = new Disk(element);
    let name = d._get_mount();
    let free = d._get_size_disk_free() + ' / ';
    let size = d._get_size_disk();
    let info = new ListDisks(name, free, size, {});
    this.menu.addMenuItem(info, 0);
  },

  _openSettings: function () {
    //Util.spawn([ "gnome-shell-extension-prefs", Me.uuid ]);
  },

  destroy: function() {
  },
});

/*
 * System disk information
 */
const Disk = new Lang.Class({
  Name: 'SystemMonitor.Disk',

  _init: function(path) {
    this.path = path;
    this.gtop = new GTop.glibtop_fsusage();
    GTop.glibtop_get_fsusage(this.gtop, this.path);

    this.flag = this.gtop.flags;
    this.block = this.gtop.blocks;
    this.bfree = this.gtop.bfree;
    this.bavail = this.gtop.bavail;
    this.file = this.gtop.files;
    this.ffree = this.gtop.ffree;
    this.unit = 'go';

    /*
    this.settings = Prefs.getSettings();
    this.unit = 'percent';
    if (this.settings.get_boolean('field_diskpercent')) {
      this.unit = 'percent';
    } else {
      this.unit = 'go';
    }
    */
  },

  _get_mount: function() {
    return this.path;
  },

  /* Return disk size */
  _get_size_disk: function() {
    let size = Math.floor((this.block * 3.814697265625) / 1000000);
    let calcul = Math.floor((this.block * 3.814697265625) / 1000000);
    switch(this.unit) {
      case 'percent':
        calcul = 100;
      break;
      case 'go':
        calcul = size;
      break;
    }
    return calcul + this._get_unit();
  },

  /* Return disk size free */
  _get_size_disk_free: function() {
    let free = Math.floor((this.bavail * 3.814697265625) / 1000000);
    let calcul = '';
    switch(this.unit) {
      case 'percent':
        calcul = Math.floor((free * 100) / ((this.block * 3.814697265625) / 1000000));
      break;
      case 'go':
        calcul = free;
      break;
    }
    return '' + calcul;
  },

  _get_unit: function() {
    unit = '';
    switch(this.unit) {
      case 'percent':
        unit = ' %';
      break;
      case 'go':
        unit = ' Go';
      break;
    }
    return unit;
  }
});

/*
 * Enable/Disable extension
 */
let spacediskindicator;

function enable() {
  spacediskindicator = new SpaceIndicator();
  let menu = Main.panel;
  menu.addToStatusArea("SpaceIndicator", spacediskindicator);
}

function disable() {
  spacediskindicator.destroy();
}