Added code to process CLASS in CalendarEventObject.

[xestiacalendar/.git] / source / objects / calendarevent / CalendarEvent.cpp

diff --git a/source/objects/calendarevent/CalendarEvent.cpp b/source/objects/calendarevent/CalendarEvent.cpp
index d175eed..7cc99a4 100644 (file)
--- a/source/objects/calendarevent/CalendarEvent.cpp
+++ b/source/objects/calendarevent/CalendarEvent.cpp
@@ -8,6 +8,7 @@ CalendarObjectValidResult CalendarEventObject::ValidObject(){
        bool ValidEnd = false;
        bool ValidDateTimeStamp = false;
        bool ValidUniqueID = false;
+       bool ValidDateTimeStart = false;
        int SeekCount = 0;
        string PropertyName;
        
@@ -99,6 +100,41 @@ CalendarObjectValidResult CalendarEventObject::ValidObject(){
        
        SeekCount = 0;
        
+       // Look for DTSTART if nothing is set for METHOD..
+       
+       if (MethodData.size() == 0){
+       
+               for (vector<string>::iterator iter = ObjectName.begin();
+                       iter != ObjectName.end(); iter++){
+                       
+                       try{
+                               PropertyName = ObjectName[SeekCount].substr(0,7);
+                       }
+                       
+                       catch(const out_of_range& oor){
+                               continue;
+                       }
+               
+                       if (PropertyName == "DTSTART"){
+                       
+                               if (ValidDateTimeStart == false){
+                                       ValidDateTimeStart = true;
+                               } else {
+                                       return CALENDAROBJECTVALID_INVALIDFORMAT;
+                               }
+                               
+                       }
+                       
+                       SeekCount++;
+                       
+               }
+       
+       } else {
+               ValidDateTimeStart = true;
+       }
+       
+       SeekCount = 0;
+       
        // Look for END:VEVENT.
        
        for (vector<string>::iterator iter = ObjectName.begin();
@@ -124,6 +160,7 @@ CalendarObjectValidResult CalendarEventObject::ValidObject(){
        if (ValidBegin == true && 
                ValidEnd == true && 
                ValidDateTimeStamp == true &&
+               ValidDateTimeStart == true &&
                ValidUniqueID == true){
                
                return CALENDAROBJECTVALID_OK;
@@ -141,17 +178,115 @@ void CalendarEventObject::ProcessData(){
        // Process the data.
        
        multimap<string,string> DataReceived;
+       map<string,string> PropertyData;
+       string *PropertyNameData = nullptr;
        
        // Get the Date Time Stamp (DTSTAMP).
        
        DataReceived = ProcessTextVectors(&ObjectName, &ObjectData, false, "DTSTAMP");
        
-       // Process the data.
+       // Process the data from DTSTAMP.
        
        if (DataReceived.begin() != DataReceived.end()){
        
+               try {
+                       DateTimeStampTokens = DataReceived.begin()->first.substr(8);
+               }
+               
+               catch(const out_of_range &oor){
+                       // Do nothing as there is no data.
+               }               
+               
                DateTimeStampData = DataReceived.begin()->second;
                
        }
+
+       // Get the Unique ID (UID).
+       
+       DataReceived = ProcessTextVectors(&ObjectName, &ObjectData, false, "UID");
+       
+       // Process the data from UID.
+       
+       if (DataReceived.begin() != DataReceived.end()){
+       
+               try {
+                       UniqueIDTokens = DataReceived.begin()->first.substr(4);
+               }
+               
+               catch(const out_of_range &oor){
+                       // Do nothing as there is no data.
+               }               
+               
+               UniqueID = DataReceived.begin()->second;
+               
+       }
+       
+       // Get the Date Time Start value.
+       
+       DataReceived = ProcessTextVectors(&ObjectName, &ObjectData, false, "DTSTART");
+       
+       // Process the data from DTSTART.
+       
+       if (DataReceived.begin() != DataReceived.end()){
+       
+               bool TokenData = false;
+               string PropertyTokens;
+               
+               PropertyNameData = (string*)&DataReceived.begin()->first;
+               
+               PropertyData = SplitValues(*PropertyNameData);
+               
+               for(map<string,string>::iterator iter = PropertyData.begin();
+                       iter != PropertyData.end(); iter++){
+                       
+                       if (iter->first == "VALUE"){
+                               
+                               DateTimeStartDataValue = iter->second;
+                               
+                       } else if (iter->first == "TZID"){
+                               
+                               DateTimeStartDataTimeZoneID = iter->second;
+                               
+                       } else {
+                               
+                               if (TokenData == false){
+                                       TokenData = true;
+                               } else {
+                                       PropertyTokens += ";";
+                               }
+                               
+                               PropertyTokens += iter->first;
+                               PropertyTokens += "=";
+                               PropertyTokens += iter->second;
+                               
+                       }
+                               
+               }
+               
+               if (PropertyTokens.size() > 0){
+                       DateTimeStartDataTokens = PropertyTokens;
+               }
+               
+               DateTimeStartData = DataReceived.begin()->second;
+               
+       }
+       
+       // Get the Class value.
+       
+       DataReceived = ProcessTextVectors(&ObjectName, &ObjectData, false, "CLASS");
+       
+       if (DataReceived.begin() != DataReceived.end()){
+       
+               try {
+                       ClassDataTokens = DataReceived.begin()->first.substr(6);
+               }
+               
+               catch(const out_of_range &oor){
+                       // Do nothing as there is no data.
+               }               
+               
+               ClassData = DataReceived.begin()->second;
+               
+       }
        
 }