From 2961be636a29ee4668987dd26294c40cf2c09838 Mon Sep 17 00:00:00 2001
From: Albert Yi <albert@carbonfive.com>
Date: Wed, 30 Jan 2019 10:54:27 -0800
Subject: [PATCH] add job hook when post is created

---
 app/logical/automod/update_dynamo_db_job.rb | 13 ++++++++++++-
 app/logical/upload_service.rb               |  5 +++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/app/logical/automod/update_dynamo_db_job.rb b/app/logical/automod/update_dynamo_db_job.rb
index 3e973104c..f59f08d35 100644
--- a/app/logical/automod/update_dynamo_db_job.rb
+++ b/app/logical/automod/update_dynamo_db_job.rb
@@ -2,6 +2,10 @@ module Automod
   class UpdateDynamoDbJob < Struct.new(:post_id)
     extend Memoist
 
+    def self.enabled?
+      Danbooru.config.aws_access_key_id.present?
+    end
+
     def perform
       post = Post.find(post_id)
       data = {
@@ -27,7 +31,14 @@ module Automod
     end
 
     def dynamo_db_client
-      Aws::DynamoDB::Client.new(region: "us-west-1")
+      credentials = Aws::Credentials.new(
+        Danbooru.config.aws_access_key_id,
+        Danbooru.config.aws_secret_access_key
+      )
+      Aws::DynamoDB::Client.new(
+        credentials: credentials,
+        region: "us-west-1"
+      )
     end
     memoize :dynamo_db_client
 
diff --git a/app/logical/upload_service.rb b/app/logical/upload_service.rb
index 1974726b9..8e9b78847 100644
--- a/app/logical/upload_service.rb
+++ b/app/logical/upload_service.rb
@@ -90,6 +90,11 @@ class UploadService
     end
 
     upload.update(status: "completed", post_id: @post.id)
+
+    if @post.is_pending? && Automod::UpdateDynamoDbJob.enabled?
+      Delayed::Job.enqueue(Automod::UpdateDynamoDbJob.new(@post.id), run_at: 84.hours.from_now, queue: "default")
+    end
+
     @post
   end