polserver/pol-core/bscript/impstr.h

/** @file
 *
 * @par History
 */

#ifndef H_BSCRIPT_IMPSTR_H
#define H_BSCRIPT_IMPSTR_H

#ifndef BSCRIPT_BOBJECT_H
#include "bobject.h"
#endif

#include <stack>
#include <string>
#include <string_view>


namespace Pol::Bscript
{
class SubString;

class String : public BObjectImp
{
  using base = BObjectImp;

public:
  enum class Tainted
  {
    YES,  // performs unicode sanitize should be done for every external value (assuming ISO8859)
    NO    // performs no unicode sanitize should only be used for internal usage
  };
  String() : BObjectImp( OTString ), value_( "" ) {}
  explicit String( const char* str, size_t nchars, Tainted san = Tainted::NO );
  explicit String( const char* str, Tainted san = Tainted::NO );
  explicit String( const std::string& str, Tainted san = Tainted::NO );
  explicit String( const std::string_view& str, Tainted san = Tainted::NO );
  explicit String( const std::wstring& str, Tainted san = Tainted::NO );
  explicit String( BObjectImp& objimp );
  String( const String& str ) : BObjectImp( OTString ), value_( str.value_ ) {}
  ~String() override = default;

private:
  explicit String( const std::string& str, std::string::size_type pos, std::string::size_type n );

public:
  static BObjectImp* unpack( std::istream& is );
  static BObjectImp* unpackWithLen( std::istream& is );
  std::string pack() const override;
  void packonto( std::string& str ) const override;
  static void packonto( std::string& str, const std::string& value );
  BObjectImp* copy() const override { return new String( *this ); }
  size_t sizeEstimate() const override;
  ContIterator* createIterator( BObject* pIterVal ) override;

  // FIXME: Possibly eliminate this later and have [ ] operator support?
  // Or stick to functions, overload them with other substring getting methods.
  String* StrStr( int begin, int len ) const;
  // FIXME: Possibly upgrade this later with overload functions in order to support different
  // trim methods, or add ELTrim, etc?
  String* ETrim( const char* CRSet, int type ) const;
  void EStrReplace( String* str1, String* str2 );
  void ESubStrReplace( String* replace_with, unsigned int index, unsigned int len );

  const char* data() const { return value_.c_str(); }
  const std::string& value() const { return value_; }
  size_t length() const;
  void toUpper();
  void toLower();

  // Static:  mf_SplitWords allows non-String types as input (uses objimp->getStringRep() as source)
  static ObjArray* getCharacters( const std::string& input, size_t limit );

  bool hasUTF8Characters() const;
  static bool hasUTF8Characters( const std::string& str );
  std::vector<unsigned short> toUTF16() const;
  static std::string fromUTF16( unsigned short code );
  static std::string fromUTF16( const unsigned short* code, size_t len, bool big_endian = false );
  static std::string fromUTF8( const char* code, size_t len );
  static std::vector<unsigned short> toUTF16( const std::string& text );
  static String* fromUCArray( ObjArray* array, bool break_first_null = true );

  String& operator=( const char* s )
  {
    value_ = s;
    return *this;
  }
  String& operator=( const String& str )
  {
    value_ = str.value_;
    return *this;
  }
  void copyvalue( const String& str ) { value_ = str.value_; }

private:
  void remove( const std::string& s );
  bool isTrue() const override { return !value_.empty(); }

public:
  BObjectImp* selfPlusObjImp( const BObjectImp& objimp ) const override;
  BObjectImp* selfPlusObj( const BObjectImp& objimp ) const override;
  BObjectImp* selfPlusObj( const BLong& objimp ) const override;
  BObjectImp* selfPlusObj( const Double& objimp ) const override;
  BObjectImp* selfPlusObj( const String& objimp ) const override;
  BObjectImp* selfPlusObj( const ObjArray& objimp ) const override;
  void selfPlusObjImp( BObjectImp& objimp, BObject& obj ) override;
  void selfPlusObj( BObjectImp& objimp, BObject& obj ) override;
  void selfPlusObj( BLong& objimp, BObject& obj ) override;
  void selfPlusObj( Double& objimp, BObject& obj ) override;
  void selfPlusObj( String& objimp, BObject& obj ) override;
  void selfPlusObj( ObjArray& objimp, BObject& obj ) override;

  BObjectImp* selfMinusObjImp( const BObjectImp& objimp ) const override;
  BObjectImp* selfMinusObj( const BObjectImp& objimp ) const override;
  BObjectImp* selfMinusObj( const BLong& objimp ) const override;
  BObjectImp* selfMinusObj( const Double& objimp ) const override;
  BObjectImp* selfMinusObj( const String& objimp ) const override;
  BObjectImp* selfMinusObj( const ObjArray& objimp ) const override;
  void selfMinusObjImp( BObjectImp& objimp, BObject& obj ) override;
  void selfMinusObj( BObjectImp& objimp, BObject& obj ) override;
  void selfMinusObj( BLong& objimp, BObject& obj ) override;
  void selfMinusObj( Double& objimp, BObject& obj ) override;
  void selfMinusObj( String& objimp, BObject& obj ) override;
  void selfMinusObj( ObjArray& objimp, BObject& obj ) override;

  BObjectRef OperSubscript( const BObject& obj ) override;
  BObjectRef OperMultiSubscript( std::stack<BObjectRef>& indices ) override;
  BObjectRef OperMultiSubscriptAssign( std::stack<BObjectRef>& indices,
                                       BObjectImp* target ) override;

  int find( int begin, const char* target ) const;
  unsigned int SafeCharAmt() const;

  BObjectImp* array_assign( BObjectImp* idx, BObjectImp* target, bool copy ) override;

  std::string getStringRep() const override { return value_; }
  std::string getFormattedStringRep() const override { return "\"" + value_ + "\""; }
  void printOn( std::ostream& ) const override;

  bool compare( const String& str ) const;
  bool compare( size_t pos1, size_t len1, const String& str ) const;
  bool compare( size_t pos1, size_t len1, const String& str, size_t pos2, size_t len2 ) const;

protected:
  bool operator==( const BObjectImp& objimp ) const override;
  bool operator<( const BObjectImp& objimp ) const override;

  BObjectImp* call_method( const char* methodname, Executor& ex ) override;
  BObjectImp* call_method_id( const int id, Executor& ex, bool forcebuiltin = false ) override;

private:
  size_t getBytePosition( std::string::const_iterator* itr, size_t codeindex ) const;

  std::string value_;
  friend class SubString;
};

class ConstString : public String
{
public:
  explicit ConstString( const std::string& str ) : String( str ) {}
};
}  // namespace Pol::Bscript

#endif