Add /api/v1/stats

This commit is contained in:
Omar Roth 2019-03-01 19:25:16 -06:00
parent 0eaf8f38a1
commit 4be82c5ca6
3 changed files with 55 additions and 5 deletions

View File

@ -122,11 +122,32 @@ config.video_threads.times do |i|
end end
end end
# stats = Statistics.new statistics = {
# if config.statistics "error" => "Statistics are not availabile.",
# spawn do }
# end if config.statistics_enabled
# end spawn do
loop do
statistics = {
"version" => "2.0",
"software" => {
"name" => "invidious",
"version" => "#{CURRENT_VERSION}-#{CURRENT_COMMIT}",
},
"openRegistrations" => config.registration_enabled,
"usage" => {
"total" => PG_DB.query_one("SELECT count(*) FROM users", as: Int64),
"activeHalfyear" => PG_DB.query_one("SELECT count(*) FROM users WHERE CURRENT_TIMESTAMP - updated < '6 months'", as: Int64),
"activeMonth" => PG_DB.query_one("SELECT count(*) FROM users WHERE CURRENT_TIMESTAMP - updated < '1 month'", as: Int64),
},
"updatedAt" => Time.now.to_unix,
}
sleep 1.minute
Fiber.yield
end
end
end
top_videos = [] of Video top_videos = [] of Video
if config.top_enabled if config.top_enabled
@ -1291,6 +1312,10 @@ post "/preferences" do |env|
registration_enabled ||= "off" registration_enabled ||= "off"
config.registration_enabled = registration_enabled == "on" config.registration_enabled = registration_enabled == "on"
statistics_enabled = env.params.body["statistics_enabled"]?.try &.as(String)
statistics_enabled ||= "off"
config.statistics_enabled = statistics_enabled == "on"
File.write("config/config.yml", config.to_yaml) File.write("config/config.yml", config.to_yaml)
end end
else else
@ -2381,6 +2406,25 @@ end
# API Endpoints # API Endpoints
get "/api/v1/stats" do |env|
env.response.content_type = "application/json"
if statistics["error"]?
halt env, status_code: 500, response: statistics.to_json
end
if !config.statistics_enabled
error_message = {"error" => "Statistics are not enabled."}.to_json
halt env, status_code: 400, response: error_message
end
if env.params.query["pretty"]? && env.params.query["pretty"] == "1"
statistics.to_pretty_json
else
statistics.to_json
end
end
get "/api/v1/captions/:id" do |env| get "/api/v1/captions/:id" do |env|
locale = LOCALES[env.get("locale").as(String)]? locale = LOCALES[env.get("locale").as(String)]?

View File

@ -22,6 +22,7 @@ user: String,
captcha_enabled: {type: Bool, default: true}, captcha_enabled: {type: Bool, default: true},
login_enabled: {type: Bool, default: true}, login_enabled: {type: Bool, default: true},
registration_enabled: {type: Bool, default: true}, registration_enabled: {type: Bool, default: true},
statistics_enabled: {type: Bool, default: false},
admins: {type: Array(String), default: [] of String}, admins: {type: Array(String), default: [] of String},
}) })
end end

View File

@ -189,6 +189,11 @@ function update_value(element) {
<label for="registration_enabled"><%= translate(locale, "Registration enabled? ") %></label> <label for="registration_enabled"><%= translate(locale, "Registration enabled? ") %></label>
<input name="registration_enabled" id="registration_enabled" type="checkbox" <% if config.registration_enabled %>checked<% end %>> <input name="registration_enabled" id="registration_enabled" type="checkbox" <% if config.registration_enabled %>checked<% end %>>
</div> </div>
<div class="pure-control-group">
<label for="statistics_enabled"><%= translate(locale, "Report statistics? ") %></label>
<input name="statistics_enabled" id="statistics_enabled" type="checkbox" <% if config.statistics_enabled %>checked<% end %>>
</div>
<% end %> <% end %>
<% if env.get? "user" %> <% if env.get? "user" %>