/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/*
 * The contents of this file are subject to the Netscape Public
 * License Version 1.1 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy of
 * the License at http://www.mozilla.org/NPL/
 *
 * Software distributed under the License is distributed on an "AS
 * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * rights and limitations under the License.
 *
 * The Original Code is mozilla.org code.
 *
 * The Initial Developer of the Original Code is Netscape
 * Communications Corporation.  Portions created by Netscape are
 * Copyright (C) 1998 Netscape Communications Corporation. All
 * Rights Reserved.
 *
 * Contributor(s):
 *   Scott Collins <scc@netscape.com>
 */

#ifndef _nsAWritableString_h__
#define _nsAWritableString_h__

  // WORK IN PROGRESS

  // See also...
#include "nsAReadableString.h"


/*
  This file defines the abstract interfaces |nsAWritableString| and
  |nsAWritableCString|.

  |nsAWritableString| is a string of |PRUnichar|s.  |nsAWritableCString| (note the
  'C') is a string of |char|s.
*/

template <class CharT>
class basic_nsAWritableString
      : public basic_nsAReadableString<CharT>
    /*
      ...
    */
  {
    protected:
      typedef typename basic_nsAReadableString<CharT>::FragmentRequest  FragmentRequest;

      struct Fragment
        {
          CharT*    mStart;
          CharT*    mEnd;
          PRUint32  mFragmentIdentifier;

          Fragment()
              : mStart(0), mEnd(0), mFragmentIdentifier(0)
            {
              // nothing else to do here
            }
        };

    public:
      virtual CharT* GetFragment( Fragment&, FragmentRequest, PRUint32 = 0 ) = 0;

      friend class Iterator;
      class Iterator
          : public std::bidirectional_iterator_tag
        {
          public:
            typedef ptrdiff_t                   difference_type;
            typedef CharT                       value_type;
            typedef CharT*                      pointer;
            typedef CharT&                      reference;
            typedef bidirectional_iterator_tag  iterator_category;

          private:
            friend class basic_nsAWritableString<CharT>;

            Fragment  mFragment;
            CharT*    mPosition;
            basic_nsAWritableString<CharT>* mOwningString;

            void
            normalize_forward()
              {
                if ( mPosition == mFragment.mEnd )
                  if ( mOwningString->GetFragment(mFragment, kNextFragment) )
                    mPosition = mFragment.mStart;
              }

            void
            normalize_backward()
              {
                if ( mPosition == mFragment.mStart )
                  if ( mOwningString->GetFragment(mFragment, kPrevFragment) )
                    mPosition = mFragment.mEnd;
              }

            Iterator( Fragment& aFragment,
                      CharT* aStartingPosition,
                      basic_nsAWritableString<CharT>& aOwningString )
                : mFragment(aFragment),
                  mPosition(aStartingPosition),
                  mOwningString(&aOwningString)
              {
                // nothing else to do here
              }

          public:
            // Iterator( const Iterator& ); ...use default copy-constructor
            // Iterator& operator=( const Iterator& ); ...use default copy-assignment operator

            
            reference
            operator*() const
              {
                return *mPosition;
              }

            pointer
            operator->() const
              {
                return mPosition;
              }

            Iterator&
            operator++()
              {
                ++mPosition;
                normalize_forward();
                return *this;
              }

            Iterator
            operator++( int )
              {
                Iterator result(*this);
                ++mPosition;
                normalize_forward();
                return result;
              }

            Iterator&
            operator--()
              {
                normalize_backward();
                --mPosition;
                return *this;
              }

            Iterator
            operator--( int )
              {
                Iterator result(*this);
                normalize_backward();
                --mPosition;
                return result;
              }

            const Fragment&
            fragment() const
              {
                return mFragment;
              }

            difference_type
            size_forward() const
              {
                return mFragment.mEnd - mPosition;
              }

            difference_type
            size_backward() const
              {
                return mPosition - mFragment.mStart;
              }

            Iterator&
            operator+=( difference_type n )
              {
                if ( n < 0 )
                  return operator-=(-n);

                while ( n )
                  {
                    difference_type one_hop = std::min(n, size_forward());
                    mPosition += one_hop;
                    normalize_forward();
                    n -= one_hop;
                  }

                return *this;
              }

            Iterator&
            operator-=( difference_type n )
              {
                if ( n < 0 )
                  return operator+=(-n);

                while ( n )
                  {
                    difference_type one_hop = std::min(n, size_backward());
                    mPosition -= one_hop;
                    normalize_backward();
                    n -= one_hop;
                  }

                return *this;
              }

            PRBool
            operator==( const Iterator& rhs ) const
              {
                return mPosition == rhs.mPosition;
              }

            PRBool
            operator!=( const Iterator& rhs ) const
              {
                return mPosition != rhs.mPosition;
              }
        };

    public:

#ifdef HAVE_CPP_USING
      using basic_nsAReadableString<CharT>::Begin;
      using basic_nsAReadableString<CharT>::End;
#else
      basic_nsAReadableString<CharT>::ConstIterator
      Begin( PRUint32 aOffset = 0 ) const
        {
          return basic_nsAReadableString<CharT>::Begin(aOffset);
        }

      basic_nsAReadableString<CharT>::ConstIterator
      End( PRUint32 aOffset = 0 ) const
        {
          return basic_nsAReadableString<CharT>::End(aOffset);
        }
#endif

      Iterator
      Begin( PRUint32 aOffset = 0 )
        {
          Fragment fragment;
          CharT* startPos = GetFragment(fragment, kFragmentAt, aOffset);
          return Iterator(fragment, startPos, *this);
        }


      Iterator
      End( PRUint32 aOffset = 0 )
        {
          Fragment fragment;
          CharT* startPos = GetFragment(fragment, kFragmentAt, max(0U, Length()-aOffset));
          return Iterator(fragment, startPos, *this);
        }


      virtual void SetCapacity( PRUint32 ) = 0;
      virtual void SetLength( PRUint32 ) = 0;


      void
      Truncate( PRUint32 aNewLength=0 )
        {
          NS_ASSERTION(aNewLength<=Length(), "Can't use |Truncate()| to make a string longer.");

          if ( aNewLength < Length() )
            SetLength(aNewLength);
        }


      // PRBool SetCharAt( char_type, index_type ) = 0;



      // void ToLowerCase();
      // void ToUpperCase();

      // void StripChars( const CharT* aSet );
      // void StripChar( ... );
      // void StripWhitespace();
      // void ReplaceChar( ... );
      // void ReplaceSubstring( ... );
      // void Trim( ... );
      // void CompressSet( ... );
      // void CompressWhitespace( ... );



      virtual void Assign( const basic_nsAReadableString<CharT>& rhs );
      // virtual void AssignChar( CharT ) = 0;

      virtual void Append( const basic_nsAReadableString<CharT>& );
      virtual void AppendChar( CharT );

      virtual void Insert( const basic_nsAReadableString<CharT>&, PRUint32 atPosition );
      // virtual void InsertChar( CharT, PRUint32 atPosition ) = 0;

      virtual void Cut( PRUint32 cutStart, PRUint32 cutLength );

      virtual void Replace( PRUint32 cutStart, PRUint32 cutLength, const basic_nsAReadableString<CharT>& );


      basic_nsAWritableString<CharT>&
      operator+=( const basic_nsAReadableString<CharT>& rhs )
        {
          Append(rhs);
          return *this;
        }

      basic_nsAWritableString<CharT>&
      operator+=( const basic_nsLiteralString<CharT>& rhs )
        {
          Append(rhs);
          return *this;
        }

      basic_nsAWritableString<CharT>&
      operator=( const basic_nsAReadableString<CharT>& rhs )
        {
          Assign(rhs);
          return *this;
        }

      basic_nsAWritableString<CharT>&
      operator=( const basic_nsLiteralString<CharT>& rhs )
        {
          Assign(rhs);
          return *this;
        }
  };

NS_DEF_STRING_COMPARISONS(basic_nsAWritableString<CharT>)

template <class CharT>
typename basic_nsAWritableString<CharT>::Iterator
copy_chunky( typename basic_nsAReadableString<CharT>::ConstIterator first,
             typename basic_nsAReadableString<CharT>::ConstIterator last,
             typename basic_nsAWritableString<CharT>::Iterator      result )
  {
    while ( first != last )
      {
        basic_nsAReadableString<CharT>::ConstIterator::difference_type lengthToCopy = std::min(first.size_forward(), result.size_forward());
        if ( first.fragment().mStart == last.fragment().mStart )
          lengthToCopy = std::min(lengthToCopy, last.operator->() - first.operator->());

        // assert(lengthToCopy > 0);

        std::char_traits<CharT>::copy(result.operator->(), first.operator->(), lengthToCopy);

        first += lengthToCopy;
        result += lengthToCopy;
      }

    return result;
  }

template <class CharT>
typename basic_nsAWritableString<CharT>::Iterator
copy_backward_chunky( typename basic_nsAReadableString<CharT>::ConstIterator first,
             typename basic_nsAReadableString<CharT>::ConstIterator last,
             typename basic_nsAWritableString<CharT>::Iterator      result )
  {
    while ( first != last )
      {
        typename basic_nsAReadableString<CharT>::ConstIterator::difference_type lengthToCopy = std::min(first.size_backward(), result.size_backward());
        if ( first.fragment().mStart == last.fragment().mStart )
          lengthToCopy = std::min(lengthToCopy, first.operator->() - last.operator->());

        std::char_traits<CharT>::move(result.operator->(), first.operator->(), lengthToCopy);

        first -= lengthToCopy;
        result -= lengthToCopy;
      }

    return result;
  }




template <class CharT>
void
basic_nsAWritableString<CharT>::Assign( const basic_nsAReadableString<CharT>& rhs )
  {
    SetLength(rhs.Length());
    copy_chunky<CharT>(rhs.Begin(), rhs.End(), Begin());
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::Append( const basic_nsAReadableString<CharT>& rhs )
  {
    PRUint32 oldLength = Length();
    SetLength(oldLength + rhs.Length());
    copy_chunky<CharT>(rhs.Begin(), rhs.End(), Begin(oldLength));
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::AppendChar( CharT aChar )
  {
    SetLength(Length()+1);
    *End(1) = aChar;
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::Insert( const basic_nsAReadableString<CharT>& aReadable, PRUint32 aPosition )
  {
    typedef typename basic_nsAReadableString<CharT> readable_t;

    PRUint32 oldLength = Length();
    SetLength(oldLength + aReadable.Length());
    if ( aPosition < oldLength )
      copy_backward_chunky<CharT>(readable_t::Begin(aPosition), readable_t::Begin(oldLength), End());
    else
      aPosition = oldLength;
    copy_chunky<CharT>(aReadable.Begin(), aReadable.End(), Begin(aPosition));
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::Cut( PRUint32 cutStart, PRUint32 cutLength )
  {
    typedef typename basic_nsAReadableString<CharT> readable_t;

    copy_chunky<CharT>(readable_t::Begin(cutStart+cutLength), readable_t::End(), Begin(cutStart));
    SetLength(Length()-cutLength);
  }

template <class CharT>
void
basic_nsAWritableString<CharT>::Replace( PRUint32 cutStart, PRUint32 cutLength, const basic_nsAReadableString<CharT>& aReplacement )
  {
    PRUint32 oldLength = Length();

    cutStart = min(cutStart, oldLength);
    cutLength = min(cutLength, oldLength-cutStart);
    PRUint32 cutEnd = cutStart + cutLength;

    PRUint32 replacementLength = aReplacement.Length();
    PRUint32 replacementEnd = cutStart + replacementLength;

    PRUint32 newLength = oldLength - cutLength + replacementLength;

    typedef typename basic_nsAReadableString<CharT> readable_t;

    if ( cutLength > replacementLength )
      copy_chunky<CharT>(readable_t::Begin(cutEnd), readable_t::End(), Begin(replacementEnd));
    SetLength(newLength);
    if ( cutLength < replacementLength )
      copy_backward_chunky<CharT>(readable_t::Begin(cutEnd), readable_t::Begin(oldLength), Begin(replacementEnd));

    copy_chunky<CharT>(aReplacement.Begin(), aReplacement.End(), Begin(cutStart));
  }

// operator>>
// getline (maybe)

typedef basic_nsAWritableString<PRUnichar> nsAWritableString;
typedef basic_nsAWritableString<char>      nsAWritableCString;

#endif // !defined(_nsAWritableString_h__)