pig-media-server 0.3.2 → 2

data/javascripts/components/query_list.js

@@ -0,0 +1,68 @@
+class Item extends React.Component {
+  url(){ return `/?query=${encodeURIComponent(this.props.query)}` }
+
+  click(){
+    this.props.state.open(this.url());
+  }
+
+  delete_query(){
+    if(confirm('Really?')){
+      $.post("/api/r/delete_query_list", {query: this.props.query}).done((data)=>{
+        this.props.update_list();
+      });
+    }
+  }
+  render(){
+    return <span className='main_span'>
+      <a href='javascript:void(0)' onClick={()=> this.click()}>
+        {this.props.query}
+      </a>
+      <a href='javascript:void(0)' className='delete' onClick={()=> this.delete_query()}>
+        Delete
+      </a>
+    </span>
+  }
+}
+
+class QueryList extends React.Component {
+  constructor(props){
+    super(props);
+    this.state = {};
+    this.state.list = [];
+    this.controller = new Controller()
+  }
+
+  update_list(){
+    $.get("/api/r/query_list").done((data)=>{
+      this.state.list = data;
+      this.setState(this.state);
+    });
+  }
+  
+  componentDidMount(){
+    this.update_list();
+  }
+
+  show(){
+    if(location.pathname == '/'){
+      if(!this.controller.query()){
+        return true;
+      } else {
+        return false;
+      }
+    } else {
+      return false
+    }
+  }
+
+  render(){
+    var items = $.map(this.state.list, (query)=>{
+      return <Item key={query} query={query} state={this.props.state} update_list={()=> this.update_list()}/>
+    });
+    return <div>
+    {this.show() ? items : null}
+    </div>
+  }
+}
+
+window.QueryList = QueryList;

data/javascripts/components/sort.js

@@ -0,0 +1,64 @@
+class QuerySave extends React.Component{
+  constructor(props){
+    super(props);
+    this.controller = new Controller();
+  }
+  
+  click(){
+    var query = decodeURIComponent(this.controller.query());
+    $.post("/api/r/query_list", {query: query})
+  }
+  render(){
+    return <span>
+      <a href='javascript:void(0)' onClick={()=> this.click()}>Save Query</a>
+    </span>
+  }
+}
+
+class Sort extends React.Component{
+  constructor(props){
+    super(props);
+    this.controller = new Controller();
+  }
+
+  sort_by_date(){
+    var query = this.controller.query();
+    var new_url = `/?query=${query}&sort=date&order=descending`
+    history.pushState("", "", new_url);
+    this.props.state.initialize();
+  }
+
+  sort_by_name(){
+    var query = this.controller.query();
+    var new_url = `/?query=${query}&sort=name&order=ascending`
+    history.pushState("", "", new_url);
+    this.props.state.initialize();
+  }
+
+  change_order(order){
+    var query = this.controller.query();
+    var sort = this.controller.params().sort;
+    var new_url = `/?query=${query}&sort=${sort}&order=${order}`
+    history.pushState("", "", new_url);
+    this.props.state.initialize();
+
+  }
+
+  can_change_order(){ return !!this.controller.params().sort; }
+  render(){
+    return <p>
+      <a href="javascript:void(0)" onClick={()=> this.sort_by_date()}>Sort By Date</a>
+      <a href="javascript:void(0)" onClick={()=> this.sort_by_name()}>Sort By Name</a>
+      {this.can_change_order() ?
+        <span>
+          <a href="javascript:void(0)" onClick={()=> this.change_order('ascending')}>Ascending</a>
+          <a href="javascript:void(0)" onClick={()=> this.change_order('descending')}>Descending</a>
+        </span>
+        : null
+      }
+      <QuerySave />
+    </p>
+  }
+}
+
+window.Sort = Sort;

data/javascripts/components/watch.js

@@ -0,0 +1,48 @@
+class ChromeCast extends React.Component {
+  is_video(){ var e = ext(this.props.item.name); return e == 'mp4' || e == 'flv'; }
+
+  click(){ window.chrome_cast(this.props.item.url, this.props.item.key) } 
+  render(){
+    return <span>
+      {this.is_video() ?
+        <span>
+          <a 
+            className="watch" 
+            href="javascript:void(0)"
+            title={this.props.item.name}
+            onClick={()=> this.click()}
+          >ChromeCast</a>
+          &nbsp;
+        </span>
+        : null}
+    </span>
+  }
+}
+
+
+class Watch extends React.Component {
+  is_video(){
+    var e = ext(this.props.item.name);
+    return e == 'mp4' || e == 'flv';
+  }
+
+  set_video(){ this.props.state.models.video.set(this.props.item); }
+  render(){
+    return <span>
+      {this.is_video() ?
+        <span>
+          <a 
+            className="watch" 
+            href="javascript:void(0)"
+            title={this.props.item.name}
+            onClick={()=> this.set_video()}
+          >Watch</a>
+          &nbsp;
+        </span>
+        : null}
+    </span>
+  }
+}
+
+window.Watch = Watch;
+window.ChromeCast = ChromeCast;

data/javascripts/controller.js

@@ -0,0 +1,62 @@
+class Controller {
+  route(){
+    var query =this.params().query; 
+    var result = {}
+    switch(location.pathname){
+      case "/":
+        result.page = "list";
+        if(!!query){
+          result.api_url = `/api/r/search?query=${query}`
+          if(!!this.params().sort){ result.api_url += `&sort=${this.params().sort}` }
+          if(!!this.params().order){ result.api_url += `&order=${this.params().order}` }
+          if(!!this.params().page){ result.api_url += `&page=${this.params().page}` }
+          $('title').text(decodeURIComponent(this.query()));
+        } else {
+          result.api_url = null;
+        }
+        break;
+
+      case "/latest":
+        result.page = "list";
+        result.api_url = '/api/r/latest';
+        $('title').text("Latest - Pig Media Server");
+        break;
+      case "/custom":
+        result.page = "list";
+        result.api_url = `/api/r/custom?name=${this.params().name}`;
+        $('title').text(decodeURIComponent(this.params().name));
+        break;
+
+      case "/recommend":
+        result.page = "list";
+        result.api_url = `/api/r/recommend?name=${this.params().name}`;
+        $('title').text('Recommend');
+        break;
+    }
+    return result;
+  }
+
+  
+
+  params(){
+    var arg  = new Object;
+    var pair = location.search.substring(1).split('&');
+    for(var i=0; pair[i]; i++) {
+      var kv = pair[i].split('=');
+      arg[kv[0]] = kv[1];
+    }
+    return arg;
+  }
+
+  query(){ return this.params().query; }
+
+  can_sort_and_paging(){
+    if(location.pathname == '/'){
+      return !!this.query();
+    } else {
+      return false;
+    }
+  }
+}
+
+window.Controller = Controller;

data/javascripts/custom_list.js

@@ -0,0 +1,14 @@
+class CustomList {
+  constructor(app){
+    this.app = app;
+    this.list = []
+  }
+  load(){
+    $.get('/api/r/custom_list').done((data)=>{
+      this.list = data;
+      this.app.update_state();
+    });
+  }
+}
+
+window.CustomList = CustomList;

data/javascripts/event_dispatcher.js

@@ -0,0 +1,21 @@
+class EventDispatcher {
+  constructor(){
+    this.listeners = {};
+  }
+
+  addEventListener(type, callback){
+    if (!this.listeners[type]) { this.listeners[type] = []; }
+    this.listeners[type].push(callback);
+  }
+  clearEventListener(){ this.listeners = {}; }
+
+  dispatchEvent(event) {
+    if (this.listeners[event.type]) {
+      for (var listener in this.listeners[event.type]) {
+        this.listeners[event.type][listener].apply(this.listeners, arguments);
+      }
+    }
+  }
+}
+
+window.EventDispatcher = EventDispatcher;

data/javascripts/recent.js

@@ -0,0 +1,34 @@
+class Recent {
+  constructor(app){
+    this.app = app;
+  }
+
+  login(){ return !!this.app.state.session.user_id }
+
+  load(){
+    if(this.login()){
+      var dd = new Date();
+      $.get('/recents', {stamp: dd.getTime()}).done((data)=>{
+        this.app.state.recents = data;
+        this.app.update_state();
+      });
+    } else {
+    }
+  }
+
+  use(key){
+    if(this.login()){
+      var dd = new Date();
+      $.get('/recents', {stamp: dd.getTime()}).done((recents)=>{
+        recents[key] = {time: parseInt((new Date)/1000), type: 'movie'}
+        $.post("/recents", {data: JSON.stringify(recents)});
+        this.app.state.recents = recents;
+        this.app.update_state();
+      });
+    } else {
+      console.log("kotti");
+    }
+  }
+}
+
+window.Recent = Recent;

data/javascripts/utils.js

@@ -0,0 +1,22 @@
+window.round = (i)=>{
+  i = i*10;
+  return Math.round(i)/10
+}
+
+window.size_pretty = (size)=>{
+  var size = parseFloat(String(size));
+  var result = size;
+  if(size < 1024){
+    result = `${size} Bytes`
+  } else if(size < 1024*1024){
+    result = `${round(size / 1024)} KB`
+  } else if(size < 1024*1024*1024){
+    result = `${round(size / (1024*1024))} MB`
+  } else if(size < 1024*1024*1024*1024){
+    result = `${round(size / (1024*1024*1024))} GB`
+  }
+  return result
+}
+
+
+window.ext = (str)=>{ return String(str).split('.').pop().toLowerCase(); }

data/javascripts/video.js

@@ -0,0 +1,10 @@
+class Video extends EventDispatcher{
+  set(item){
+    this.item = null;
+    this.dispatchEvent({type: 'videoUpdated'});
+    this.item = item;
+    this.dispatchEvent({type: 'videoUpdated'});
+  }
+}
+
+window.Video = Video;

data/lib/pig-media-server/api.rb

@@ -0,0 +1,96 @@
+require 'active_support/concern'
+module PigMediaServer
+  module API
+    extend ActiveSupport::Concern
+    def page
+      params[:page].to_i < 1 ? 1 : params[:page].to_i
+    end
+
+    def size
+      params[:size] ? params[:size].to_i : 50
+    end
+
+    def list_to_json list
+      list.map{|x| 
+        hash = x.to_hash
+        hash['custom_links'] = partial :_custom_links, locals: {record: x}
+        hash['metadata'] = !!x.metadata and x.metadata != ''
+        hash['srt'] = !!x.metadata and x.metadata != ''
+        hash
+      }.to_json
+    end
+
+    included do
+      get '/api/r/latest' do
+        content_type :json
+        list_to_json(Groonga['Files'].select.paginate([key: 'mtime', order: 'descending'], size: size, page: page).map{|x| Pig.new(x)})
+      end
+
+      get '/api/r/custom' do
+        content_type :json
+        c = config['custom_list'][params[:name]]
+        list_to_json(Pig.find JSON.parse(open(c).read))
+      end
+
+      get '/api/r/recommend' do
+        content_type :json
+        list = []
+        begin
+          keys = open("#{config['user_data_path']}/recommend/#{params[:name]}").read.split("\n")
+          list = Pig.find(keys)
+        rescue
+        end
+        list_to_json(list)
+      end
+
+      get '/api/r/search' do
+        content_type :json
+        list_to_json(Pig.search params.merge(page: page))
+      end
+
+      get '/api/r/config' do
+        content_type :json
+        config.to_json
+      end
+
+      get '/api/r/session' do
+        content_type :json
+        session.to_hash.to_json
+      end
+
+      get '/api/r/custom_list' do
+        content_type :json
+        if config['custom_list']
+          config['custom_list'].to_json
+        else
+          {}.to_json
+        end
+      end
+
+      get '/api/r/query_list' do
+        content_type :json
+        Groonga['QueryList'].select.map{|x| x.query }.sort.to_json
+      end
+
+      post '/api/r/query_list' do
+        content_type :json
+        key = Digest::MD5.hexdigest(params[:query])
+        unless Groonga['QueryList'][key]
+          Groonga['QueryList'].add(key)
+          Groonga['QueryList'][key].query = params[:query]
+        end
+        {result: true}.to_json
+      end
+
+      post '/api/r/delete_query_list' do
+        content_type :json
+        key = Digest::MD5.hexdigest(params[:query])
+        if Groonga['QueryList'][key]
+          Groonga['QueryList'].delete key
+        end
+        {result: true}.to_json
+
+      end
+    end
+  end
+end

data/lib/pig-media-server/aspect.rb

@@ -7,22 +7,27 @@ module PigMediaServer
       pit_config = Pit.get 'Pig Media Server'
       while true
         GC.start
-        begin
+        #begin
           q = open("#{pit_config['user_data_path']}/rate/queue.txt").read.split("\n")
           q.each do |x|
-            next if x == ''
+            #begin
+            #next if x == ''
             key = x.split(' ').first
             rate = x.split(' ').last
             pig = Pig.find key
+            next unless pig
+            puts pig.path
             if rate == '16:9'
               system 'MP4Box', '-add', "#{pig.record.path}:par=1:1", '-new', "#{pig.record.path}.MP4"
               FileUtils.mv "#{pig.record.path}.MP4", pig.record.path
             end
+            #rescue
+            #end
           end
           open("#{pit_config['user_data_path']}/rate/queue.txt", 'w'){|x| x.puts ''}
-        rescue => e
-          p e
-        end
+        #rescue => e
+        #  p e
+        #end
         sleep 5
       end
     end

data/lib/pig-media-server/backup.rb

@@ -0,0 +1,69 @@
+require 'pig-media-server/script'
+require 'tmpdir'
+
+module PigMediaServer
+  class Backup
+    def backup
+      pit_config = Pit.get 'Pig Media Server'
+      backup_path = pit_config['backup_path']
+      Dir.mktmpdir do |tmpdir|
+        FileUtils.cd tmpdir
+        date = Date.today
+
+        a = $f.select.to_a
+        open("backup.json", "w"){|f|
+          a.each_with_index{|x,i|
+            f.puts [x._key, x.path, x.metadata, x.srt].to_json
+            puts "#{i+1} / #{a.count}" if i%100 ==0
+          }
+        }
+        open('groonga.schema', 'w'){|x| x.puts Groonga::Schema.dump}
+        system "tar zcvf #{date}.tar.gz backup.json groonga.schema"
+        FileUtils.mv "#{date}.tar.gz", backup_path
+      end
+    end
+
+    def restore_from path
+      raise unless File.exist? path
+      pit_config = Pit.get 'Pig Media Server'
+      Dir.glob("#{pit_config['groonga']}/**/*").each{|x| FileUtils.rm x}
+      Groonga::Database.create path: pit_config['groonga']+"/search"
+      Dir.mktmpdir do |tmpdir|
+        FileUtils.cd tmpdir
+        FileUtils.cp path, "."
+        system 'tar', 'xvf', Dir.glob("*.gz").first
+        system 'ls'
+        FileUtils.rm Dir.glob("*.gz").first 
+        Groonga::Schema.restore(open('groonga.schema').read){|schema| schema.define }
+        $f = Groonga['Files']
+        json =open("backup.json").read.split("\n")
+        json.each_with_index{|x,i|
+          x = JSON.parse x
+          $f.add(x[0])
+          $f[x[0]].path = x[1]
+          $f[x[0]].metadata = x[2]
+          $f[x[0]].srt = x[3]
+          puts "#{i+1} / #{json.count}" if i%100 == 0 or i+1 == json.count
+        }
+      end
+      ary = $f.select.to_a
+
+      lost = []
+
+      ary.each_with_index{|x,i|
+        if x.path.nil?
+          $f.delete x._key
+          next
+        end
+        unless File.exist? x.path
+          puts x.path
+          $f.delete x._key
+          next
+        end
+        x.mtime = File::Stat.new(x.path).mtime.to_i
+        puts "#{i+1} / #{ary.count}" if (i+1)%1000 == 0 or i+1 == ary.count
+      }
+      lost.sort.each{|x| puts x}
+    end
+  end
+end

data/lib/pig-media-server/cli.rb

@@ -1,27 +1,62 @@
-case ARGV[0] 
-when 'setup'
-  require 'fileutils'
-  config = Pit.get("Pig Media Server", require:{
-    'path' => 'Path of your storage',
-    'groonga' => "Path of groonga's files",
-    'exclude_path' => 'Exclude Path(Array)',
-    'user_data_path' => 'Path of User Data'
-  })
+require 'thor'
 
-  puts "Path: #{config['path']}"
-  puts "Groonga: #{config['groonga']}"
-  FileUtils.mkdir_p config['user_data_path']
-when 'crawl'
-  require 'pig-media-server/crawl'
-  PigMediaServer::Crawl.new.all
-when 'server'
-  port = ARGV[1] ? ARGV[1].to_i : 8080
-  require 'pig-media-server/web'
-  PigMediaServer::Web.run! host: '0.0.0.0', port: port
-when 'aspect'
-  require 'pig-media-server/aspect'
-  PigMediaServer::Aspect.new.run
-when 'kindle-send'
-  require 'pig-media-server/kindle_send'
-  PigMediaServer::KindleSend.new.run
+module PigMediaServer
+  class CLI < Thor
+    desc 'setup', 'setup configurations'
+    def setup
+      require 'fileutils'
+      config = Pit.get("Pig Media Server", require:{
+        'hostname' => 'Your host name',
+        'path' => 'Path of your storage',
+        'groonga' => "Path of groonga's files",
+        'exclude_path' => 'Exclude Path(Array)',
+        'user_data_path' => 'Path of User Data'
+      })
+
+      puts "Path: #{config['path']}"
+      puts "Groonga: #{config['groonga']}"
+      FileUtils.mkdir_p config['user_data_path']
+    end
+
+    desc 'crawl', 'run crawler'
+    def crawl
+      require 'pig-media-server/crawl'
+      PigMediaServer::Crawl.new.all
+    end
+
+    desc 'server [PORT]', 'run server'
+    option :bind, :type => :string, :default => '0.0.0.0'
+    option :port, :type => :numeric, :default => 8080
+    def server(port = nil)
+      require 'pig-media-server/web'
+      port ||= options[:port]
+      Sinatra::Base.server.delete 'HTTP' # conflict with HTTP.gem
+      PigMediaServer::Web.run! :bind => options[:bind], :port => port.to_i
+    end
+
+    desc 'aspect', 'fix aspect'
+    def aspect
+      require 'pig-media-server/aspect'
+      PigMediaServer::Aspect.new.run
+    end
+
+    desc 'kindle-send', 'send to kindle'
+    def kindle_send
+      require 'pig-media-server/kindle_send'
+      PigMediaServer::KindleSend.new.run
+    end
+
+    desc 'backup', 'backup groonga\'s data'
+    def backup
+      require 'pig-media-server/backup'
+      PigMediaServer::Backup.new.backup
+    end
+    
+    option :path, required: true
+    desc 'restore', 'restore groonga\'s data'
+    def restore
+      require 'pig-media-server/backup'
+      PigMediaServer::Backup.new.restore_from options[:path]
+    end
+  end
 end