#include "dataString.h"
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>

char *strAppendWithSize_unlimited(char *strOut, char *pData, int32_t Size)
{
	int32_t strOut_i = strGetSize(strOut);
	for (int32_t i = 0; i < Size; i++)
	{
		strOut[strOut_i + i] = pData[i];
	}
	strOut_i += Size;
	// add \0 to the end of strOut
	strOut[strOut_i] = 0;

	return strOut;
}

char *strCut(char *strOut, char *strIn, char startSign, char endSign)
{
	int32_t Size = strGetSize(strIn);
	int32_t isStart = 0;
	int32_t isEnd = 0;
	int32_t iOut = 0;
	for (int32_t i = 0; i < Size; i++)
	{
		if (!isStart && (strIn[i] == startSign))
		{
			isStart = 1;
			continue;
		}
		if (isStart && (strIn[i] == endSign))
		{
			isEnd = 1;
			break;
		}
		if (isStart)
		{
			strOut[iOut] = strIn[i];
			iOut++;
		}
	}
	/* add \0 */
	strOut[iOut] = 0;
	if (isStart && isEnd)
	{
		/* succeed */
		return strOut;
	}
	/* faild */
	return NULL;
}

char *strDeleteChar(char *strOut, char *strIn, char ch)
{
	int32_t iOut = 0;
	for (int32_t i = 0; i < strGetSize(strIn); i++)
	{
		if (ch == strIn[i])
		{
			continue;
		}
		strOut[iOut] = strIn[i];
		iOut++;
	}
	/* add \0 */
	strOut[iOut] = 0;
	return strOut;
}

char *strDeleteEnter(char *str)
{
	return strDeleteChar(str, str, '\n');
}

char *strAppendWithSize(char *strOut, char *pData, int32_t Size)
{
	int32_t strOut_i = strGetSize(strOut);
	for (int32_t i = 0; i < Size; i++)
	{
		strOut[strOut_i + i] = pData[i];
	}
	strOut_i += Size;
	// add \0 to the end of strOut
	strOut[strOut_i] = 0;

	return strOut;
}

int32_t strCountSign(char *strIn, char sign)
{
	int32_t count = 0;
	for (int32_t i = 0; i < strGetSize(strIn); i++)
	{
		if (sign == strIn[i])
		{
			count++;
		}
	}
	return count;
}

int32_t strGetTokenNum(char *strIn, char sign)
{
	return strCountSign(strIn, sign) + 1;
}

uint32_t strGetSize(char *pData)
{
	uint32_t Size = 0;
	if (NULL == pData)
	{
		return 0;
	}
	for (int32_t i = 0; pData[i] != 0; i++)
	{
		Size++;
	}
	return Size;
}

char *strAppend_unlimited(char *strOut, char *pData)
{
	uint32_t Size = 0;
	Size = strGetSize(pData);
	return strAppendWithSize_unlimited(strOut, pData, Size);
}

char *strGetLastToken(char *strOut, char *strIn, char sign)
{
	int32_t size = strGetSize(strIn);
	int32_t isGetSign = 0;
	int32_t buffSize = 0;
	for (int32_t i = size - 1; i > -1; i--)
	{
		if (strIn[i] != sign)
		{
			strOut[size - i - 1] = strIn[i];
			buffSize++;
		}
		if (strIn[i] == sign)
		{
			isGetSign = 1;
			break;
		}
	}
	int32_t i = 0;
	for (i = 0; i < buffSize / 2; i++)
	{
		char buff = strOut[i];
		strOut[i] = strOut[buffSize - i - 1];
		strOut[buffSize - i - 1] = buff;
	}
	strOut[buffSize] = 0;
	return strOut;
}

char *strPopToken(char *strOut, char *strIn, char sign)
{
	int32_t getSign = 0;
	int32_t iPoped = 0;
	int32_t iOut = 0;
	int32_t size = strGetSize(strIn);
	int32_t i = 0;
	for (i = 0; i < size; i++)
	{
		if (getSign)
		{
			strIn[iPoped++] = strIn[i];
			continue;
		}
		if (strIn[i] != sign)
		{
			strOut[iOut++] = strIn[i];
			continue;
		}
		if (strIn[i] == sign)
		{
			getSign = 1;
			continue;
		}
	}
	strOut[iOut] = 0;
	strIn[iPoped] = 0;
	return strOut;
}

char *strGetFirstToken(char *strOut, char *strIn, char sign)
{
	int32_t size = strGetSize(strIn);
	for (int32_t i = 0; i < size; i++)
	{
		if (strIn[i] != sign)
		{
			strOut[i] = strIn[i];
		}
		if (strIn[i] == sign)
		{
			break;
		}
	}
	return strOut;
}

int32_t strGetToken(char *string, char **argv, char sign)
{
	int32_t argc = 0;
	int32_t i = 0;
	//arg_i point32_t to the arg operated now
	int32_t arg_i = 0;
	// if not found ' ', then put chars from CMD to argv_tem
	int32_t char_i = 0;
	for (i = 0; (i < strGetSize(string)); i++)
	{
		if (string[i] != sign)
		{
			argv[arg_i][char_i] = string[i];
			char_i++;
		}
		if (string[i] == sign)
		{
			// write '\0' to the end of argv
			argv[arg_i][char_i] = 0;
			arg_i++;
			char_i = 0;
		}
		// write '\0' to the end of last argv
		argv[arg_i][char_i] = 0;
	}
	argc = arg_i + 1;
	return argc;
}

char *strAppend(char *strOut, char *pData)
{
	uint32_t Size = 0;
	Size = strGetSize(pData);
	return strAppendWithSize(strOut, pData, Size);
}

int32_t strIsStartWith(char *str, char *strStart)
{
	if (NULL == str || NULL == strStart)
	{
		/* input is null */
		return 0;
	}
	uint32_t size = strGetSize(strStart);
	uint32_t CMDName_get = 1;
	for (int32_t i = 0; i < size; i++)
	{
		if (str[i] != strStart[i])
		{
			CMDName_get = 0;
		}
	}
	return CMDName_get;
}

int32_t strEqu(char *str1, char *str2)
{
	if (str1 == NULL || str2 == NULL)
	{
		return 0;
	}
	int32_t size1 = strGetSize(str1);
	int32_t size2 = strGetSize(str2);
	if (size1 != size2)
	{
		return 0; // not equal
	}
	for (int32_t i = 0; i < size1; i++)
	{
		if (str1[i] != str2[i])
		{
			return 0; // not equal
		}
	}
	return 1;
}

char *strRemovePrefix(char *inputStr, char *prefix, char *outputStr)
{
	if (!strIsStartWith(inputStr, prefix))
	{
		return NULL;
	}

	for (int32_t i = strGetSize(prefix); i < strGetSize(inputStr); i++)
	{
		outputStr[i - strGetSize(prefix)] = inputStr[i];
	}
	return outputStr;
}

char *strClear(char *str)
{
	for (int32_t i = 0; i < sizeof(str); i++)
	{
		str[i] = 0;
	}
	return str;
}

int32_t strIsContain(char *str, char ch)
{
	for (int32_t i = 0; i < strGetSize(str); i++)
	{
		if (str[i] == ch)
		{
			return 1;
		}
	}
	return 0;
}

char *strCopy(char *strBuff, char *strIn)
{
	memcpy(strBuff, strIn, strGetSize(strIn));
	return strBuff;
}