Convert String to Integer

#include <stdio.h>
#include <stdint.h>

int custom_atoi(const char *str) {
    // Your logic here
    int n = 0;
    bool neg = false;
    int  count =0;
    while(*str != '\0')
    {
        if(*str == '-' && count == 0)
        {
            neg = true;
        }
        else if(*str == '+' && count ==0)
        {
            neg = false;
        }
        else
        {
            if(*str >= '0' && *str <= '9' && count == 0)
            {
                n = n*10 + (*str - '0');
            }
            else
            {
                count++;
            }
        }
        str++;
    }
    if(neg == true)
    {
        n*=-1;
    }
    return n;
}

int main() {
    char str[101];
    fgets(str, sizeof(str), stdin);

    // Remove newline
    uint8_t i = 0;
    while (str[i]) {
        if (str[i] == '\n') {
            str[i] = '\0';
            break;
        }
        i++;
    }

    printf("%d", custom_atoi(str));
    return 0;
}