Merge pull request #2 from Nyae44/add-logic

add shortening url logic and db persistence

Author: Nyae44

cmd/webapp/db.sqlite3

@@ -1,24 +1,31 @@
 package main
 
 import (
+	"database/sql"
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/nyae44/url-shortener-app/internal/controllers"
+	"github.com/nyae44/url-shortener-app/internal/db"
 	"log"
 	"net/http"
-	"text/template"
 )
 
 func main() {
-	http.HandleFunc("/", ShowHomePage)
-	log.Fatal(http.ListenAndServe(":8080", nil))
-}
-
-func ShowHomePage(w http.ResponseWriter, r *http.Request) {
-	temp, err := template.ParseFiles("internal/views/index.html")
+	sqlite, err := sql.Open("sqlite3", "db.sqlite")
 	if err != nil {
-		http.Error(w, err.Error(), http.StatusInternalServerError)
-		return
+		log.Fatal(err)
 	}
-	if err = temp.Execute(w, nil); err != nil {
-		http.Error(w, err.Error(), http.StatusInternalServerError)
-		return
+	defer sqlite.Close()
+
+	if err := db.CreateTable(sqlite); err != nil {
+		log.Fatal(err)
 	}
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path == "/" {
+			controllers.ShowIndex(w, r)
+		} else {
+			controllers.Proxy(sqlite)(w, r)
+		}
+	})
+	http.HandleFunc("/shorten", controllers.Shorten(sqlite))
+	log.Fatal(http.ListenAndServe(":8080", nil))
 }

cmd/webapp/main.go

@@ -0,0 +1 @@
+package controllers

cmd/webapp/webapp

@@ -0,0 +1 @@
+package controllers

internal/controllers/index.go

@@ -0,0 +1 @@
+package db

internal/controllers/shorten.go

@@ -0,0 +1 @@
+package url