hashirama/invidious-mod-JP
1
0
Fork 0
Code Issues Pull requests Actions 2 Projects Releases Wiki Activity

Add email to user table

Browse source
This commit is contained in:
Omar Roth 2018-03-31 10:30:17 -05:00
parent 0e30ddb728
commit ffdf2ea4c5
2 changed files with 29 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
config/sql/users.sql
View file

@ -8,6 +8,9 @@ CREATE TABLE public.users
updated timestamp with time zone, updated timestamp with time zone,
notifications integer, notifications integer,
subscriptions text[] COLLATE pg_catalog."default", subscriptions text[] COLLATE pg_catalog."default",
notifications_viewed timestamp with time zone,
email text COLLATE pg_catalog."default" NOT NULL,
CONSTRAINT users_email_key UNIQUE (email),
CONSTRAINT users_id_key UNIQUE (id) CONSTRAINT users_id_key UNIQUE (id)
) )
WITH ( WITH (

30
src/helpers.cr
View file

@ -85,6 +85,8 @@ class User
updated: Time, updated: Time,
notifications: Int32, notifications: Int32,
subscriptions: Array(String), subscriptions: Array(String),
notifications_viewed: Time,
email: String,
}) })
end end
@ -565,29 +567,41 @@ def get_user(sid, client, headers, db)
args = arg_array(user_array) args = arg_array(user_array)
db.exec("INSERT INTO users VALUES (#{args}) \ db.exec("INSERT INTO users VALUES (#{args}) \
ON CONFLICT (id) DO UPDATE SET updated = $2, subscriptions = $4", user_array) ON CONFLICT (email) DO UPDATE SET id = $1, updated = $2, subscriptions = $4", user_array)
end end
else else
user = fetch_user(sid, client, headers) user = fetch_user(sid, client, headers)
user_array = user.to_a
args = arg_array(user.to_a) args = arg_array(user.to_a)
db.exec("INSERT INTO users VALUES (#{args})", user.to_a)
db.exec("INSERT INTO users VALUES (#{args}) \
ON CONFLICT (email) DO UPDATE SET id = $1, updated = $2, subscriptions = $4", user_array)
end end
return user return user
end end
def fetch_user(sid, client, headers) def fetch_user(sid, client, headers)
feed = client.get("/subscription_manager?action_takeout=1", headers).body feed = client.get("/subscription_manager?disable_polymer=1", headers).body
File.write("feed.html", feed)
channels = [] of String channels = [] of String
feed = XML.parse_html(feed) feed = XML.parse_html(feed)
feed.xpath_nodes("//opml/outline/outline").each do |channel|
id = channel["xmlurl"][-24..-1]
get_channel(id, client, PG_DB)
channels << id feed.xpath_nodes(%q(//a[@class="subscription-title yt-uix-sessionlink"]/@href)).each do |channel|
channel_id = channel.content.lstrip("/channel/").not_nil!
get_channel(channel_id, client, PG_DB)
channels << channel_id
end end
user = User.new(sid, Time.now, 0, channels) email = feed.xpath_node(%q(//a[@class="yt-masthead-picker-header yt-masthead-picker-active-account"]))
if email
email = email.content.lstrip.rstrip
else
email = ""
end
user = User.new(sid, Time.now, 0, channels, Time.now, email)
return user return user
end end