/*  Copyright (C) 2004 Garrett A. Kajmowicz
 *
 * This file is part of the uClibc C++ Library.  This library is free
 * software; you can redistribute it and/or modify it under the
 * terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this library; see the file COPYING.  If not, write to the Free
 * Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307,
 * USA.
 */

#include <ios>
#include <cctype>
#include <streambuf>
#include <istream_helpers>
#include <ostream>

#ifndef __STD_HEADER_ISTREAM
#define __STD_HEADER_ISTREAM 1

#pragma GCC visibility push(default)

extern "C++"
{
namespace std
{
  typedef basic_istream<char> istream;

#ifdef __UCLIBCXX_HAS_WCHAR__
  typedef basic_istream<wchar_t> wistream;
#endif

  template <class charT, class traits> basic_istream<charT,traits>& ws(basic_istream<charT,traits>& is);

  template <class charT, class traits> class _UCXXEXPORT basic_istream :
    virtual public basic_ios<charT,traits>
  {
  public:

    typedef charT                     char_type;
    typedef typename traits::int_type int_type;
    typedef typename traits::pos_type pos_type;
    typedef typename traits::off_type off_type;
    typedef basic_streambuf<charT,traits>  streambuf_type;
    typedef traits                    traits_type;

    explicit basic_istream(basic_streambuf<charT,traits>* sb)
      : basic_ios<charT, traits>(sb), count_last_ufmt_input(0)
    {
      basic_ios<charT, traits>::init(sb);
    }

    virtual ~basic_istream() { }

    class sentry;

    basic_istream<charT,traits>& operator>>(basic_istream<charT,traits>& (*pf)(basic_istream<charT,traits>&));
    basic_istream<charT,traits>& operator>>(basic_ios<charT,traits>& (*pf)(basic_ios<charT,traits>&));
    basic_istream<charT,traits>& operator>>(ios_base& (*pf)(ios_base&));
    basic_istream<charT,traits>& operator>>(bool& n);
    basic_istream<charT,traits>& operator>>(short& n);
    basic_istream<charT,traits>& operator>>(unsigned short& n);
    basic_istream<charT,traits>& operator>>(int& n);
    basic_istream<charT,traits>& operator>>(unsigned int& n);
    basic_istream<charT,traits>& operator>>(long& n);
    basic_istream<charT,traits>& operator>>(unsigned long& n);
    basic_istream<charT,traits>& operator>>(void*& p);
    basic_istream<charT,traits>& operator>>(basic_streambuf<char_type,traits>* sb);

#ifdef CONFIG_HAVE_FLOAT
    basic_istream<charT,traits>& operator>>(float& f);
    basic_istream<charT,traits>& operator>>(double& f);
    basic_istream<charT,traits>& operator>>(long double& f);
#endif

    _UCXXEXPORT streamsize gcount() const
    {
      return count_last_ufmt_input;
    }

    _UCXXEXPORT int_type get();    //below
    _UCXXEXPORT basic_istream<charT,traits>& get(char_type& c);  //Below

    _UCXXEXPORT basic_istream<charT,traits>& get(char_type* s, streamsize n)
    {
      return get(s, n, basic_ios<charT,traits>::widen('\n'));
    }

    _UCXXEXPORT basic_istream<charT,traits>& get(char_type* s, streamsize n, char_type delim)
    {
      sentry(*this, true);
      streamsize i = 0;
      int_type c;
      for (i = 0; i < n-1; ++i)
      {
        c = basic_ios<charT, traits>::mstreambuf->sgetc();
        basic_ios<charT, traits>::mstreambuf->sbumpc();
        if (c == traits::eof())
        {
          if (i==0)
          {
            basic_ios<charT,traits>::setstate(ios_base::failbit);
          }
          else
          {
            basic_ios<charT,traits>::setstate(ios_base::eofbit);
          }

          break;
        }

        if (c == delim)
        {
          if (i==0)
          {
            basic_ios<charT,traits>::setstate(ios_base::failbit);
          }

          basic_ios<charT, traits>::mstreambuf->sputbackc(c);
          break;
        }

        s[i] = c;
      }

      s[i] = traits::eos();
      count_last_ufmt_input = i;
      return *this;
    }

    _UCXXEXPORT basic_istream<charT,traits>& get(basic_streambuf<char_type,traits>& sb)
    {
      return get(sb, basic_ios<charT,traits>::widen('\n'));
    }

    _UCXXEXPORT basic_istream<charT,traits>& get(basic_streambuf<char_type,traits>& sb, char_type delim)
    {
      sentry(*this, true);
      streamsize i = 0;
      int_type c;
      while (1)
      {
        // We will exit internally based upon error conditions

        c = basic_ios<charT, traits>::mstreambuf->sgetc();
        if (c == traits::eof())
        {
          if (i==0)
          {
            basic_ios<charT,traits>::setstate(ios_base::failbit);
          }
          else
          {
            basic_ios<charT,traits>::setstate(ios_base::eofbit);
          }

          count_last_ufmt_input = i;
          return *this;
        }

        if (c == delim)
        {
          if (i == 0)
          {
            basic_ios<charT,traits>::setstate(ios_base::failbit);
          }

          count_last_ufmt_input = i;
          return *this;
        }

        if (sb.sputc(c) != c)
        {
          // Error doing output

          count_last_ufmt_input = i;
          return *this;
        }

        ++i;
        basic_ios<charT, traits>::mstreambuf->sbumpc();
      }
    }

    _UCXXEXPORT basic_istream<charT,traits>& getline(char_type* s, streamsize n)
    {
      return getline(s, n, basic_ios<charT,traits>::widen('\n'));
    }

    _UCXXEXPORT basic_istream<charT,traits>& getline(char_type* s, streamsize n, char_type delim)
    {
      sentry(*this, true);
      streamsize i = 0;
      int_type c;
      for (i = 0; i < n-1; ++i)
      {
        c = basic_ios<charT, traits>::mstreambuf->sgetc();
        if (c == traits::eof())
        {
          if (basic_ios<charT,traits>::eof())
          {
            basic_ios<charT,traits>::setstate(ios_base::failbit);
          }
          else
          {
            basic_ios<charT,traits>::setstate(ios_base::eofbit);
          }

          count_last_ufmt_input = i;
          s[i] = traits::eos();
          return *this;
        }

        if (basic_ios<charT, traits>::mstreambuf->sbumpc()==traits::eof())
        {
          basic_ios<charT,traits>::setstate(ios_base::eofbit);
        }

        if (c == delim)
        {
          count_last_ufmt_input = i+1;
          s[i] = traits::eos();
          return *this;
        }

        s[i] = c;
      }

      s[n-1] = traits::eos();
      return *this;
    }

    _UCXXEXPORT basic_istream<charT,traits>& ignore (streamsize n = 1, int_type delim = traits::eof())
    {
      sentry(*this, true);
      streamsize i;
      int_type c;
      for (i = 0; i < n; ++i)
      {
        c = basic_ios<charT, traits>::mstreambuf->sgetc();
        if (c == traits::eof())
        {
          basic_ios<charT,traits>::setstate(ios_base::eofbit);
          return *this;
        }

        basic_ios<charT, traits>::mstreambuf->sbumpc();
        if (c == delim)
        {
          return *this;
        }
      }

      return *this;
    }

    _UCXXEXPORT int_type peek()
    {
      if (basic_ios<charT,traits>::good() == false)
      {
        return traits::eof();
      }
      else
      {
        int_type c = basic_ios<charT, traits>::mstreambuf->sgetc();
        if (c == traits::eof())
        {
          basic_ios<charT,traits>::setstate(ios_base::eofbit);
        }

        return basic_ios<charT, traits>::mstreambuf->sgetc();
      }
    }

    _UCXXEXPORT basic_istream<charT,traits>& read (char_type* s, streamsize n)
    {
      sentry(*this, true);
      streamsize i;
      int_type c;
      for (i = 0; i < n; ++i)
      {
        c = basic_ios<charT, traits>::mstreambuf->sgetc();

        if (c == traits::eof())
        {
          basic_ios<charT,traits>::setstate(ios_base::failbit);
          basic_ios<charT,traits>::setstate(ios_base::eofbit);
          count_last_ufmt_input = i;
          return *this;
        }

        basic_ios<charT, traits>::mstreambuf->sbumpc();
        s[i] = c;
      }

      count_last_ufmt_input = n;
      return *this;
    }

    _UCXXEXPORT streamsize readsome(char_type* s, streamsize n)
    {
      sentry(*this, true);
      if (!basic_ios<charT,traits>::good())
      {
        count_last_ufmt_input = 0;
        basic_ios<charT,traits>::setstate(ios_base::failbit);
        return 0;
      }

      if (basic_ios<charT, traits>::mstreambuf->in_avail()  ==  -1)
      {
        count_last_ufmt_input=0;
        basic_ios<charT,traits>::setstate(ios_base::eofbit);
        return 0;
      }

      if (n > basic_ios<charT, traits>::mstreambuf->in_avail())
      {
        n = basic_ios<charT, traits>::mstreambuf->in_avail();
      }

      streamsize i;
      int_type c;

      for (i = 0; i < n; ++i)
      {
        c = basic_ios<charT, traits>::mstreambuf->sgetc();
        basic_ios<charT, traits>::mstreambuf->sbumpc();
        s[i] = c;
      }

      count_last_ufmt_input = n;
      return n;
    }

    _UCXXEXPORT basic_istream<charT,traits>& putback(char_type c)
    {
      sentry(*this, true);
      if (!basic_ios<charT,traits>::good())
      {
        basic_ios<charT,traits>::setstate(ios_base::failbit);
        return *this;
      }

      if (basic_ios<charT, traits>::mstreambuf == 0){

        basic_ios<charT,traits>::setstate(ios_base::badbit);
        return *this;
      }

      if (basic_ios<charT, traits>::mstreambuf->sputbackc(c) == traits::eof())
      {
        basic_ios<charT,traits>::setstate(ios_base::badbit);
        return *this;
      }

      return *this;
    }

    _UCXXEXPORT basic_istream<charT,traits>& unget()
    {
      sentry(*this, true);
      if (!basic_ios<charT,traits>::good())
      {
        basic_ios<charT,traits>::setstate(ios_base::failbit);
        return *this;
      }

      if (basic_ios<charT, traits>::mstreambuf == 0)
      {
        basic_ios<charT,traits>::setstate(ios_base::failbit);
        return *this;
      }

      if (basic_ios<charT, traits>::mstreambuf->sungetc() == traits::eof())
      {
        basic_ios<charT,traits>::setstate(ios_base::failbit);
      }

      return *this;
    }

    _UCXXEXPORT int sync()
    {
      sentry(*this, true);
      if (basic_ios<charT, traits>::mstreambuf == 0)
      {
        return -1;
      }

      if (basic_ios<charT, traits>::mstreambuf->pubsync() == -1)
      {
        basic_ios<charT,traits>::setstate(ios_base::badbit);
        return traits::eof();
      }

      return 0;
    }

    _UCXXEXPORT pos_type tellg()
    {
      if (basic_ios<charT,traits>::fail() !=false)
      {
        return pos_type(-1);
      }

      return basic_ios<charT, traits>::mstreambuf->pubseekoff(0, ios_base::cur, ios_base::in);
    }

    _UCXXEXPORT basic_istream<charT,traits>& seekg(pos_type pos)
    {
       if (basic_ios<charT,traits>::fail() !=true)
       {
        basic_ios<charT, traits>::mstreambuf->pubseekpos(pos);
       }

       return *this;
    }

    _UCXXEXPORT basic_istream<charT,traits>& seekg(off_type off, ios_base::seekdir dir)
    {
      if (basic_ios<charT,traits>::fail() !=true)
      {
        basic_ios<charT, traits>::mstreambuf->pubseekoff(off, dir);
      }

      return *this;
    }

  protected:
    _UCXXEXPORT basic_istream(const basic_istream<charT,traits> &): basic_ios<charT, traits>() { }
    _UCXXEXPORT basic_istream<charT,traits> & operator=(const basic_istream<charT,traits> &){ return *this; }
    streamsize count_last_ufmt_input;
  };

  template <class charT,class traits = char_traits<charT> > class _UCXXEXPORT basic_istream<charT,traits>::sentry
  {
    bool ok;
  public:
    explicit _UCXXEXPORT sentry(basic_istream<charT,traits>& os, bool noskipws = false){
      if (os.good() !=0)
      {
        // Prepare for output
      }

      // Flush any tied buffer

      if (os.tie() != 0)
      {
        os.tie()->flush();
      }

      if (!noskipws)
      {
        __skipws(os);
      }

      ok = true;
    }

    _UCXXEXPORT ~sentry() { }

    _UCXXEXPORT operator bool()
    {
      return ok;
    }
  };

  // Template implementations of basic_istream functions which may be partially specialized
  // For code reduction

  template <class charT, class traits>
  _UCXXEXPORT typename basic_istream<charT,traits>::int_type basic_istream<charT,traits>::get()
  {
    sentry(*this, true);
    int_type retval = basic_ios<charT, traits>::mstreambuf->sgetc();
    if (retval == traits::eof())
    {
      count_last_ufmt_input = 0;
      basic_ios<charT,traits>::setstate(ios_base::eofbit);
    }
    else
    {
      count_last_ufmt_input = 1;
      basic_ios<charT, traits>::mstreambuf->sbumpc();
    }

    return retval;
  }

  template <class charT, class traits>
  _UCXXEXPORT basic_istream<charT,traits>& basic_istream<charT,traits>::get(char_type& c)
  {
    sentry(*this, true);
    int_type retval = basic_ios<charT, traits>::mstreambuf->sgetc();
    if (retval == traits::eof())
    {
      count_last_ufmt_input = 0;
      basic_ios<charT,traits>::setstate(ios_base::eofbit);
      basic_ios<charT,traits>::setstate(ios_base::failbit);
    }
    else
    {
      count_last_ufmt_input = 1;
      c = traits::to_char_type(retval);
      basic_ios<charT, traits>::mstreambuf->sbumpc();
    }

    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(bool& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, bool>::readin(*this, n);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(short& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, short>::readin(*this, n);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(unsigned short& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, unsigned short>::readin(*this, n);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>& basic_istream<charT,traits>::operator>>(int& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, int>::readin(*this, n);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>& basic_istream<charT,traits>::operator>>(unsigned int& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, unsigned int>::readin(*this, n);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>& basic_istream<charT,traits>::operator>>(long int& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, long int>::readin(*this, n);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(unsigned long int& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, unsigned long int>::readin(*this, n);
    return *this;
  }

#ifdef CONFIG_HAVE_FLOAT
  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(float& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, float>::readin(*this, n);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(double& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, double>::readin(*this, n);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(long double& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, long double>::readin(*this, n);
    return *this;
  }
#endif

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(void *& n)
  {
    sentry(*this);
    __istream_readin<traits, charT, void*>::readin(*this, n);
    return *this;
  }

  template<class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    operator>>(basic_istream<charT,traits>& is, charT& c)
  {
    typename basic_istream<charT,traits>::sentry s(is);
    is.get(c);
    return is;
  }

  template<class traits> _UCXXEXPORT basic_istream<char,traits>&
    operator>>(basic_istream<char,traits>& is, unsigned char& c)
  {
    typename basic_istream<char,traits>::sentry s(is);
    char b;
    is.get(b);
    c = b;
    return is;
  }

  template<class traits> _UCXXEXPORT basic_istream<char,traits>&
    operator>>(basic_istream<char,traits>& is, signed char& c)
  {
    typename basic_istream<char,traits>::sentry s(is);
    is.get(c);
    return is;
  }

  template<class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    operator>>(basic_istream<charT,traits>& is, charT* c)
  {
    typename basic_istream<charT,traits>::sentry s(is);
    int n = is.width();
    if (n == 0)
    {
      n = __STRING_MAX_UNITS;
    }

    is.get(c, n);
    return is;

  }

  template<class traits> _UCXXEXPORT basic_istream<char,traits>&
    operator>>(basic_istream<char,traits>& is, unsigned char* c)
  {
    typename basic_istream<char,traits>::sentry s(is);
    int n = is.width();
    if (n == 0)
    {
      n = __STRING_MAX_UNITS;
    }

    is.get(c, n);
    return is;
  }

  template<class traits> _UCXXEXPORT basic_istream<char,traits>&
    operator>>(basic_istream<char,traits>& is, signed char* c)
  {
    typename basic_istream<char,traits>::sentry s(is);
    int n = is.width();
    if (n == 0)
    {
      n = __STRING_MAX_UNITS;
    }
    is.get(c, n);
    return is;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(basic_istream<charT,traits>& (*pf)(basic_istream<charT,traits>&))
  {
    sentry(*this);
    pf(*this);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    basic_istream<charT,traits>::operator>>(basic_ios<charT,traits>& (*pf)(basic_ios<charT,traits>&))
  {
    sentry(*this);
    pf(*this);
    return *this;
  }

  template <class charT, class traits> _UCXXEXPORT basic_istream<charT,traits>&
    ws(basic_istream<charT,traits>& is)
  {
    __skipws(is);
    return is;
  }

#ifdef __UCLIBCXX_EXPAND_ISTREAM_CHAR__
#ifndef __UCLIBCXX_COMPILE_ISTREAM__

  template <> _UCXXEXPORT istream & basic_istream<char, char_traits<char> >::get(char & c);
  template <> _UCXXEXPORT istream::int_type basic_istream<char, char_traits<char> >::get();

  template <> _UCXXEXPORT istream & istream::operator>>(bool &n);
  template <> _UCXXEXPORT istream & istream::operator>>(short &n);
  template <> _UCXXEXPORT istream & istream::operator>>(unsigned short &n);
  template <> _UCXXEXPORT istream & istream::operator>>(int &n);
  template <> _UCXXEXPORT istream & istream::operator>>(unsigned int &n);
  template <> _UCXXEXPORT istream & istream::operator>>(long unsigned &n);
  template <> _UCXXEXPORT istream & istream::operator>>(long int &n);
  template <> _UCXXEXPORT istream & istream::operator>>(void *& p);

#ifdef CONFIG_HAVE_FLOAT
  template <> _UCXXEXPORT istream & istream::operator>>(float &f);
  template <> _UCXXEXPORT istream & istream::operator>>(double &f);
  template <> _UCXXEXPORT istream & istream::operator>>(long double &f);
#endif

  template <> _UCXXEXPORT istream & operator>>(istream & is, char & c);

  template <> _UCXXEXPORT void __skipws(basic_istream<char,char_traits<char> >& is);

#endif
#endif

} // namespace
} // extern "C++"

#pragma GCC visibility pop

#endif