/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * Copyright 2011-2018 Danny Robson <danny@nerdcruft.net>
 */

#pragma once

#include "debug.hpp"
#include "view.hpp"

namespace cruft {
    std::string to_utf8 (const wchar_t*);
    std::string to_utf8 (const std::wstring&);
}


bool
strbegins(const char *restrict str,
          const char *restrict prefix);


namespace cruft {
    template <typename Iterator>
    struct tokeniser {
    public:
        using value_type = typename std::iterator_traits<Iterator>::value_type;
        using range_type = view<Iterator>;

        tokeniser (cruft::view<Iterator,Iterator> _range, value_type _separator):
            m_range (_range),
            m_separator (_separator)
        { ; }

        tokeniser (const char *_range, char _separator):
            m_range (_range),
            m_separator (_separator)
        { ; }

        struct iterator : public std::iterator<
            std::forward_iterator_tag,
            range_type,
            std::size_t
        > {
        public:
            iterator  operator++ (int)
            {
                iterator res(*this);
                ++*this;
                return res;
            }

            iterator& operator++ (void)&
            {
                auto newend = m_range.cend ();
                if (newend != m_end) {
                    CHECK_EQ (*m_range.cend (), m_separator);
                    newend++;
                }

                m_range = {
                    newend,
                    std::find (newend, m_end, m_separator)
                };

                return *this;
            }

            iterator operator+ (int count)
            {
                return std::next (*this, count);
            }

            auto const& operator* (void) const&
            {
                return m_range;
            }

            auto const* operator-> (void) const&
            {
                return &m_range;
            }

            bool operator== (const iterator &rhs) const
            {
                return m_range == rhs.m_range && m_separator == rhs.m_separator;
            }

            bool operator!= (const iterator &rhs) const
            {
                return !(*this == rhs);
            }

        private:
            iterator (range_type _range, value_type _separator):
                m_separator (_separator),
                m_range (_range.cbegin (),
                    std::find (_range.cbegin (),
                    _range.cend (),
                    _separator)
                ),
                m_end (_range.cend ())
            { ;  }

            value_type m_separator;
            range_type m_range;
            Iterator m_end;

            friend tokeniser;
        };

        iterator cbegin (void) const { return { m_range, m_separator }; }
        iterator  begin (void) const { return { m_range, m_separator }; }

        iterator cend (void) const { return { { m_range.cend (), m_range.cend () }, m_separator }; }
        iterator  end (void) const { return { { m_range.cend (), m_range.cend () }, m_separator }; }

        std::size_t
        size (void) const
        {
            return std::count (
                m_range.begin (),
                m_range.end (),
                m_separator
            ) + 1u;
        }

    private:
        const range_type m_range;
        const value_type m_separator;
    };


    template <typename ContainerT>
    tokeniser (ContainerT&, typename ContainerT::value_type) -> tokeniser<typename ContainerT::iterator>;

    tokeniser (const char*,char) -> tokeniser<const char*>;

    ///////////////////////////////////////////////////////////////////////////
    template <typename CharT, std::size_t LengthV>
    auto
    make_tokeniser (CharT (&data)[LengthV], CharT separator)
    {
        return tokeniser { std::begin (data), std::end (data), separator };
    }


    ///////////////////////////////////////////////////////////////////////////
    /// Calculate the line and column of an iterator within a view.
    template <typename IteratorT>
    auto
    character_position (
        cruft::view<IteratorT> data,
        IteratorT const cursor
    ) {
        using difference_type = typename std::iterator_traits<IteratorT>::difference_type;
        difference_type line_count = 0;
        auto current_line = data.begin ();

        do {
            auto next_line = std::find (current_line, data.end (), '\n');
            if (next_line == data.end () || next_line > cursor)
                break;

            ++line_count;
            current_line = next_line + 1;
        } while (1);

        struct {
            difference_type line;
            difference_type column;
        } val = {
            .line = line_count,
            .column = cursor - current_line,
        };

        return val;
    }
}