Convert String to Integer

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


int custom_atoi(const char *str){
    int i=0;
    int ans=0;
    int sign =1;
    if(str[i]=='+'){
        sign=1;
        i++;
    }
    else if(str[i]=='-'){
        sign=-1;
        i++;
    }   
    
    while(str[i]!='\0'){
        int c = str[i];
        
        if( c >='0' && c<='9'){
            ans=ans*10 + c-'0';
        }
        else
            break;

        i++;
    }



    return ans*sign;
}

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