グカネータ作成 (#041) (#362)

Reviewed-on: #362
Co-authored-by: miteruzo <miteruzo@naver.com>
Co-committed-by: miteruzo <miteruzo@naver.com>

backend/app/controllers/gekanator_games_controller.rb

@@ -0,0 +1,118 @@
+class GekanatorGamesController < ApplicationController
+  def create
+    return head :not_found unless current_user&.admin?
+
+    guessed_post_id = params.require(:guessed_post_id)
+    correct_post_id = params[:correct_post_id].presence
+    answers = params.require(:answers).as_json
+
+    game = GekanatorGame.new(
+        user: current_user,
+        guessed_post_id:,
+        correct_post_id:,
+        won: correct_post_id.present? && guessed_post_id.to_i == correct_post_id.to_i,
+        question_count: answers.length,
+        answers:)
+
+    if game.save
+      render json: { id: game.id }, status: :created
+    else
+      render json: { errors: game.errors.full_messages }, status: :unprocessable_entity
+    end
+  end
+
+  def extra_questions
+    return head :not_found unless current_user&.admin?
+
+    game = GekanatorGame.find_by(id: params[:id])
+    return head :not_found unless game
+
+    asked_ids = Array(game.answers).filter_map { |answer| answer_question_id(answer) }
+    existing_example_ids =
+        GekanatorQuestionExample.where(post_id: game.correct_post_id)
+                                .select(:gekanator_question_id)
+    # Direct examples only for now; post_similarity-based expansion is deferred.
+    questions =
+        GekanatorQuestion
+        .accepted
+        .where(kind: 'post_similarity', source: 'user_suggested')
+        .where.not(id: existing_example_ids)
+        .order(priority_weight: :desc, id: :asc)
+
+    render json: {
+      questions: questions.filter_map { |question|
+        json = extra_question_json(question)
+        next if asked_ids.include?(json[:id].to_s)
+        next if asked_ids.include?("post-similarity:#{ json[:id] }")
+
+        json
+      }.first(2)
+    }
+  end
+
+  def extra_question_answers
+    return head :not_found unless current_user&.admin?
+
+    game = GekanatorGame.find_by(id: params[:id])
+    return head :not_found unless game
+
+    answer_params = params.require(:answers)
+    if !answer_params.is_a?(Array)
+      return render_validation_error fields: { answers: ['配列で指定してください．'] }
+    end
+
+    answers = answer_params.map { |answer|
+      {
+        question_id: answer.require(:question_id).to_i,
+        answer: answer.require(:answer)
+      }
+    }
+    questions = GekanatorQuestion.where(id: answers.map { _1[:question_id] })
+    question_by_id = questions.index_by(&:id)
+    if questions.length != answers.length
+      return render_validation_error fields: { answers: ['質問が見つかりません．'] }
+    end
+    if questions.any? { |question| question.status != 'accepted' || question.kind != 'post_similarity' }
+      return render_validation_error fields: { answers: ['質問が不正です．'] }
+    end
+
+    ActiveRecord::Base.transaction do
+      answers.each do |item|
+        question = question_by_id[item[:question_id]]
+        example =
+            GekanatorQuestionExample.find_or_initialize_by(
+                gekanator_question: question,
+                post: game.correct_post,
+                user: current_user)
+        example.assign_attributes(
+            gekanator_game: game,
+            answer: item[:answer],
+            source: 'post_game_extra',
+            weight: 1.0)
+        example.save!
+      end
+    end
+
+    render json: { count: answers.length }, status: :created
+  end
+
+  private
+
+  def extra_question_json question
+    {
+      id: question.id,
+      text: question.text,
+      source: question.source,
+      priority_weight: question.priority_weight
+    }
+  end
+
+  def answer_question_id answer
+    value = if answer.is_a?(Hash)
+      answer['question_id'].presence || answer[:question_id].presence ||
+        answer['questionId'].presence || answer[:questionId].presence
+    end
+
+    value&.to_s
+  end
+end

backend/app/controllers/gekanator_posts_controller.rb

@@ -0,0 +1,46 @@
+class GekanatorPostsController < ApplicationController
+  def index
+    return head :not_found unless current_user&.admin?
+
+    posts =
+        Post
+        .preload(tags: :tag_name)
+        .with_attached_thumbnail
+        .order(Arel.sql(
+            'COALESCE(posts.original_created_before - INTERVAL 1 MINUTE, ' \
+            'posts.original_created_from, posts.created_at) DESC, posts.id DESC'))
+
+    render json: { posts: posts.map { |post| post_json(post) } }
+  end
+
+  private
+
+  def post_json post
+    {
+      id: post.id,
+      url: post.url,
+      title: post.title,
+      thumbnail: thumbnail_url(post),
+      thumbnail_base: post.thumbnail_base,
+      original_created_from: post.original_created_from,
+      original_created_before: post.original_created_before,
+      tags: post.tags.map { |tag| tag_json(tag) }
+    }
+  end
+
+  def tag_json tag
+    {
+      id: tag.id,
+      name: tag.name,
+      category: tag.category
+    }
+  end
+
+  def thumbnail_url post
+    return nil unless post.thumbnail.attached?
+
+    rails_storage_proxy_url(post.thumbnail, only_path: false)
+  rescue
+    nil
+  end
+end

backend/app/controllers/gekanator_question_suggestions_controller.rb

@@ -0,0 +1,53 @@
+class GekanatorQuestionSuggestionsController < ApplicationController
+  def create
+    return head :not_found unless current_user&.admin?
+
+    game = GekanatorGame.find_by(id: params.require(:gekanator_game_id))
+    return head :not_found unless game
+
+    suggestion = GekanatorQuestionSuggestion.new(
+        gekanator_game: game,
+        user: current_user,
+        question_text: params.require(:question_text),
+        answer: params.require(:answer))
+
+    if suggestion.valid?
+      ActiveRecord::Base.transaction do
+        suggestion.save!
+        Gekanator::QuestionSuggestionPromoter.call(
+            suggestion: suggestion,
+            user: current_user)
+      end
+
+      render json: {
+        id: suggestion.id,
+        count: game.question_suggestions.count
+      }, status: :created
+    else
+      render_validation_error suggestion
+    end
+  end
+
+  def ai_convert
+    return head :not_found unless current_user&.admin?
+
+    suggestion = GekanatorQuestionSuggestion.find_by(id: params[:id])
+    return head :not_found unless suggestion
+    if Gekanator::AiRunBudget.exceeded_after_next_run?
+      suggestion.gekanator_ai_runs.create!(
+          model: 'budget_guard',
+          status: 'blocked_budget',
+          input_tokens: 0,
+          output_tokens: 0,
+          estimated_cost_jpy: 0)
+      return head :payment_required
+    end
+
+    Gekanator::QuestionSuggestionAiConverter.call(
+        suggestion: suggestion,
+        user: current_user)
+    head :no_content
+  rescue NotImplementedError
+    head :not_implemented
+  end
+end

backend/app/controllers/gekanator_questions_controller.rb

@@ -0,0 +1,83 @@
+class GekanatorQuestionsController < ApplicationController
+  def index
+    return head :not_found unless current_user&.admin?
+
+    questions =
+        GekanatorQuestion
+        .accepted
+        .includes(:gekanator_question_examples)
+        .order(priority_weight: :desc, id: :asc)
+
+    render json: {
+      questions: questions.map { |question| question_json(question) }
+    }
+  end
+
+  private
+
+  def question_json question
+    json = {
+      id: question_id_for(question),
+      text: question.text,
+      kind: question.kind,
+      condition: condition_json(question.condition),
+      source: question.source,
+      priority_weight: question.priority_weight
+    }
+    if question.kind == 'post_similarity'
+      json[:example_answers] = example_answers_json(question)
+    end
+    json
+  end
+
+  def question_id_for question
+    condition = condition_json(question.condition).deep_symbolize_keys
+
+    case condition[:type]
+    when 'tag'
+      "tag:#{ condition[:key] }"
+    when 'source'
+      "source:#{ condition[:host] }"
+    when 'original-year'
+      "original-year:#{ condition[:year] }"
+    when 'original-month'
+      "original-month:#{ condition[:month] }"
+    when 'original-month-day'
+      "original-month-day:#{ condition[:monthDay] || condition[:month_day] }"
+    when 'title-length-greater-than'
+      "title:length-greater-than:#{ condition[:length] }"
+    when 'title-has-ascii'
+      'title:ascii'
+    when 'post-similarity'
+      "post-similarity:#{ question.id }"
+    else
+      "catalog:#{ question.id }"
+    end
+  end
+
+  def condition_json condition
+    json = condition.deep_dup.as_json
+
+    if json['type'] == 'original-month-day' && json['monthDay'].blank?
+      json['monthDay'] = json.delete('month_day')
+    end
+
+    json
+  end
+
+  def example_answers_json question
+    question
+        .gekanator_question_examples
+        .group_by(&:post_id)
+        .transform_values { |examples| aggregate_answer(examples) }
+  end
+
+  def aggregate_answer examples
+    examples
+        .group_by(&:answer)
+        .map { |answer, grouped| [answer, grouped.sum(&:weight), grouped.max_by(&:updated_at)&.updated_at] }
+        .sort_by { |(_answer, weight, updated_at)| [-weight, -(updated_at&.to_f || 0)] }
+        .first
+        &.first
+  end
+end

backend/app/models/gekanator_ai_run.rb

@@ -0,0 +1,21 @@
+class GekanatorAiRun < ApplicationRecord
+  STATUSES = ['pending', 'running', 'succeeded', 'failed', 'blocked_budget'].freeze
+
+  belongs_to :gekanator_question_suggestion
+
+  validates :model, presence: true, length: { maximum: 255 }
+  validates :status, presence: true, inclusion: { in: STATUSES }
+  validates :input_tokens,
+            presence: true,
+            numericality: { only_integer: true, greater_than_or_equal_to: 0 }
+  validates :output_tokens,
+            presence: true,
+            numericality: { only_integer: true, greater_than_or_equal_to: 0 }
+  validates :estimated_cost_jpy,
+            presence: true,
+            numericality: { greater_than_or_equal_to: 0 }
+
+  scope :this_month, lambda {
+    where(created_at: Time.current.beginning_of_month..Time.current.end_of_month)
+  }
+end

backend/app/models/gekanator_game.rb

@@ -0,0 +1,15 @@
+class GekanatorGame < ApplicationRecord
+  belongs_to :user
+  belongs_to :guessed_post, class_name: 'Post'
+  belongs_to :correct_post, class_name: 'Post'
+  has_many :question_suggestions,
+           class_name: 'GekanatorQuestionSuggestion',
+           dependent: :delete_all
+  has_many :question_examples,
+           class_name: 'GekanatorQuestionExample',
+           dependent: :delete_all
+
+  validates :answers, presence: true
+  validates :question_count, numericality: { greater_than_or_equal_to: 0 }
+  validates :won, inclusion: { in: [true, false] }
+end

backend/app/models/gekanator_question.rb

@@ -0,0 +1,23 @@
+class GekanatorQuestion < ApplicationRecord
+  KINDS = ['tag', 'source', 'title', 'original_date', 'post_similarity'].freeze
+  SOURCES = ['user_suggested', 'ai_generated', 'admin_curated'].freeze
+  STATUSES = ['pending', 'accepted', 'rejected', 'disabled'].freeze
+
+  belongs_to :gekanator_question_suggestion, optional: true
+  belongs_to :created_by, class_name: 'User', optional: true
+  has_many :gekanator_question_examples, dependent: :delete_all
+
+  validates :kind, presence: true, inclusion: { in: KINDS }
+  validates :source, presence: true, inclusion: { in: SOURCES }
+  validates :status, presence: true, inclusion: { in: STATUSES }
+  validates :text, presence: true, length: { maximum: 1000 }
+  validates :condition, presence: true
+  validates :priority_weight,
+            presence: true,
+            numericality: {
+              greater_than: 0,
+              less_than_or_equal_to: 3
+            }
+
+  scope :accepted, -> { where(status: 'accepted') }
+end

backend/app/models/gekanator_question_example.rb

@@ -0,0 +1,17 @@
+class GekanatorQuestionExample < ApplicationRecord
+  ANSWERS = GekanatorQuestionSuggestion::ANSWERS
+  SOURCES = ['initial_suggestion', 'post_game_extra'].freeze
+
+  belongs_to :gekanator_question
+  belongs_to :post
+  belongs_to :user
+  belongs_to :gekanator_game, optional: true
+
+  validates :answer, presence: true, inclusion: { in: ANSWERS }
+  validates :source, presence: true, inclusion: { in: SOURCES }
+  validates :weight,
+            presence: true,
+            numericality: {
+              greater_than: 0
+            }
+end

backend/app/models/gekanator_question_suggestion.rb

@@ -0,0 +1,25 @@
+class GekanatorQuestionSuggestion < ApplicationRecord
+  MAX_QUESTIONS_PER_GAME = 3
+  ANSWERS = ['yes', 'no', 'partial', 'probably_no', 'unknown'].freeze
+
+  belongs_to :gekanator_game
+  belongs_to :user
+  has_many :gekanator_questions, dependent: :nullify
+  has_many :gekanator_ai_runs, dependent: :destroy
+
+  validates :question_text, presence: true, length: { maximum: 1000 }
+  validates :answer, presence: true, inclusion: { in: ANSWERS }
+  validates :processed, inclusion: { in: [true, false] }
+  validate :question_suggestion_limit_per_game, on: :create
+
+  private
+
+  def question_suggestion_limit_per_game
+    return if gekanator_game_id.blank?
+
+    count = GekanatorQuestionSuggestion.where(gekanator_game_id:).count
+    if count >= MAX_QUESTIONS_PER_GAME
+      errors.add(:base, '質問追加数を超えてゐます．')
+    end
+  end
+end

backend/app/models/post.rb

@@ -11,6 +11,17 @@ class Post < ApplicationRecord
   has_many :user_post_views, dependent: :delete_all
   has_many :post_similarities, dependent: :delete_all
   has_many :post_versions
+  has_many :gekanator_guessed_games,
+           class_name: 'GekanatorGame',
+           foreign_key: :guessed_post_id,
+           dependent: :delete_all,
+           inverse_of: :guessed_post
+  has_many :gekanator_correct_games,
+           class_name: 'GekanatorGame',
+           foreign_key: :correct_post_id,
+           dependent: :delete_all,
+           inverse_of: :correct_post
+  has_many :gekanator_question_examples, dependent: :delete_all
 
   has_many :parent_post_implications,
            class_name: 'PostImplication',

backend/app/services/gekanator/ai_run_budget.rb

@@ -0,0 +1,22 @@
+module Gekanator
+  class AiRunBudget
+    MONTHLY_LIMIT_JPY = BigDecimal('450').freeze
+    MAX_RUN_ESTIMATED_COST_JPY = BigDecimal('5').freeze
+
+    def self.remaining_monthly_budget_jpy
+      MONTHLY_LIMIT_JPY - monthly_cost_jpy
+    end
+
+    def self.monthly_cost_jpy
+      GekanatorAiRun.this_month.sum(:estimated_cost_jpy)
+    end
+
+    def self.exceeded?
+      monthly_cost_jpy >= MONTHLY_LIMIT_JPY
+    end
+
+    def self.exceeded_after_next_run?
+      monthly_cost_jpy + MAX_RUN_ESTIMATED_COST_JPY >= MONTHLY_LIMIT_JPY
+    end
+  end
+end

backend/app/services/gekanator/question_suggestion_ai_converter.rb

@@ -0,0 +1,18 @@
+module Gekanator
+  class QuestionSuggestionAiConverter
+    def self.call(...) = new(...).call
+
+    def initialize suggestion:, user:
+      @suggestion = suggestion
+      @user = user
+    end
+
+    def call
+      raise NotImplementedError, 'AI question conversion is not implemented yet.'
+    end
+
+    private
+
+    attr_reader :suggestion, :user
+  end
+end

backend/app/services/gekanator/question_suggestion_promoter.rb

@@ -0,0 +1,49 @@
+module Gekanator
+  class QuestionSuggestionPromoter
+    def self.call(...) = new(...).call
+
+    def initialize suggestion:, user:
+      @suggestion = suggestion
+      @user = user
+    end
+
+    def call
+      suggestion.with_lock do
+        return promoted_question if suggestion.processed?
+        return suggestion if suggestion.answer == 'unknown'
+
+        question = GekanatorQuestion.create!(
+            text: suggestion.question_text,
+            kind: 'post_similarity',
+            source: 'user_suggested',
+            status: 'accepted',
+            priority_weight: 1.2,
+            condition: {
+              type: 'post-similarity',
+              postId: suggestion.gekanator_game.correct_post_id,
+              answer: suggestion.answer,
+              threshold: 0.65
+            },
+            gekanator_question_suggestion: suggestion,
+            created_by: user)
+        GekanatorQuestionExample.create!(
+            gekanator_question: question,
+            post: suggestion.gekanator_game.correct_post,
+            user: user,
+            gekanator_game: suggestion.gekanator_game,
+            answer: suggestion.answer,
+            source: 'initial_suggestion')
+        suggestion.update!(processed: true)
+        question
+      end
+    end
+
+    private
+
+    attr_reader :suggestion, :user
+
+    def promoted_question
+      suggestion.gekanator_questions.order(id: :desc).first
+    end
+  end
+end