package sqlite3

import (
	"context"

	"github.com/c9s/rockhopper"
)

func init() {
	AddMigration(upAddProfitTable, downAddProfitTable)

}

func upAddProfitTable(ctx context.Context, tx rockhopper.SQLExecutor) (err error) {
	// This code is executed when the migration is applied.

	_, err = tx.ExecContext(ctx, "CREATE TABLE `profits`\n(\n    `gid`                  INTEGER PRIMARY KEY AUTOINCREMENT,\n    `strategy`             VARCHAR(32)    NOT NULL,\n    `strategy_instance_id` VARCHAR(64)    NOT NULL,\n    `symbol`               VARCHAR(8)     NOT NULL,\n    -- average_cost is the position average cost\n    `average_cost`         DECIMAL(16, 8) NOT NULL,\n    -- profit is the pnl (profit and loss)\n    `profit`               DECIMAL(16, 8) NOT NULL,\n    -- net_profit is the pnl (profit and loss)\n    `net_profit`           DECIMAL(16, 8) NOT NULL,\n    -- profit_margin is the pnl (profit and loss)\n    `profit_margin`        DECIMAL(16, 8) NOT NULL,\n    -- net_profit_margin is the pnl (profit and loss)\n    `net_profit_margin`    DECIMAL(16, 8) NOT NULL,\n    `quote_currency`       VARCHAR(10)    NOT NULL,\n    `base_currency`        VARCHAR(10)    NOT NULL,\n    -- -------------------------------------------------------\n    -- embedded trade data --\n    -- -------------------------------------------------------\n    `exchange`             VARCHAR(24)    NOT NULL DEFAULT '',\n    `is_futures`           BOOLEAN        NOT NULL DEFAULT FALSE,\n    `is_margin`            BOOLEAN        NOT NULL DEFAULT FALSE,\n    `is_isolated`          BOOLEAN        NOT NULL DEFAULT FALSE,\n    `trade_id`             BIGINT         NOT NULL,\n    -- side is the side of the trade that makes profit\n    `side`                 VARCHAR(4)     NOT NULL DEFAULT '',\n    `is_buyer`             BOOLEAN        NOT NULL DEFAULT FALSE,\n    `is_maker`             BOOLEAN        NOT NULL DEFAULT FALSE,\n    -- price is the price of the trade that makes profit\n    `price`                DECIMAL(16, 8) NOT NULL,\n    -- quantity is the quantity of the trade that makes profit\n    `quantity`             DECIMAL(16, 8) NOT NULL,\n    -- trade_amount is the quote quantity of the trade that makes profit\n    `quote_quantity`       DECIMAL(16, 8) NOT NULL,\n    `traded_at`            DATETIME(3)    NOT NULL,\n    -- fee\n    `fee_in_usd`           DECIMAL(16, 8),\n    `fee`                  DECIMAL(16, 8) NOT NULL,\n    `fee_currency`         VARCHAR(10)    NOT NULL\n);")
	if err != nil {
		return err
	}

	return err
}

func downAddProfitTable(ctx context.Context, tx rockhopper.SQLExecutor) (err error) {
	// This code is executed when the migration is rolled back.

	_, err = tx.ExecContext(ctx, "DROP TABLE IF EXISTS `profits`;")
	if err != nil {
		return err
	}

	return err
}