#!/bin/bash

projectName="<project_name>"

getStatefulWidgetCode() {
    fileName=$1
    # echo $'import \'package:flutter/material.dart\';\r\nimport \'package:provider/provider.dart\';\r\nimport \'package:'${projectName}'/ProviderModels/'${fileName}$'ProviderModel/'${fileName}$'_pm.dart\';\r\n// Copy the line below and add as a child where you want to use '${fileName^}$' Widget\r\n// ChangeNotifierProvider(create:(context)=>'${fileName^}$'Pm(),child: '${fileName^}$'())\r\n\nclass '${fileName^}$' extends StatefulWidget {\r\n@override\r\n_'${fileName^}$'State createState() => _'${fileName^}$'State();\r\n}\r\n\r\nclass _'${fileName^}$'State extends State<'${fileName^}$'> {\r\n @override\r\n  Widget build(BuildContext context) {\r\n'${fileName^}'Pm '${fileName,}'Pm = Provider.of<'${fileName^}'Pm>(context);\r\nreturn Container();\r\n }\r\n}'
    echo $'import \'package:flutter/material.dart\';\r\nimport \'package:provider/provider.dart\';\r\nimport \'./'${fileName}$'_pm.dart\';\r\n\r\n// Copy the line below and add as a child where you want to use '${fileName^}$' Widget\r\n// ChangeNotifierProvider(create:(context)=>'${fileName^}$'Pm(),child: '${fileName^}$'())\r\n\r\nclass '${fileName^}$' extends StatefulWidget {\r\n@override\r\n_'${fileName^}$'State createState() => _'${fileName^}$'State();\r\n}\r\n\nclass _'${fileName^}$'State extends State<'${fileName^}$'> {\r\n @override\r\n  Widget build(BuildContext context) {\r\n'${fileName^}'Pm '${fileName,}'Pm = Provider.of<'${fileName^}$'Pm>(context);\r\nreturn Scaffold(appBar: AppBar(title: Text("'${fileName^}$'"), centerTitle: true));\r\n } \r\n }'
}

getProviderModelCode() {
    fileName=$1
    echo $'import \'package:flutter/material.dart\';\r\n\nclass '${fileName^}$'Pm extends ChangeNotifier {\r\n'${fileName^}$'Pm(){\r\n//This is the constructor\r\n}\r\n}'
}

getStatelessWidgetCode() {
    fileName=$1
    echo $'import \'package:flutter/material.dart\';\r\nimport \'package:provider/provider.dart\';\r\nimport \'./'${fileName}$'_pm.dart\';\r\n\r\n// Copy the line below and add as a child where you want to use '${fileName^}$' Widget\r\n// ChangeNotifierProvider(create:(context)=>'${fileName^}$'Pm(),child: '${fileName^}$'())\r\n\r\nclass '${fileName^}$' extends StatelessWidget {\r\n @override\r\n  Widget build(BuildContext context) {\r\n'${fileName^}'Pm '${fileName,}'Pm = Provider.of<'${fileName^}$'Pm>(context);\r\nreturn Scaffold(appBar: AppBar(title: Text("'${fileName^}$'"), centerTitle: true));\r\n } \r\n }'
}

getWidgetCode() {
    fileName=$1
    option=$2
    statelessCode=1
    statefulCode=2
    # If option == 1
    # getStatelessWidgetCode()
    # If option == 2
    # getStatelessWidgetCode()
    widgetCode=""
    if [ ${option} -eq ${statelessCode} ]; then
        # echo "Inside 1"
        widgetCode=$(getStatelessWidgetCode ${fileName})
    elif [ ${option} -eq ${statefulCode} ]; then
        # echo "Inside 2"
        widgetCode=$(getStatefulWidgetCode ${fileName})
    else
        widgetCode=$(getStatelessWidgetCode ${fileName})
    fi
    echo ${widgetCode}
}

askForStatefulOrStatelessWidget() {
    while true; do
        read -p $'Do you want to make it a stateful or stateless Widget? 1/2\n1. Stateless Widget (Default)\n2. Stateful Widget\n' option
        echo ''
        case $option in
        [1]*)
            echo 1
            break
            ;;
        [2]*)
            echo 2
            break
            ;;
        *)
            echo 1
            break
            ;;
        esac
    done
}

# Function for adding Widgets folder
addWidgetsFolder() {
    folderName="Widgets"
    if [ -d "${folderName}" ]; then
        echo ${folderName}$' folder already exists.\nEntering into the folder.'
    else
        mkdir "${folderName}"
    fi
    cd "${folderName}"
}
# Function for adding ProviderModels folder
addProviderModelsFolder() {
    folderName="ProviderModels"
    if [ -d "${folderName}" ]; then
        echo ${folderName}$' folder already exists.\nEntering into the folder.'
    else
        mkdir "${folderName}"
    fi
    cd "${folderName}"
}

addWidgetFile() {
    fileName=$1
    mkdir "${fileName}"
    moveInsideLocation "${fileName}"
    touch "${fileName}.dart"
    option=$(askForStatefulOrStatelessWidget)
    code=$(getWidgetCode ${fileName} ${option})
    echo -e $code >>"${fileName}.dart"
    echo "${fileName}.dart added"
}

addProviderFile() {
    fileName=$1
    touch "${fileName}_pm.dart"
    providerModelCode=$(getProviderModelCode ${fileName})
    echo -e $providerModelCode >>"${fileName}_pm.dart"
    cd ..
    cd ..
    echo "${fileName}_pm.dart added"
}

moveInsideLocation() {
    locaiton=$1
    cd ${locaiton}
}

moveInsideLib() {
    if [ ! -d "lib" ]; then
        echo $'No lib folder found.\nCreating lib folder'
        mkdir lib
    fi
    cd lib
}

askForProjetcLocation() {
    while true; do
        read -p $'Are you inside your Flutter project\'s root location ? y/N\n' yN
        echo ''
        case $yN in
        [Yy]*)
            projectName=${PWD##*/}
            moveInsideLib
            break
            ;;
        [Nn]*)
            # ask user for project root location
            read -p $'Please paste your Flutter project root location here\n' projectLocation
            moveInsideLocation ${projectLocation}
            moveInsideLib
            break
            ;;
        *)
            echo "Please answer yes or no.(y/N)"
            ;;
        esac
    done
}

moveIntoWidgets() {
    if [ ! -d "Widgets" ]; then
        echo $'No Widgets folder found.\nCreating Widgets folder'
        mkdir Widgets
    fi
    moveInsideLocation "Widgets"
}

# --------------------------------------------------------------- #
############ START OF SCRIPT ################
echo $'Follow the steps to add the provider model files\nBoth the Provider class and Widget class(Stateful Widget) will be automatically added to your project and linked with proper imports.'
############## ASK USER FOR PROJECT LOCATION ##########
askForProjetcLocation
######################################################

############# AFTER ENTERING INTO /lib FOLDER OF FLUTTER PROJECT ############
echo $'Name the WIDGET you are building (try using "lowerCamelCasing") \nEg - signIn, signUp, dahboard, mainPage, firstPage\n'
while true; do
    read -p "Name - " fileName
    fileName=${fileName//-/}
    fileName=${fileName//_/}
    nameLength=${#fileName}
    minLength=0
    if
        [ $nameLength -gt $minLength ]
    then
        echo '----------------'

        moveIntoWidgets

        addWidgetFile ${fileName}
        addProviderFile ${fileName}
        echo ''
        echo $fileName$' Widget and ProviderModel added and linked.'
        echo '----------------'
        read -p $'Do you want to add another WIDGET? y/N \n' yn
        case $yn in
        [Yy]*)
            # Nothing here as of now if user enters yes
            continue
            ;;
        [Nn]*)
            echo $'Make sure you have "provider:" already added under "dependencies:" section inside "pubspec.yaml" file.'
            echo "Exiting from program..."
            break
            ;;
        *)
            echo "Please answer yes or no.(y/N)"
            ;;
        esac
    else
        echo "Please enter a valid file name again."
    fi
done