from django.db import models
from django.utils import timezone

# Create your models here.
class Payments(models.Model):
    creation_date = models.DateTimeField()
    payment_hash = models.CharField(max_length=64, primary_key=True)
    value = models.FloatField()
    fee = models.FloatField()
    status = models.IntegerField()
    index = models.IntegerField()
    chan_out = models.CharField(max_length=20, null=True)
    chan_out_alias = models.CharField(null=True, max_length=32)
    keysend_preimage = models.CharField(null=True, max_length=64)
    message = models.CharField(null=True, max_length=1000)
    cleaned = models.BooleanField(default=False)
    rebal_chan = models.CharField(max_length=20, null=True)
    class Meta:
        app_label = 'gui'

class PaymentHops(models.Model):
    payment_hash = models.ForeignKey('Payments', on_delete=models.CASCADE)
    attempt_id = models.IntegerField()
    step = models.IntegerField()
    chan_id = models.CharField(max_length=20)
    alias = models.CharField(max_length=32)
    chan_capacity = models.BigIntegerField()
    node_pubkey = models.CharField(max_length=66)
    amt = models.FloatField()
    fee = models.FloatField()
    cost_to = models.FloatField()
    class Meta:
        app_label = 'gui'
        unique_together = (('payment_hash', 'attempt_id', 'step'),)

class Invoices(models.Model):
    creation_date = models.DateTimeField()
    settle_date = models.DateTimeField(null=True, default=None)
    r_hash = models.CharField(max_length=64, primary_key=True)
    value = models.FloatField()
    amt_paid = models.BigIntegerField()
    state = models.IntegerField()
    chan_in = models.CharField(max_length=20, null=True)
    chan_in_alias = models.CharField(null=True, max_length=32)
    keysend_preimage = models.CharField(null=True, max_length=64)
    message = models.CharField(null=True, max_length=1000)
    sender = models.CharField(null=True, max_length=66)
    sender_alias = models.CharField(null=True, max_length=32)
    index = models.IntegerField()
    is_revenue = models.BooleanField(default=False)
    class Meta:
        app_label = 'gui'

class Forwards(models.Model):
    forward_date = models.DateTimeField()
    chan_id_in = models.CharField(max_length=20)
    chan_id_out = models.CharField(max_length=20)
    chan_in_alias = models.CharField(null=True, max_length=32)
    chan_out_alias = models.CharField(null=True, max_length=32)
    amt_in_msat = models.BigIntegerField()
    amt_out_msat = models.BigIntegerField()
    fee = models.FloatField()
    class Meta:
        app_label = 'gui'

class Channels(models.Model):
    remote_pubkey = models.CharField(max_length=66)
    chan_id = models.CharField(max_length=20, primary_key=True)
    short_chan_id = models.CharField(max_length=20)
    funding_txid = models.CharField(max_length=64)
    output_index = models.IntegerField()
    capacity = models.BigIntegerField()
    local_balance = models.BigIntegerField()
    remote_balance = models.BigIntegerField()
    unsettled_balance = models.BigIntegerField()
    local_commit = models.IntegerField()
    local_chan_reserve = models.IntegerField()
    num_updates = models.IntegerField()
    initiator = models.BooleanField()
    alias = models.CharField(max_length=32)
    total_sent = models.BigIntegerField()
    total_received = models.BigIntegerField()
    private = models.BooleanField()
    pending_outbound = models.BigIntegerField()
    pending_inbound = models.BigIntegerField()
    htlc_count = models.IntegerField()
    local_base_fee = models.IntegerField()
    local_fee_rate = models.IntegerField()
    local_disabled = models.BooleanField()
    local_cltv = models.IntegerField()
    local_min_htlc_msat = models.BigIntegerField()
    local_max_htlc_msat = models.BigIntegerField()
    remote_base_fee = models.IntegerField()
    remote_fee_rate = models.IntegerField()
    remote_disabled = models.BooleanField()
    remote_cltv = models.IntegerField()
    remote_min_htlc_msat = models.BigIntegerField()
    remote_max_htlc_msat = models.BigIntegerField()
    push_amt = models.BigIntegerField()
    close_address = models.CharField(max_length=100)
    is_active = models.BooleanField()
    is_open = models.BooleanField()
    last_update = models.DateTimeField()
    auto_rebalance = models.BooleanField(default=False)
    ar_amt_target = models.BigIntegerField()
    ar_in_target = models.IntegerField()
    ar_out_target = models.IntegerField()
    ar_max_cost = models.IntegerField()
    fees_updated = models.DateTimeField(default=timezone.now)
    auto_fees = models.BooleanField()
    notes = models.TextField(default='', blank=True)

    def save(self, *args, **kwargs):
        if self.auto_fees is None:
            if LocalSettings.objects.filter(key='AF-Enabled').exists():
                enabled = int(LocalSettings.objects.filter(key='AF-Enabled')[0].value)
            else:
                LocalSettings(key='AF-Enabled', value='0').save()
                enabled = 0
            self.auto_fees = False if enabled == 0 else True
        if not self.ar_out_target:
            if LocalSettings.objects.filter(key='AR-Outbound%').exists():
                outbound_setting = int(LocalSettings.objects.filter(key='AR-Outbound%')[0].value)
            else:
                LocalSettings(key='AR-Outbound%', value='75').save()
                outbound_setting = 75
            self.ar_out_target = outbound_setting
        if not self.ar_in_target:
            if LocalSettings.objects.filter(key='AR-Inbound%').exists():
                inbound_setting = int(LocalSettings.objects.filter(key='AR-Inbound%')[0].value)
            else:
                LocalSettings(key='AR-Inbound%', value='100').save()
                inbound_setting = 100
            self.ar_in_target = inbound_setting
        if not self.ar_amt_target:
            if LocalSettings.objects.filter(key='AR-Target%').exists():
                amt_setting = float(LocalSettings.objects.filter(key='AR-Target%')[0].value)
            else:
                LocalSettings(key='AR-Target%', value='5').save()
                amt_setting = 5
            self.ar_amt_target = int((amt_setting/100) * self.capacity)
        if not self.ar_max_cost:
            if LocalSettings.objects.filter(key='AR-MaxCost%').exists():
                cost_setting = int(LocalSettings.objects.filter(key='AR-MaxCost%')[0].value)
            else:
                LocalSettings(key='AR-MaxCost%', value='65').save()
                cost_setting = 65
            self.ar_max_cost = cost_setting
        super(Channels, self).save(*args, **kwargs)

    class Meta:
        app_label = 'gui'

class Peers(models.Model):
    pubkey = models.CharField(max_length=66, primary_key=True)
    alias = models.CharField(null=True, max_length=32)
    address = models.CharField(max_length=100)
    sat_sent = models.BigIntegerField()
    sat_recv = models.BigIntegerField()
    inbound = models.BooleanField()
    connected = models.BooleanField()
    last_reconnected = models.DateTimeField(null=True, default=None)
    ping_time = models.BigIntegerField(default=0)
    class Meta:
        app_label = 'gui'

class Rebalancer(models.Model):
    requested = models.DateTimeField(default=timezone.now)
    value = models.IntegerField()
    fee_limit = models.FloatField()
    outgoing_chan_ids = models.TextField(default='[]')
    last_hop_pubkey = models.CharField(default='', max_length=66)
    target_alias = models.CharField(default='', max_length=32)
    duration = models.IntegerField()
    start = models.DateTimeField(null=True, default=None)
    stop = models.DateTimeField(null=True, default=None)
    status = models.IntegerField(default=0)
    payment_hash = models.CharField(max_length=64, null=True, default=None)
    manual = models.BooleanField(default=False)
    fees_paid = models.FloatField(null=True, default=None)
    class Meta:
        app_label = 'gui'

class LocalSettings(models.Model):
    key = models.CharField(primary_key=True, default=None, max_length=20)
    value = models.CharField(default=None, max_length=50)
    class Meta:
        app_label = 'gui'

class Onchain(models.Model):
    tx_hash = models.CharField(max_length=64, primary_key=True)
    amount = models.BigIntegerField()
    block_hash = models.CharField(max_length=64)
    block_height = models.IntegerField()
    time_stamp = models.DateTimeField()
    fee = models.IntegerField()
    label = models.CharField(max_length=100)
    class Meta:
        app_label = 'gui'

class Closures(models.Model):
    chan_id = models.CharField(max_length=20)
    funding_txid = models.CharField(max_length=64)
    funding_index = models.IntegerField()
    closing_tx = models.CharField(max_length=64)
    remote_pubkey = models.CharField(max_length=66)
    capacity = models.BigIntegerField()
    close_height = models.IntegerField()
    settled_balance = models.BigIntegerField()
    time_locked_balance = models.BigIntegerField()
    close_type = models.IntegerField()
    open_initiator = models.IntegerField()
    close_initiator = models.IntegerField()
    resolution_count = models.IntegerField()
    closing_costs = models.IntegerField(default=0)
    class Meta:
        app_label = 'gui'
        unique_together = (('funding_txid', 'funding_index'),)

class Resolutions(models.Model):
    chan_id = models.CharField(max_length=20)
    resolution_type = models.IntegerField()
    outcome = models.IntegerField()
    outpoint_tx = models.CharField(max_length=64)
    outpoint_index = models.IntegerField()
    amount_sat = models.BigIntegerField()
    sweep_txid = models.CharField(max_length=64)
    class Meta:
        app_label = 'gui'

class PendingHTLCs(models.Model):
    chan_id = models.CharField(max_length=20)
    alias = models.CharField(max_length=32)
    incoming = models.BooleanField()
    amount = models.BigIntegerField()
    hash_lock = models.CharField(max_length=64)
    expiration_height = models.IntegerField()
    forwarding_channel = models.CharField(max_length=20)
    forwarding_alias = models.CharField(max_length=32)
    class Meta:
        app_label = 'gui'

class FailedHTLCs(models.Model):
    timestamp = models.DateTimeField(default=timezone.now)
    amount = models.IntegerField()
    chan_id_in = models.CharField(max_length=20)
    chan_id_out = models.CharField(max_length=20)
    chan_in_alias = models.CharField(null=True, max_length=32)
    chan_out_alias = models.CharField(null=True, max_length=32)
    chan_out_liq = models.BigIntegerField(null=True)
    chan_out_pending = models.BigIntegerField(null=True)
    wire_failure = models.IntegerField()
    failure_detail = models.IntegerField()
    missed_fee = models.FloatField()
    class Meta:
        app_label = 'gui'

class Autopilot(models.Model):
    timestamp = models.DateTimeField(default=timezone.now)
    chan_id = models.CharField(max_length=20)
    peer_alias = models.CharField(max_length=32)
    setting = models.CharField(max_length=20)
    old_value = models.IntegerField()
    new_value = models.IntegerField()
    class Meta:
        app_label = 'gui'

class Autofees(models.Model):
    timestamp = models.DateTimeField(default=timezone.now)
    chan_id = models.CharField(max_length=20)
    peer_alias = models.CharField(max_length=32)
    setting = models.CharField(max_length=20)
    old_value = models.IntegerField()
    new_value = models.IntegerField()
    class Meta:
        app_label = 'gui'

class PendingChannels(models.Model):
    funding_txid = models.CharField(max_length=64)
    output_index = models.IntegerField()
    local_base_fee = models.IntegerField(null=True, default=None)
    local_fee_rate = models.IntegerField(null=True, default=None)
    local_cltv = models.IntegerField(null=True, default=None)
    auto_rebalance = models.BooleanField(null=True, default=None)
    ar_amt_target = models.BigIntegerField(null=True, default=None)
    ar_in_target = models.IntegerField(null=True, default=None)
    ar_out_target = models.IntegerField(null=True, default=None)
    ar_max_cost = models.IntegerField(null=True, default=None)
    auto_fees = models.BooleanField(null=True, default=None)
    class Meta:
        app_label = 'gui'
        unique_together = (('funding_txid', 'output_index'),)

class AvoidNodes(models.Model):
    pubkey = models.CharField(max_length=66, primary_key=True)
    notes = models.CharField(null=True, max_length=1000)
    updated = models.DateTimeField(default=timezone.now)
    class Meta:
        app_label = 'gui'

class PeerEvents(models.Model):
    timestamp = models.DateTimeField(default=timezone.now)
    chan_id = models.CharField(max_length=20)
    peer_alias = models.CharField(max_length=32)
    event = models.CharField(max_length=20)
    old_value = models.BigIntegerField(null=True)
    new_value = models.BigIntegerField()
    out_liq = models.BigIntegerField()
    class Meta:
        app_label = 'gui'

class HistFailedHTLC(models.Model):
    date = models.DateField(default=timezone.now)
    chan_id_in = models.CharField(max_length=20)
    chan_id_out = models.CharField(max_length=20)
    chan_in_alias = models.CharField(null=True, max_length=32)
    chan_out_alias = models.CharField(null=True, max_length=32)
    htlc_count = models.IntegerField()
    amount_sum = models.BigIntegerField()
    fee_sum = models.BigIntegerField()
    liq_avg = models.BigIntegerField()
    pending_avg = models.BigIntegerField()
    balance_count = models.IntegerField()
    downstream_count = models.IntegerField()
    other_count = models.IntegerField()
    class Meta:
        app_label = 'gui'
        unique_together = (('date', 'chan_id_in', 'chan_id_out'),)