extension.js

const St = imports.gi.St;
const Clipboard = St.Clipboard.get_default();
const CLIPBOARD_TYPE = St.ClipboardType.CLIPBOARD;
const Main = imports.ui.main;
const PanelMenu = imports.ui.panelMenu;
const PopupMenu = imports.ui.popupMenu;
const MessageTray = imports.ui.messageTray;
const GObject = imports.gi.GObject;
const GLib = imports.gi.GLib;
const Gio = imports.gi.Gio;

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

const EXTENSIONDIR = Me.dir.get_path ();

const CUSTOM_ID = "-1";
const AUTO_VIDEO_ID = "-2";
const AUTO_AUDIO_ID = "-3";
const NO_VIDEO_ID = "-4";
const NO_AUDIO_ID = "-5";

const DEBUG_KEY = 'debug';
let DEBUG = false;
const AUDIO_KEY = 'audio-folder';
let AUDIODIR = "";
const VIDEO_KEY = 'video-folder';
let VIDEODIR = "";
const PLAYLISTS_KEY = 'playlists';
let PLAYLISTS = false;
const QUALITY_KEY = 'preferred-quality';
let QUALITY = 720;
const LANGUAGE_KEY = 'language';
let LANGUAGE = "en";

const A_TIMEOUT = 500;

let animate_event = 0;
let animate_idx = 0;
let installed = false;
let updated = true;
let ydl = "";
let last_text = "";
let uris = [];
let downloads = 0;
let icons = [];

const U2Indicator = GObject.registerClass(class U2Indicator extends PanelMenu.Button {

  _init () {
    super._init (0.0, "Gnome Youtube Downloader", false);

    this.settings = Convenience.getSettings();
    this.get_settings ();

    installed = Convenience.check_install_ydl ();
    if (!installed) Convenience.install_ydl (this.installation_cb.bind(this));
    else Convenience.check_update_ydl (this.version_cb.bind(this));
    ydl = Convenience.ydl;
    this._icon_on = new St.Icon ({
      gicon:Gio.icon_new_for_string (EXTENSIONDIR + "/data/icons/u2b.svg")
    });
    icons.push (new St.Icon ({
      gicon:Gio.icon_new_for_string (EXTENSIONDIR + "/data/icons/025.svg")
    }));
    icons.push (new St.Icon ({
      gicon:Gio.icon_new_for_string (EXTENSIONDIR + "/data/icons/050.svg")
    }));
    icons.push (new St.Icon ({
      gicon:Gio.icon_new_for_string (EXTENSIONDIR + "/data/icons/075.svg")
    }));
    icons.push (new St.Icon ({
      gicon:Gio.icon_new_for_string (EXTENSIONDIR + "/data/icons/100.svg")
    }));
    this.status = new St.Icon ({style: 'icon-size: 20px'});
    this.status.gicon = this._icon_on.gicon;
    let _box = new St.BoxLayout();
    _box.add_actor(this.status);
    this.add_actor (_box);
    this.connect('button-press-event', () => {
      if (!this.menu.isOpen) return;
      if (!installed) installed = Convenience.check_install_ydl ();
      if (this.install) {
        this.install.actor.visible = !installed || !updated;
      }
      this.get_settings ();
      if (installed) this.get_clipboard ();
    });

    this.build_menu ();
    if (downloads) this.add_animation ();
  }

  get_settings () {
    DEBUG = this.settings.get_boolean (DEBUG_KEY);
    AUDIODIR = this.settings.get_string (AUDIO_KEY);
    if (!AUDIODIR) AUDIODIR = Convenience.get_special_dir (GLib.UserDirectory.DIRECTORY_MUSIC);
    VIDEODIR = this.settings.get_string (VIDEO_KEY);
    if (!VIDEODIR) VIDEODIR = Convenience.get_special_dir (GLib.UserDirectory.DIRECTORY_VIDEOS);
    PLAYLISTS = this.settings.get_boolean (PLAYLISTS_KEY);
    QUALITY = this.settings.get_int (QUALITY_KEY);
    LANGUAGE = this.settings.get_string (LANGUAGE_KEY);
    if (Convenience.LANGS.indexOf (LANGUAGE) == -1)
      LANGUAGE = "en";
  }

  installation_cb () {
    let s = updated ? "Installation" : "Updating";
    installed = !!Convenience.ydl;
    s = installed ? s + " complete." : s + " failed.";
    show_notification (s);
    updated = true;
    ydl = Convenience.ydl;
    if (this.install) this.install.actor.visible = !installed;
  }

  version_cb (state) {
    updated = state;
    if (!updated) Convenience.install_ydl (this.installation_cb.bind(this));
  }

  get_clipboard () {
    let self = this;
    Clipboard.get_text (CLIPBOARD_TYPE, function (c, text) {
      if (text && text != last_text) {
        last_text = text;
        self.on_new_text ();
      }
    });
    return true;
  }

  add_animation () {
    if (animate_event) {
      GLib.source_remove (animate_event);
      animate_event = 0;
    }
    animate_event = GLib.timeout_add (100, A_TIMEOUT, this.animate.bind(this));
  }

  animate () {
    if (!downloads) {
      this.status.gicon = this._icon_on.gicon;
      return false;
    }
    this.status.gicon = icons[animate_idx].gicon;
    animate_idx++;
    if (animate_idx > 3) animate_idx = 0;
    return true;
  }

  on_new_text () {
    let ar = last_text.split ("\n");
    uris = [];
    ar.forEach (s=>{
      if (this.is_y2b (s)) uris.push (s.trim ());
    });
    if (uris.length) {
      this.item.set_uri (uris[0]);
    }
  }

  is_y2b (text) {
    let res = GLib.Uri.is_valid (text, 0);

    if (res){
      res = text.indexOf ("https://www.youtube.com") == 0 || text.indexOf ("https://youtu.be") == 0;
    }

    return res;
  }

  build_menu () {
    this.menu.removeAll ();

    this.item = new YoutubeItem ();
    this.menu.addMenuItem (this.item.section);
    this.item.connect ('audio', (item) => {
      if (!installed || !item.uri) return;
      var args = [ydl,"-o",AUDIODIR + "/%(title)s.%(ext)s","-x","-f"];
      if (item.audio.profile.id != AUTO_AUDIO_ID) args.push (item.audio.profile.id);
      else args.push ("m4a");
      if (!PLAYLISTS) args.push ("--no-playlist");
      args.push (item.uri);
      spawn_async (args, (p,s,o) => {
        show_notification ("Complete " + item.uri + s);
        downloads--;
      });
      show_notification ("Starting " + item.uri);
      downloads++;
      this.add_animation ();
    });
    this.item.connect ('video', (item) => {
      if (!installed || !item.uri) return;
      var args = [ydl,"-o",VIDEODIR + "/%(title)s.%(ext)s"];
      var auto, ap = item.audio.profile, vp = item.video.profile;
      if (item.quality.profile.id != CUSTOM_ID) {
        auto = item.quality.profile.auto;
        args.push ("-f");
        if (auto <=240) args.push ("bestvideo[height<="+ auto +"][ext=webm][fps<=30]+worstaudio[ext=webm]/best[height<="+ auto +"][ext=webm]/best[height<="+ auto +"]");
        else if (auto <=720) args.push ("bestvideo[height<="+ auto +"][ext=webm][fps<=30]+bestaudio[ext=webm]/best[height<="+ auto +"][ext=webm]/best[height<="+ auto +"]");
        else args.push ("bestvideo[height<="+ auto +"][ext=webm][fps<=30]+bestaudio/best[height<="+ auto +"]");
      } else if (vp.id != AUTO_VIDEO_ID) {
        args.push ("-f");
        if (((ap.id == AUTO_AUDIO_ID)&&vp.audio) || (ap.id == NO_AUDIO_ID)) {
          args.push (vp.id);
        } else if ((ap.id == AUTO_AUDIO_ID) && !vp.audio) {
          var ext = "webm";
          if (vp.desc.indexOf("webm") == -1) ext = "m4a";
          args.push (vp.id + "+bestaudio[ext=" + ext + "]/bestaudio");
        } else {
          args.push (vp.id + "+" + ap.id);
        }
      } else if ((ap.id != NO_AUDIO_ID) && (ap.id != AUTO_AUDIO_ID)) {
        args.push ("-f");
        var ext = ap.desc.indexOf("webm")>-1?"webm":"mp4";
        args.push ("best[ext="+ ext +"]/best+" + ap.id);
      }
      if (!PLAYLISTS) args.push ("--no-playlist");
      args.push (item.uri);
      var s = "";
      args.forEach(p=>{s += p + " ";});
      print (s);
      spawn_async (args, (p,s,o) => {
        show_notification ("Complete " + item.uri + s);
        downloads--;
      });
      show_notification ("Starting " + item.uri);
      downloads++;
      this.add_animation ();
    });

    this.prefs = new PrefsMenuItem ();
    this.menu.addMenuItem (this.prefs.content);

    this.install = new PopupMenu.PopupMenuItem ("\u26a0 Installing youtube-dl...", {reactive: false});
    this.menu.addMenuItem (this.install);
  }

  remove_events () {
    if (animate_event != 0) GLib.source_remove (animate_event);
    animate_event = 0;
  }
});

const YoutubeItem = GObject.registerClass ({
  Signals: {
    'audio': {},
    'video': {},
  },
}, class YoutubeItem extends GObject.Object {

  _init () {
    super._init ();
    this.section = new PopupMenu.PopupMenuSection ();
    this.item = new PopupMenu.PopupBaseMenuItem ({ reactive: false, can_focus: false });
    this.section.addMenuItem (this.item);

    this.vbox = new St.BoxLayout({ vertical:true, style:"padding:0px;spacing:0", x_expand:true });
    this.item.actor.add_child (this.vbox);

    this.label = new St.Label ({text: " ", style: ''});
    this.label.align = St.Align.START;
    this.vbox.add_child (this.label);

    let box = new St.BoxLayout({ vertical: false, style: 'padding: 4px' });
    this.vbox.add (box);
    this.audio_button = new St.Button ({ label: "Audio", style_class: 'audio-button', x_expand:true });
    box.add (this.audio_button);
    this.video_button = new St.Button ({ label: "Video", style_class: 'video-button', x_expand:true});
    box.add (this.video_button);

    this.quality = new ProfileSubMenuItem (CUSTOM_ID);
    this.section.addMenuItem (this.quality.section);
    this.video = new ProfileSubMenuItem (AUTO_VIDEO_ID);
    this.section.addMenuItem (this.video.section);
    this.audio = new ProfileSubMenuItem (AUTO_AUDIO_ID);
    this.section.addMenuItem (this.audio.section);
    this.quality.connect ('select', (o) => {
      this.on_profile (o.profile);
    });
    this.video.connect ('select', (o) => {
      this.on_profile (o.profile);
    });
    this.audio.connect ('select', (o) => {
      this.on_profile (o.profile);
    });
    this.subtitles = new PopupMenu.PopupSubMenuMenuItem ("Subtitles", false);
    let icon = new St.Icon({ style_class: 'popup-menu-icon' });
    icon.icon_name = "format-text-underline-symbolic";
    this.subtitles.actor.add_child (icon);
    this.section.addMenuItem (this.subtitles);

    this.audio_button.connect ('clicked', () => {
      this.emit ('audio');
      this.item.activate (null);
    });
    this.video_button.connect ('clicked', () => {
      this.emit ('video');
      this.item.activate (null);
    });
    this.label.connect ('notify::text', () => {
      this.section.actor.visible = this.label.text.length > 0;
    });
    this.set_text ("");
    this.uri = "";
    this.profile = {id:"",desc:"Auto Profile",audio:true,video:true};
  }

  set_text (text) {
    this.label.set_text (text);
  }

  set_uri (uri) {
    this.uri = uri;
    this.set_text (uri);
    if (!ydl) return;
    var pipe = new SpawnPipe ([ydl,"-e",this.uri], null, (stdout, err) => {
      if (stdout.length) this.set_text (stdout[0]);
      else if (err) this.set_text (err);
    });
    this.get_quality ();
    this.get_subtitles ();
  }

  get_subtitles (text) {
    this.subtitles.actor.visible = false;
    this.subtitles.menu.removeAll ();
    this.subs = []; this.caps = [];
    var pipe = new SpawnPipe ([ydl,"--list-subs",this.uri], null, (stdout, err) => {
      if (stdout.length) this.get_subs (stdout);
    });
  }

  get_subs (text) {
    var s = "", auto = true;
    for (let i=0; i<text.length; i++) {
      if (text[i].trim().length < 2) continue;
      if (text[i].indexOf ("subtitles for") > 0) auto = false;
      s = text[i].split (" ")[0];
      if (Convenience.LANGS.indexOf (s.toLowerCase()) > -1)
        if (auto) this.caps.push (s);
        else this.subs.push (s);
    }
    let mi = new PopupMenu.PopupMenuItem ("Auto-generated (" + LANGUAGE + ")");
    this.subtitles.menu.addMenuItem (mi);
    mi.connect ('activate', (o) => {
      var pl = PLAYLISTS?"":"--no-playlist ";
      if (GLib.spawn_command_line_async (ydl + " -o " + VIDEODIR +
        "/%(title)s.%(ext)s --write-auto-sub --sub-lang " + LANGUAGE +
        " --sub-format best " +
        "--convert-subs srt --skip-download " + pl + this.uri))
        show_notification ("Starting " + this.uri);
      else show_notification ("Error " + this.uri);
    });
    if (this.subs.length > 0) {
      mi = new PopupMenu.PopupMenuItem ("All Available Languages");
      this.subtitles.menu.addMenuItem (mi);
      mi.connect ('activate', (o) => {
        var pl = PLAYLISTS?"":"--no-playlist ";
        if (GLib.spawn_command_line_async (ydl + " -o " + VIDEODIR +
          "/%(title)s.%(ext)s --write-sub --sub-format best --all-subs " +
          "--convert-subs srt --skip-download " + pl + this.uri))
          show_notification ("Starting " + this.uri);
        else show_notification ("Error " + this.uri);
      });
    }
    this.subs.forEach (p=>{
      mi = new PopupMenu.PopupMenuItem (p);
      this.subtitles.menu.addMenuItem (mi);
      mi.connect ('activate', (o) => {
        var pl = PLAYLISTS?"":"--no-playlist ";
        if (GLib.spawn_command_line_async (ydl + " -o " + VIDEODIR +
          "/%(title)s.%(ext)s --write-sub --sub-format best --sub-lang " +
          o.label.text + " --convert-subs srt --skip-download " + pl + this.uri))
          show_notification ("Starting " + this.uri);
        else show_notification ("Error " + this.uri);
      });
    });
    this.subtitles.actor.visible = true;
  }

  get_quality (text) {
    this.quality.section.actor.visible = false;
    this.video.section.actor.visible = false;
    this.audio.section.actor.visible = false;
    this.quality.add_default ();
    this.video.add_default ();
    this.audio.add_default ();
    this.profiles = [];
    var pipe = new SpawnPipe ([ydl,"-F",this.uri], null, (stdout, err) => {
      if (stdout.length) this.get_profiles (stdout);
    });
  }

  get_profiles (text) {
    var ar = [], s = "", a = true, v = true, id = "", i;
    var qs = [144,240,360,480,720,1080,1440,2160,4320];
    var qd = ["Lowest","Mobile","Video CD","DVD","HD Ready","Full HD","2K Video","4K Video","8K Video"];
    var q = [false,false,false,false,false,false,false,false,false];
    for (i=0; i<text.length; i++) {
      if (text[i].length < 10) continue;
      ar = []; a = true; v = true; s = "";
      text[i] = text[i].replace ("DASH","");
      if (text[i].indexOf ("audio only") > 0) {
        v = false; text[i] = text[i].replace ("audio only","");
      } else if (text[i].indexOf ("video only") > 0) {
        a = false; text[i] = text[i].replace ("video only","");
      }
      text[i].split (" ").forEach (w=>{
        if (w.trim().length > 1) ar.push (w.trim().replace(',',''));
      });
      if (ar.length > 1 && Number.isInteger(parseInt (ar[0]))) {
        id = ar[0];
        if (a && v) ar[0] = "(av)";
        else if (a) ar[0] = "(a)";
        else ar[0] = "(v)";
        ar.forEach (a=>{s += a + " ";});
        this.profiles.push ({id:id,desc:s.trim(),audio:a,video:v});
        if (s.indexOf(" 144p ") > -1) q[0] = true;
        else if (s.indexOf(" 240p ") > -1) q[1] = true;
        else if (s.indexOf(" 360p") > -1) q[2] = true;
        else if (s.indexOf(" 480p") > -1) q[3] = true;
        else if (s.indexOf(" 720p") > -1) q[4] = true;
        else if (s.indexOf(" 1080p") > -1) q[5] = true;
        else if (s.indexOf(" 1440p") > -1) q[6] = true;
        else if (s.indexOf(" 2160p") > -1) q[7] = true;
        else if (s.indexOf(" 4320p") > -1) q[8] = true;
      }
    }
    this.profiles.forEach (p=>{
      if (p.audio) this.audio.add_profile (p);
      if (p.video) this.video.add_profile (p);
    });
    this.audio.add_profile ({id:NO_AUDIO_ID,desc:"No Audio", audio:false});
    this.video.add_profile ({id:NO_VIDEO_ID,desc:"No Video", video:false});
    //this.quality.menu.addMenuItem (new PopupMenu.PopupSeparatorMenuItem());
    for (i=0; i < q.length; i++) {
      if (q[i]) {
        let p = {id:"", desc:qs[i]+"p - "+qd[i], audio:true, video:true, auto:qs[i]};
        this.quality.add_profile (p);
        if (p.auto<=QUALITY) this.quality.on_profile (p);
      }
    }
    if (this.profiles.length > 0) {
      this.quality.section.actor.visible = true;
      this.on_profile (this.quality.profile);
    }
  }

  on_profile (profile) {
    var custom = this.quality.profile.id == CUSTOM_ID;
    this.video.section.actor.visible = custom;
    this.audio.section.actor.visible = custom;

    this.audio_button.visible = !(custom && !this.audio.profile.audio);
    this.video_button.visible = !(custom && !this.video.profile.video);
  }
});

const ProfileSubMenuItem = GObject.registerClass ({
  Signals: {
    'select': {}
  }
}, class ProfileSubMenuItem extends GObject.Object {

  _init (id) {
    super._init ();
    this.section = new PopupMenu.PopupMenuSection ();
    let label_text = "";
    this._icon = new St.Icon({ style_class: 'popup-menu-icon' });
    if (id == CUSTOM_ID) {
      this.default_profile = {id:id,desc:"Custom",audio:true,video:true};
      label_text = "Quality Preset";
      this._icon.icon_name = "emblem-system-symbolic";
    } else if (id == AUTO_VIDEO_ID) {
      this.default_profile = {id:id,desc:"Auto",audio:true,video:true};
      label_text = "Video Track Format";
      this._icon.icon_name = "camera-video-symbolic";
    } else if (id == AUTO_AUDIO_ID) {
      this.default_profile = {id:id,desc:"Auto",audio:true,video:true};
      label_text = "Audio Track Format";
      this._icon.icon_name = "audio-speakers-symbolic";
    } else this.default_profile = {id:"",desc:""};
    this.profile = this.default_profile;
    this.label = new PopupMenu.PopupMenuItem (label_text, { reactive: false, can_focus: false });
    //this.addMenuItem (this.label);
    this.submenu = new PopupMenu.PopupSubMenuMenuItem (this.default_profile.desc, false);
    this.section.addMenuItem (this.submenu);
    this.submenu.actor.add_child (this._icon);
  }

  add_default () {
    this.submenu.menu.removeAll ();
    if (this.default_profile.id != "") this.add_profile (this.default_profile);
  }

  add_profile (profile) {
    let mi = new QualityMenuItem (profile);
    this.submenu.menu.addMenuItem (mi.content);
    mi.connect ('select', (o) => {
      this.on_profile (o.profile);
    });
  }

  on_profile (profile) {
    this.profile = profile;
    this.submenu.label.text = profile.desc;
    this.submenu.setSubmenuShown (false);
    this.emit ('select');
  }
});

const QualityMenuItem = GObject.registerClass ({
  Signals: {
    'select': {}
  }
}, class QualityMenuItem extends GObject.Object {

  _init (profile) {
    super._init ();
    this.content = new PopupMenu.PopupMenuItem (profile.desc);
    this.profile = profile;
    this.content.activate = this.activate.bind (this);
  }

  activate (event) {
    this.emit ('select');
  }
});

const PrefsMenuItem = class You2ber_PrefsMenuItem {

  constructor () {
    this.content = new PopupMenu.PopupBaseMenuItem ({ reactive: false, can_focus: false});
    let l = new St.Label ({text: ' '});
    l.x_expand = true;
    this.content.actor.add (l);
    this.preferences = new St.Button ({ child: new St.Icon ({ icon_name: 'preferences-system-symbolic', icon_size: 28 }), style_class: 'system-menu-action'});
    this.content.actor.add (this.preferences);
    this.preferences.connect ('clicked', () => {
      GLib.spawn_command_line_async (EXTENSIONDIR + "/preferences");
      this.emit ('activate');
    });
    l = new St.Label ({text: ' '});
    l.x_expand = true;
    this.content.actor.add (l);
  }
};

var SpawnPipe = class You2ber_SpawnPipe {

  constructor (args, dir, callback) {
    debug (args);
    dir = dir || "/";
    let exit, pid, stdin_fd, stdout_fd, stderr_fd;
    this.error = "";
    this.stdout = [];
    this.dest = "";

    try {
      [exit, pid, stdin_fd, stdout_fd, stderr_fd] =
        GLib.spawn_async_with_pipes (dir,args,null,GLib.SpawnFlags.DO_NOT_REAP_CHILD,null);
      GLib.close (stdin_fd);
      let outchannel = GLib.IOChannel.unix_new (stdout_fd);
      GLib.io_add_watch (outchannel,100,GLib.IOCondition.IN | GLib.IOCondition.HUP, (channel, condition) => {
        return this.process_line (channel, condition, "stdout");
      });
      let errchannel = GLib.IOChannel.unix_new (stderr_fd);
      GLib.io_add_watch (errchannel,100,GLib.IOCondition.IN | GLib.IOCondition.HUP, (channel, condition) => {
        return this.process_line (channel, condition, "stderr");
      });
      let watch = GLib.child_watch_add (100, pid, (pid, status, o) => {
        debug ("watch handler " + pid + ":" + status + ":" + o);
        GLib.source_remove (watch);
        GLib.spawn_close_pid (pid);
        if (callback) callback (this.stdout, this.error);
      });
    } catch (e) {
      error (e);
    }
  }

  process_line (channel, condition, stream_name) {
    if (condition == GLib.IOCondition.HUP) {
      debug (stream_name, ": has been closed");
      return false;
    }
    try {
      var [,line,] = channel.read_line (), i = -1;
      if (line) {
        debug (stream_name, line);
        if (stream_name == "stderr") {
          this.error = line;
        } else {
          this.stdout.push (line);
        }
      }
    } catch (e) {
       return false;
    }
    return true;
  }
};

function spawn_async (args, callback) {
  callback = callback || null;
  let r, pid;
  try {
    [r, pid] = GLib.spawn_async (null, args, null,
      GLib.SpawnFlags.SEARCH_PATH | GLib.SpawnFlags.DO_NOT_REAP_CHILD, null);
  } catch (e) {
    error (e.message);
    return;
  }
  GLib.child_watch_add (GLib.PRIORITY_DEFAULT, pid, (p, s, o) => {
    if (callback) callback (p, s, o);
  });
}

let freedesktop = null;

function show_notification (message) {
  if (!message) return;
  if (!freedesktop) freedesktop = Gio.DBusProxy.new_for_bus_sync (
    Gio.BusType.SESSION,0,null,"org.freedesktop.Notifications",
    "/org/freedesktop/Notifications", "org.freedesktop.Notifications", null
  );
  freedesktop.call ("Notify", new GLib.Variant("(susssasa{sv}i)", [
      "You2ber",
      42,
      "applications-internet",
      "Youtube",
      message,
      [],
      {},
      5000
    ]), 0,-1,null, (o,a) => {
  });
}

function debug (msg) {
  if (DEBUG) Convenience.debug (msg);
}

function error (msg) {
  Convenience.error (msg);
}

let uindicator;

function init () {
}

function enable () {
  uindicator = new U2Indicator;
  Main.panel.addToStatusArea ("u2ber-indicator", uindicator);
}

function disable () {
  uindicator.remove_events ();
  uindicator.destroy ();
  uindicator = null;
}