From 068cb4406ae4247b07376e096497cc0bcefa9b1d Mon Sep 17 00:00:00 2001
From: George Kaklamanos <gkaklas@tolabaki.gr>
Date: Sun, 29 Apr 2018 16:30:42 +0300
Subject: [PATCH] Log each downloaded video in database

---
 script.rb | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/script.rb b/script.rb
index 19a57e4..6ff9e9b 100644
--- a/script.rb
+++ b/script.rb
@@ -3,6 +3,8 @@
 require 'rss'
 require 'open-uri'
 require 'youtube-dl'
+require 'sequel'
+require 'date'
 
 if ARGV.empty?
     puts "Please specify the URL of the feed as a parameter"
@@ -11,9 +13,24 @@ end
 
 url=ARGV[0]
 
+DB = Sequel.connect('sqlite://videos.db')
+DB.create_table? :videos do
+    primary_key :id
+    String :url
+    String :filename
+    Integer :filesize
+    DateTime :downloaded
+end
+
+videos=DB[:videos]
+
 open(url) do |rss|
     feed=RSS::Parser.parse(rss)
     feed.items.each do |item|
-        YoutubeDL.download "#{item.link}"
+        video=YoutubeDL.download "#{item.link}"
+        videos.insert(:url => "#{item.link}",
+                      :filename => "#{video.filename}",
+                      :filesize => File.size("#{video.filename}"),
+                      :downloaded => DateTime.now)
     end
 end