Added ability to test for keys before extracting them, Added Generic Encoder/Decoder and Base64 Encoder/Decoder.

pjsonlib/include/pjson.h

@@ -26,114 +26,167 @@
 
 namespace ByteDance {
 //==[Interface]============================================================
-class pjson {
-  public:
-    enum jsonType : int64_t {
-      jsonNull,
-      jsonString,
-      jsonNumberInt,
-      jsonNumberFloat,
-      jsonBoolean,
-      jsonArray, //[ ] array
-      jsonMap, // { ... } map
+    class pjson {
+    public:
+        enum jsonType : int64_t {
+            jsonNull,
+            jsonString,
+            jsonNumberInt,
+            jsonNumberFloat,
+            jsonBoolean,
+            jsonArray, //[ ] array
+            jsonMap, // { ... } map
+        };
+        typedef std::vector<pjson*> PJSONARRAY;
+        //typedef std::unordered_map<std::string, pjson*> PJSONMAP;
+        typedef std::map<std::string, pjson*> PJSONMAP;
+
+        pjson(); // Default Constructor
+        ~pjson(); // Destructor
+        pjson(const pjson& aFrom); // Copy Constructor
+        pjson(pjson&& aFrom); // Move constructor
+        pjson& operator=(const pjson& aFrom); // Copy assignment
+        pjson& operator=(pjson&& a); // Move assignment
+
+        static pjson* CreateFromString(const std::string& aStr);
+        static pjson* CreateFromString(const char* aSrc, size_t a_iSize);
+
+        jsonType getType() const;
+        std::string toString(bool bPretty = false) const;
+        void copyFrom(const pjson& aFrom);
+
+        PJSONARRAY* getArray();
+        PJSONMAP* getMap();
+
+        float getFloat();
+        int getInt();
+        bool getBool();
+        std::string getString();
+
+        // Extracting from a Map
+        bool hasKey(const std::string& aKey);
+        bool hasKey(const char* aKey);
+
+        bool getIfExist(const std::string& aKey, float& a_rResult);
+        bool getIfExist(const std::string& aKey, int& a_rResult);
+        bool getIfExist(const std::string& aKey, bool& a_rResult);
+        bool getIfExist(const std::string& aKey, std::string& a_rResult);
+        bool getIfExist(const std::string& aKey, std::vector<std::string>& a_rResult);
+        bool getIfExist(const std::string& aKey, std::vector<int>& a_rResult);
+        bool getIfExist(const std::string& aKey, std::vector<float>& a_rResult);
+        bool getIfExist(const std::string& aKey, std::vector<bool>& a_rResult);
+
+        bool getIfExist(const char* aKey, float& a_rResult);
+        bool getIfExist(const char* aKey, int& a_rResult);
+        bool getIfExist(const char* aKey, bool& a_rResult);
+        bool getIfExist(const char* aKey, std::string& a_rResult);
+        bool getIfExist(const char* aKey, std::vector<std::string>& a_rResult);
+        bool getIfExist(const char* aKey, std::vector<int>& a_rResult);
+        bool getIfExist(const char* aKey, std::vector<float>& a_rResult);
+        bool getIfExist(const char* aKey, std::vector<bool>& a_rResult);
+
+        void reset();
+        void resetTo(jsonType aeType);
+
+        pjson& at(const std::string& aString);
+        pjson& at(const char* aSkey);
+        pjson& at(int index);
+
+        //Assignment overload
+        pjson& operator[] (const std::string& aString);
+        pjson& operator[] (const char* aSkey);
+        pjson& operator[] (int index);
+
+        pjson& operator=(const std::string& aString);
+        pjson& operator=(const char* aCString);
+        pjson& operator=(const int aInt);
+        pjson& operator=(const float aFloat);
+        pjson& operator=(const bool aBool);
+
+        pjson& operator=(const std::vector<std::string>& aValueArray);
+        pjson& operator=(const std::vector<char*>& aValueArray);
+        pjson& operator=(const std::vector<const char*>& aValueArray);
+        pjson& operator=(const std::vector<int>& aValueArray);
+        pjson& operator=(const std::vector<float>& aValueArray);
+        pjson& operator=(const std::vector<bool>& aValueArray);
+
+        pjson& operator+=(const std::string& aValue);
+        pjson& operator+=(const char* aValue);
+        pjson& operator+=(const int aValue);
+        pjson& operator+=(const float aValue);
+        pjson& operator+=(const bool aValue);
+
+        pjson& operator+=(const std::vector<std::string>& aValueArray);
+        pjson& operator+=(const std::vector<char*>& aValueArray);
+        pjson& operator+=(const std::vector<const char*>& aValueArray);
+        pjson& operator+=(const std::vector<int>& aValueArray);
+        pjson& operator+=(const std::vector<float>& aValueArray);
+        pjson& operator+=(const std::vector<bool>& aValueArray);
+
+        bool getArrayValues(size_t aFrom, size_t aTo, std::vector<std::string>& aDest);
+        bool getArrayValues(size_t aFrom, size_t aTo, std::vector<int>& aDest);
+        bool getArrayValues(size_t aFrom, size_t aTo, std::vector<float>& aDest);
+        bool getArrayValues(size_t aFrom, size_t aTo, std::vector<bool>& aDest);
+
+    /*
+     * Encodes a data buffer as a JSON-safe string by escaping special characters
+     * @param data Pointer to the data buffer
+     * @param length Length of the data buffer
+     * @return JSON-safe string representation
+     */
+    static std::string EncodeForJSON(const char* data, size_t length);
+
+    /*
+     * Encodes a data buffer as a Base64 string for JSON embedding
+     * @param data Pointer to the data buffer
+     * @param length Length of the data buffer
+     * @return Base64-encoded string
+     */
+    static std::string EncodeBase64ForJSON(const char* data, size_t length);
+
+    /*
+     * Decodes a JSON-safe string back to its original form
+     * @param jsonStr The JSON-encoded string
+     * @return The decoded string
+     */
+    static std::string DecodeFromJSON(const std::string& jsonStr);
+
+    /*
+     * Decodes a Base64 string back to its original binary data
+     * @param base64Str The Base64-encoded string
+     * @return The decoded binary data as a string
+     */
+    static std::string DecodeBase64FromJSON(const std::string& base64Str);
+
+    private:
+        std::string _toString(int a_iIndent) const;
+        void _resetIfneeded(jsonType aeType);
+        static bool _CreateFromString(const char* aSrc, size_t& a_iStart, size_t a_iEnd, pjson*& a_rResult);
+
+        static bool _ScanPastColon(const char* aSrc, size_t& a_iStart,const size_t a_iEnd);
+        static bool _ExtractString(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, std::string& aStrResult);
+        static bool _ScanString(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, pjson*& a_rStrResult);
+        static bool _ScanBool(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, pjson*& a_rBoolResult);
+        static bool _ScanToNext(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, char& a_rResult);
+        static bool _ScanNull(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, pjson*& a_rNUllResult);
+        static bool _ScanNumber(const char* aSrc, size_t& a_iStart, const size_t a_iEnd, pjson*& a_rNumResult);
+        static bool _ScanArray(const char* aSrc, size_t& a_iStart, const size_t a_iEnd, pjson*& a_rAResult);
+        static bool _ScanObject(const char* aSrc, size_t& a_iStart, const size_t a_iEnd, pjson*& a_rAResult);
+
+    private:
+
+        jsonType _eType = jsonType::jsonNull;
+        union {
+            void* _pValueRaw = nullptr;
+            PJSONMAP* _pValueMap;
+            PJSONARRAY* _pValueArray;
+            int* _pValueInt;
+            float* _pValueFloat;
+            bool* _pValueBool;
+            std::string* _pValueString;
+            /* data */
+        };
     };
-    typedef std::vector<pjson*> PJSONARRAY;
-    //typedef std::unordered_map<std::string, pjson*> PJSONMAP;
-    typedef std::map<std::string, pjson*> PJSONMAP;
-
-    pjson(); // Default Constructor
-    ~pjson(); // Destructor
-    pjson(const pjson& aFrom); // Copy Constructor
-    pjson(pjson&& aFrom); // Move constructor
-    pjson& operator=(const pjson& aFrom); // Copy assignment
-    pjson& operator=(pjson&& a); // Move assignment
-
-    static pjson* CreateFromString(const std::string& aStr);
-    static pjson* CreateFromString(const char* aSrc, size_t a_iSize);
-
-    jsonType getType() const;
-    std::string toString(bool bPretty = false) const;
-    void copyFrom(const pjson& aFrom);
-
-    PJSONARRAY* getArray();
-    PJSONMAP* getMap();
-    float getFloat();
-    int getInt();
-    bool getBool();
-    std::string getString();
-
-    void reset();
-    void resetTo(jsonType aeType);
-
-    pjson& at(const std::string& aString);
-    pjson& at(const char* aSkey);
-    pjson& at(int index);
-
-    //Assignment overload
-    pjson& operator[] (const std::string& aString);
-    pjson& operator[] (const char* aSkey);
-    pjson& operator[] (int index);
-
-    pjson& operator=(const std::string& aString);
-    pjson& operator=(const char* aCString);
-    pjson& operator=(const int aInt);
-    pjson& operator=(const float aFloat);
-    pjson& operator=(const bool aBool);
-
-    pjson& operator=(const std::vector<std::string>& aValueArray);
-    pjson& operator=(const std::vector<char*>& aValueArray);
-    pjson& operator=(const std::vector<const char*>& aValueArray);
-    pjson& operator=(const std::vector<int>& aValueArray);
-    pjson& operator=(const std::vector<float>& aValueArray);
-    pjson& operator=(const std::vector<bool>& aValueArray);
-
-    pjson& operator+=(const std::string& aValue);
-    pjson& operator+=(const char* aValue);
-    pjson& operator+=(const int aValue);
-    pjson& operator+=(const float aValue);
-    pjson& operator+=(const bool aValue);
-
-    pjson& operator+=(const std::vector<std::string>& aValueArray);
-    pjson& operator+=(const std::vector<char*>& aValueArray);
-    pjson& operator+=(const std::vector<const char*>& aValueArray);
-    pjson& operator+=(const std::vector<int>& aValueArray);
-    pjson& operator+=(const std::vector<float>& aValueArray);
-    pjson& operator+=(const std::vector<bool>& aValueArray);
-
-    bool getArrayValues(size_t aFrom, size_t aTo, std::vector<std::string>& aDest);
-    bool getArrayValues(size_t aFrom, size_t aTo, std::vector<int>& aDest);
-    bool getArrayValues(size_t aFrom, size_t aTo, std::vector<float>& aDest);
-    bool getArrayValues(size_t aFrom, size_t aTo, std::vector<bool>& aDest);
-
-  private:
-    std::string _toString(int a_iIndent) const;
-    void _resetIfneeded(jsonType aeType);
-    static bool _CreateFromString(const char* aSrc, size_t& a_iStart, size_t a_iEnd, pjson*& a_rResult);
-
-    static bool _ScanPastColon(const char* aSrc, size_t& a_iStart,const size_t a_iEnd);
-    static bool _ExtractString(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, std::string& aStrResult);
-    static bool _ScanString(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, pjson*& a_rStrResult);
-    static bool _ScanBool(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, pjson*& a_rBoolResult);
-    static bool _ScanToNext(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, char& a_rResult);
-    static bool _ScanNull(const char* aSrc, size_t& a_iStart,const size_t a_iEnd, pjson*& a_rNUllResult);
-    static bool _ScanNumber(const char* aSrc, size_t& a_iStart, const size_t a_iEnd, pjson*& a_rNumResult);
-    static bool _ScanArray(const char* aSrc, size_t& a_iStart, const size_t a_iEnd, pjson*& a_rAResult);
-    static bool _ScanObject(const char* aSrc, size_t& a_iStart, const size_t a_iEnd, pjson*& a_rAResult);
-
-  private:
-
-    jsonType _eType = jsonType::jsonNull;
-    union {
-      void* _pValueRaw = nullptr;
-      PJSONMAP* _pValueMap;
-      PJSONARRAY* _pValueArray;
-      int* _pValueInt;
-      float* _pValueFloat;
-      bool* _pValueBool;
-      std::string* _pValueString;
-      /* data */
-    };
-};
 //========================================================================
 };// end namespace ByteDance
-#endif /* !PRAVEENJSON_H */
+#endif /* !PRAVEENJSON_H */