import { Actor } from './actor.js';
import { Graph } from './graph.js';
import { Action } from './action.js';
import { CryptoUtil } from './crypto.js';
import params from '../params.js';

class Post extends Actor {
  constructor(forum, authorPublicKey, postContent) {
    const index = forum.posts.countVertices();
    const name = `Post${index + 1}`;
    super(name, forum.scene);
    this.id = postContent.id ?? CryptoUtil.randomUUID();
    this.authorPublicKey = authorPublicKey;
    this.value = 0;
    this.citations = postContent.citations;
    this.title = postContent.title;
    this.totalCitationWeight = this.citations.reduce((total, { weight }) => total += weight, 0);
    if (this.totalCitationWeight > params.revaluationLimit) {
      throw new Error('Post total citation weight exceeds revaluation limit '
      + `(${this.totalCitationWeight} > ${params.revaluationLimit})`);
    }
    if (this.citations.some(({ weight }) => Math.abs(weight) > 1)) {
      throw new Error('Each citation weight must be in the range [-1, 1]');
    }
  }
}

/**
 * Purpose: Maintain a directed, acyclic, weighted graph of posts referencing other posts
 */
export class Forum extends Actor {
  constructor(name, scene) {
    super(name, scene);
    this.posts = new Graph(scene);
    this.actions = {
      addPost: new Action('add post', scene),
      propagateValue: new Action('propagate value', this.scene),
    };
  }

  async addPost(authorId, postContent) {
    const post = new Post(this, authorId, postContent);
    await this.actions.addPost.log(this, post);
    this.posts.addVertex(post.id, post, post.title);
    if (this.scene.flowchart) {
      this.scene.flowchart.log(`${post.id} -- value --> ${post.id}_value[0]`);
    }
    for (const { postId: citedPostId, weight } of post.citations) {
      this.posts.addEdge('citation', post.id, citedPostId, { weight });
      if (this.scene.flowchart) {
        this.scene.flowchart.log(`${post.id} -- ${weight} --> ${citedPostId}`);
      }
    }
    return post.id;
  }

  getPost(postId) {
    return this.posts.getVertexData(postId);
  }

  getPosts() {
    return this.posts.getVerticesData();
  }

  async setPostValue(post, value) {
    post.value = value;
    await post.setValue('value', value);
    if (this.scene.flowchart) {
      this.scene.flowchart.log(`${post.id}_value[${value}]`);
    }
  }

  getTotalValue() {
    return this.getPosts().reduce((total, { value }) => total += value, 0);
  }

  async onValidate(bench, pool, postId, initialValue) {
    initialValue *= params.initialPostValue();

    if (this.scene.flowchart) {
      this.scene.flowchart.log(`${postId}_initial_value[${initialValue}] -- initial value --> ${postId}`);
    }

    const post = this.getPost(postId);
    post.setStatus('Validated');

    // Compute rewards
    const rewards = new Map();
    await this.propagateValue(rewards, pool, post, initialValue);
    // Apply computed rewards
    for (const [id, value] of rewards) {
      bench.reputations.addTokens(id, value);
    }
  }

  async propagateValue(rewards, fromActor, post, increment, depth = 0) {
    if (params.referenceChainLimit >= 0 && depth > params.referenceChainLimit) {
      return [];
    }

    this.actions.propagateValue.log(fromActor, post, `(${increment})`);

    // Recursively distribute reputation to citations, according to weights
    let downstreamRefund = 0;
    for (const { postId: citedPostId, weight } of post.citations) {
      const citedPost = this.getPost(citedPostId);
      downstreamRefund += await this.propagateValue(rewards, post, citedPost, weight * increment, depth + 1);
    }

    // Apply leaching value
    const adjustedIncrement = increment * (1 - params.leachingValue * post.totalCitationWeight) + downstreamRefund;

    // Prevent value from decreasing below zero
    const rawNewValue = post.value + adjustedIncrement;
    const newValue = Math.max(0, rawNewValue);
    const upstreamRefund = rawNewValue < 0 ? rawNewValue : 0;
    const appliedIncrement = newValue - post.value;

    // Increment the value of the post
    await this.setPostValue(post, newValue);

    // Award reputation to post author
    console.log(`reward for post author ${post.authorPublicKey}`, appliedIncrement);

    rewards.set(post.authorPublicKey, appliedIncrement);

    return upstreamRefund;
  }
}