from django.db import models
from django.contrib.sitemaps import Sitemap
from django.conf import settings
from jrnl.models import image_url_replace, extract_images
import markdown
import datetime
from itertools import chain

from utils.widgets import markdown_to_html


class Topic(models.Model):
    name = models.CharField(max_length=60)
    slug = models.SlugField()
    pluralized_name = models.CharField(max_length=60)

    def __str__(self):
        return self.name

    def get_absolute_url(self):
        return "/src/topic/%s" % (self.slug)

    @property
    def pub_date(self):
        return datetime.datetime.now()


class Post(models.Model):
    title = models.CharField(max_length=200)
    slug = models.SlugField(unique_for_date='pub_date')
    body_html = models.TextField(blank=True)
    body_markdown = models.TextField()
    pub_date = models.DateTimeField('Date published')
    topics = models.ManyToManyField(Topic, blank=True)
    last_updated = models.DateTimeField(auto_now=True)
    enable_comments = models.BooleanField(default=False)
    has_code = models.BooleanField(default=False)
    PUB_STATUS = (
        (0, 'Draft'),
        (1, 'Published'),
    )
    status = models.IntegerField(choices=PUB_STATUS, default=0)
    meta_description = models.CharField(max_length=256, null=True, blank=True)
    TEMPLATES = (
        (0, 'default'),
    )
    template_name = models.IntegerField(choices=TEMPLATES, default=0)

    class Meta:
        ordering = ('-pub_date',)
        get_latest_by = 'pub_date'
        verbose_name_plural = 'entries'

    def __str__(self):
        return self.title

    def get_absolute_url(self):
        return "/src/%s" % self.slug

    def comment_period_open(self):
        return self.enable_comments and datetime.datetime.today() - datetime.timedelta(30) <= self.pub_date

    def get_images(self):
        return extract_images(self.body_html)

    @property
    def get_previous_published(self):
        return self.get_previous_by_pub_date(status__exact=1)

    @property
    def get_next_published(self):
        return self.get_next_by_pub_date(status__exact=1)

    def save(self):
        md = image_url_replace(self.body_markdown)
        self.body_html = markdown_to_html(md)
        super(Post, self).save()


def get_upload_path(self, filename):
    return "images/src/%s" % filename


class Book(models.Model):
    title = models.CharField(max_length=200)
    image = models.FileField(blank=True, null=True, upload_to=get_upload_path)
    slug = models.SlugField(unique_for_date='pub_date')
    body_html = models.TextField(blank=True)
    body_markdown = models.TextField()
    pub_date = models.DateTimeField('Date published')
    last_updated = models.DateTimeField(auto_now=True)
    PUB_STATUS = (
        (0, 'Draft'),
        (1, 'Published'),
    )
    status = models.IntegerField(choices=PUB_STATUS, default=0)
    price = models.FloatField()
    price_sale = models.FloatField()
    meta_description = models.CharField(max_length=256, null=True, blank=True)
    pages = models.DecimalField(max_digits=3, decimal_places=0, null=True, blank=True)
    DEFAULT = 'details/src_book.html'
    BOOK2 = 'details/src_book_2.html'
    TEMPLATES = (
        (DEFAULT, 'Default'),
        (BOOK2, 'Book Two'),
    )
    template_name = models.CharField(
        max_length=200,
        choices=TEMPLATES,
        default=DEFAULT
    )

    class Meta:
        ordering = ('-pub_date',)
        get_latest_by = 'pub_date'

    def __str__(self):
        return self.title

    def get_absolute_url(self):
        return "/src/books/%s" % (self.slug)

    def get_image_url(self):
        img = self.image.url.split('src/')[1]
        return '%ssrc/%s' % (settings.IMAGES_URL, img)

    def save(self):
        md = image_url_replace(self.body_markdown)
        self.body_html = markdown_to_html(md)
        super(Book, self).save()


'''class SrcDemo(models.Model):
    title = models.CharField(max_length=254)
    slug = models.SlugField()
    body = models.TextField(blank=True, null=True)
    head = models.TextField(blank=True, null=True)
    DEMO_TEMPLATES = (
        (0, 'Blank'),
        (1, 'Basic_light'),
    )
    template = models.IntegerField(choices=DEMO_TEMPLATES, default=0)
    pub_date = models.DateTimeField('Date published', blank=True)

    class Meta:
        verbose_name_plural = "Demos"
        app_label = 'projects'
        ordering = ('-pub_date',)

    def save(self):
        if not self.id:
            self.pub_date = datetime.datetime.now()
        super(SrcDemo, self).save()
'''

class SrcSitemap(Sitemap):
    changefreq = "never"
    priority = 0.7
    protocol = "https"

    def items(self):
        return list(chain(
            Post.objects.all(),
            Book.objects.all(),
            Topic.objects.all()
        ))

    def lastmod(self, obj):
        return obj.pub_date