98. Convert String to Integer

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

int custom_atoi(const char *str) {
    // Your logic here
    int res=0, i=1;
    while(*str)
    {
        if( (*str >='0' && *str <= '9') || *str == '-' || *str == '+'){
            if( *str == '-') i = -1;
            else if(*str == '+') i = 1;
            else {
                res = res * 10 + *str - '0';  
            }
        }
        else return (res * i);
        str++;


    }
    return 0;
}

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;
}