diff --git a/assignment-2/Dockerfile b/assignment-2/Dockerfile
new file mode 100644
index 0000000..4c63ebc
--- /dev/null
+++ b/assignment-2/Dockerfile
@@ -0,0 +1,13 @@
+FROM golang:1.20 AS builder
+WORKDIR /app
+COPY . .
+
+FROM alpine:3.13
+WORKDIR /app
+COPY --from=builder /app/deploy/hacktiv-assignment-2 .
+COPY --from=builder /app/template /app/template/.
+RUN ls
+EXPOSE 8080
+
+ENV APP_NAME=hacktiv-assignment-2
+CMD ["./hacktiv-assignment-2"]
diff --git a/assignment-2/Makefile b/assignment-2/Makefile
new file mode 100644
index 0000000..ad1430d
--- /dev/null
+++ b/assignment-2/Makefile
@@ -0,0 +1,34 @@
+.PHONY: migration
+migration:
+ migrate create -ext sql -dir db/migrations/$(module) $(name)
+
+.PHONY: migrate
+migrate:
+ migrate -path db/migrations/$(module) -database "postgres://postgresuser:postgrespassword@127.0.0.1:5432/postgres?sslmode=disable&search_path=public" -verbose up
+
+.PHONY: rollback
+rollback:
+ migrate -path db/migrations/$(module) -database "postgres://postgresuser:postgrespassword@127.0.0.1:5432/postgres?sslmode=disable&search_path=public" -verbose down 1
+
+.PHONY: rollback-all
+rollback-all:
+ migrate -path db/migrations/$(module) -database "postgres://postgresuser:postgrespassword@127.0.0.1:5432/postgres?sslmode=disable&search_path=public" -verbose down -all
+
+.PHONY: force-migrate
+force-migrate:
+ migrate -path db/migrations/$(module) -database "postgres://postgresuser:postgrespassword@127.0.0.1:5432/postgres?sslmode=disable&search_path=public" -verbose force $(version)
+
+.PHONY: compile-server
+compile-server:
+ GO111MODULE=on CGO_ENABLED=0 GOOS=linux go build -a -installsuffix cgo -ldflags '-extldflags "-static"' -o deploy/hacktiv-assignment-2 main.go
+
+.PHONY: docker-build-server
+docker-build-server:
+ docker build --no-cache -t hacktiv-assignment-2:latest -f Dockerfile .
+
+.PHONY: run
+run:
+ docker-compose up
+
+.PHONY: compile-build-run
+compile-build-run: compile-server docker-build-server run
\ No newline at end of file
diff --git a/assignment-2/README.md b/assignment-2/README.md
new file mode 100644
index 0000000..960ff51
--- /dev/null
+++ b/assignment-2/README.md
@@ -0,0 +1,12 @@
+# Assignment 2 Hacktiv
+Simple login-logout page with gorilla session stored on postgres db
+
+# Dependencies
+- [Docker](https://docs.docker.com/engine/install/ubuntu/)
+- [Docker-compose](https://docs.docker.com/compose/install/)
+
+# How To Run
+1. Compile server to generate executable file `make compile-server`
+2. Build docker image `make docker-build-server`
+3. Run docker image `make run`
+4. Alternatively, you can use this shortcut to automatically run 3 above steps `make compile-build-run`
diff --git a/assignment-2/db/migrations/users/20221114032114_create_table_user.down.sql b/assignment-2/db/migrations/users/20221114032114_create_table_user.down.sql
new file mode 100644
index 0000000..1dac82b
--- /dev/null
+++ b/assignment-2/db/migrations/users/20221114032114_create_table_user.down.sql
@@ -0,0 +1,5 @@
+BEGIN;
+
+DROP TABLE users;
+
+COMMIT;
\ No newline at end of file
diff --git a/assignment-2/db/migrations/users/20221114032114_create_table_user.up.sql b/assignment-2/db/migrations/users/20221114032114_create_table_user.up.sql
new file mode 100644
index 0000000..a08b7e3
--- /dev/null
+++ b/assignment-2/db/migrations/users/20221114032114_create_table_user.up.sql
@@ -0,0 +1,11 @@
+BEGIN;
+
+CREATE TABLE users (
+ id serial not null,
+ username varchar(50) unique not null,
+ first_name varchar(200) not null,
+ last_name varchar(200) not null,
+ password varchar(120) not null
+);
+
+COMMIT;
\ No newline at end of file
diff --git a/assignment-2/deploy/hacktiv-assignment-2 b/assignment-2/deploy/hacktiv-assignment-2
new file mode 100755
index 0000000..830f74f
Binary files /dev/null and b/assignment-2/deploy/hacktiv-assignment-2 differ
diff --git a/assignment-2/docker-compose.yaml b/assignment-2/docker-compose.yaml
new file mode 100644
index 0000000..28d99db
--- /dev/null
+++ b/assignment-2/docker-compose.yaml
@@ -0,0 +1,24 @@
+version: '3'
+
+services:
+ postgres:
+ image: postgres:13.2-alpine
+ environment:
+ - POSTGRES_USER=postgresuser
+ - POSTGRES_PASSWORD=postgrespassword
+ - POSTGRES_DB=postgres
+ ports:
+ - 5432:5432
+ networks:
+ - assignment-2
+ server:
+ image: hacktiv-assignment-2:latest
+ ports:
+ - 8080:8080
+ depends_on:
+ - postgres
+ networks:
+ - assignment-2
+
+networks:
+ assignment-2:
\ No newline at end of file
diff --git a/assignment-2/driver/pgx.go b/assignment-2/driver/pgx.go
new file mode 100644
index 0000000..7d958a8
--- /dev/null
+++ b/assignment-2/driver/pgx.go
@@ -0,0 +1,11 @@
+package driver
+
+import (
+ "context"
+
+ "github.com/jackc/pgx/v4"
+)
+
+func NewPostgresConn(ctx context.Context) (*pgx.Conn, error) {
+ return pgx.Connect(ctx, POSTGRES_URL)
+}
diff --git a/assignment-2/driver/renderer.go b/assignment-2/driver/renderer.go
new file mode 100644
index 0000000..5e790d8
--- /dev/null
+++ b/assignment-2/driver/renderer.go
@@ -0,0 +1,41 @@
+package driver
+
+import (
+ "html/template"
+ "io"
+
+ "github.com/labstack/echo"
+)
+
+type Renderer struct {
+ template *template.Template
+ debug bool
+ location string
+}
+
+func NewRenderer(location string, debug bool) *Renderer {
+ tpl := new(Renderer)
+ tpl.location = location
+ tpl.debug = debug
+
+ tpl.ReloadTemplates()
+
+ return tpl
+}
+
+func (t *Renderer) ReloadTemplates() {
+ t.template = template.Must(template.ParseGlob(t.location))
+}
+
+func (t *Renderer) Render(
+ w io.Writer,
+ name string,
+ data interface{},
+ c echo.Context,
+) error {
+ if t.debug {
+ t.ReloadTemplates()
+ }
+
+ return t.template.ExecuteTemplate(w, name, data)
+}
diff --git a/assignment-2/driver/session_store.go b/assignment-2/driver/session_store.go
new file mode 100644
index 0000000..3f09b62
--- /dev/null
+++ b/assignment-2/driver/session_store.go
@@ -0,0 +1,36 @@
+package driver
+
+import (
+ "log"
+
+ "github.com/antonlindstrom/pgstore"
+ "github.com/gorilla/sessions"
+)
+
+const (
+ SESSION_ID = "test-id"
+ POSTGRES_URL = "postgres://postgresuser:postgrespassword@postgres:5432/postgres?sslmode=disable"
+)
+
+var (
+ SESSION_AUTH_KEY = []byte("my-auth-key-very-secret")
+ SESSION_ENCRYPTION_KEY = []byte("my-encryption-key-very-secret123")
+)
+
+func NewPostgresStore() *pgstore.PGStore {
+ store, err := pgstore.NewPGStore(POSTGRES_URL, SESSION_AUTH_KEY, SESSION_ENCRYPTION_KEY)
+ if err != nil {
+ log.Fatalln("ERROR", err)
+ }
+
+ return store
+}
+
+func NewCookieStore() *sessions.CookieStore {
+ store := sessions.NewCookieStore(SESSION_AUTH_KEY, SESSION_ENCRYPTION_KEY)
+ store.Options.Path = "/"
+ store.Options.MaxAge = 86400 * 7
+ store.Options.HttpOnly = true
+
+ return store
+}
diff --git a/assignment-2/entity/user.go b/assignment-2/entity/user.go
new file mode 100644
index 0000000..4213de3
--- /dev/null
+++ b/assignment-2/entity/user.go
@@ -0,0 +1,9 @@
+package entity
+
+type User struct {
+ ID int `json:"id"`
+ Username string `json:"username"`
+ FirstName string `json:"first_name"`
+ LastName string `json:"last_name"`
+ Password string `json:"password"`
+}
diff --git a/assignment-2/handler/common.go b/assignment-2/handler/common.go
new file mode 100644
index 0000000..ce37eb2
--- /dev/null
+++ b/assignment-2/handler/common.go
@@ -0,0 +1,21 @@
+package handler
+
+import (
+ "net/http"
+
+ "github.com/antonlindstrom/pgstore"
+ "github.com/ibrahimker/golang-intermediate/assignment-2/driver"
+ "github.com/labstack/echo"
+)
+
+func storeSessionHelper(c echo.Context, pgs *pgstore.PGStore, username string) error {
+ session, err := pgs.Get(c.Request(), driver.SESSION_ID)
+ if err != nil {
+ return c.String(http.StatusInternalServerError, err.Error())
+ }
+ session.Values["username"] = username
+ if err = session.Save(c.Request(), c.Response()); err != nil {
+ return c.String(http.StatusInternalServerError, err.Error())
+ }
+ return nil
+}
diff --git a/assignment-2/handler/login_handler.go b/assignment-2/handler/login_handler.go
new file mode 100644
index 0000000..ecb80e1
--- /dev/null
+++ b/assignment-2/handler/login_handler.go
@@ -0,0 +1,87 @@
+package handler
+
+import (
+ "net/http"
+
+ "github.com/antonlindstrom/pgstore"
+ "github.com/gorilla/sessions"
+ "github.com/ibrahimker/golang-intermediate/assignment-2/driver"
+ "github.com/ibrahimker/golang-intermediate/assignment-2/entity"
+ "github.com/ibrahimker/golang-intermediate/assignment-2/repository"
+ "github.com/labstack/echo"
+ log "github.com/sirupsen/logrus"
+)
+
+type HTMLTemplate struct {
+ User entity.User
+ Err string
+}
+
+type Login struct {
+ cs *sessions.CookieStore
+ pgStore *pgstore.PGStore
+ ur *repository.User
+}
+
+func NewLoginHandler(cs *sessions.CookieStore, pgs *pgstore.PGStore, ur *repository.User) *Login {
+ return &Login{
+ cs: cs,
+ pgStore: pgs,
+ ur: ur,
+ }
+}
+
+func (l *Login) LoginHandler(c echo.Context) error {
+ log.Info("Start Login Handler")
+ // authenticate in db
+ user, err := l.ur.GetByUsernamePassword(c.Request().Context(), c.FormValue("username"), c.FormValue("password"))
+ if err != nil {
+ log.Warn(err)
+ return c.Render(http.StatusOK, "login.html", HTMLTemplate{
+ User: entity.User{},
+ Err: "Error when login",
+ })
+ }
+ log.Info("Successfully authenticate user", user)
+
+ // store username in session
+ if err = storeSessionHelper(c, l.pgStore, user.Username); err != nil {
+ log.Warn("error when store session")
+ return c.Render(http.StatusOK, "login.html", HTMLTemplate{
+ User: entity.User{},
+ Err: "Error when store session",
+ })
+ }
+
+ return c.Redirect(http.StatusTemporaryRedirect, "/home")
+}
+
+func (l *Login) HomeHandler(c echo.Context) error {
+ log.Info("Start Home Handler")
+
+ session, err := l.pgStore.Get(c.Request(), driver.SESSION_ID)
+ if err != nil {
+ log.WithError(err).Warn("error when retrieve session")
+ return c.Redirect(http.StatusTemporaryRedirect, "/login")
+ }
+
+ if len(session.Values) == 0 {
+ log.Info("no session values available")
+ return c.Redirect(http.StatusTemporaryRedirect, "/login")
+ }
+
+ log.Info("Session exists, render html")
+ return c.Render(http.StatusOK, "home.html", HTMLTemplate{
+ User: entity.User{Username: session.Values["username"].(string)},
+ Err: "",
+ })
+}
+
+func (l *Login) LogoutHandler(c echo.Context) error {
+ log.Info("Start Logout Handler")
+ session, _ := l.pgStore.Get(c.Request(), driver.SESSION_ID)
+ session.Options.MaxAge = -1
+ session.Save(c.Request(), c.Response())
+
+ return c.Redirect(http.StatusTemporaryRedirect, "/home")
+}
diff --git a/assignment-2/handler/register_handler.go b/assignment-2/handler/register_handler.go
new file mode 100644
index 0000000..43a3eb0
--- /dev/null
+++ b/assignment-2/handler/register_handler.go
@@ -0,0 +1,51 @@
+package handler
+
+import (
+ "net/http"
+
+ "github.com/antonlindstrom/pgstore"
+ "github.com/gorilla/sessions"
+ "github.com/ibrahimker/golang-intermediate/assignment-2/entity"
+ "github.com/ibrahimker/golang-intermediate/assignment-2/repository"
+ "github.com/labstack/echo"
+ log "github.com/sirupsen/logrus"
+)
+
+type Register struct {
+ cs *sessions.CookieStore
+ pgs *pgstore.PGStore
+ ur *repository.User
+}
+
+func NewRegisterHandler(cs *sessions.CookieStore, pgs *pgstore.PGStore, ur *repository.User) *Register {
+ return &Register{
+ cs: cs,
+ pgs: pgs,
+ ur: ur,
+ }
+}
+
+func (r *Register) RegisterHandler(c echo.Context) error {
+ data := "Hello from /html"
+ u := &entity.User{
+ Username: c.FormValue("username"),
+ FirstName: c.FormValue("firstname"),
+ LastName: c.FormValue("lastname"),
+ Password: c.FormValue("password"),
+ }
+ log.Info("user", u)
+ if u.Username == "" || u.Password == "" {
+ log.Warn("username password cannot be empty")
+ return c.Redirect(http.StatusTemporaryRedirect, "/home")
+ }
+ if err := r.ur.Insert(c.Request().Context(), u); err != nil {
+ return c.Render(http.StatusOK, "login.html", data)
+ }
+
+ // store username in session
+ if err := storeSessionHelper(c, r.pgs, u.Username); err != nil {
+ return c.String(http.StatusInternalServerError, err.Error())
+ }
+
+ return c.Redirect(http.StatusTemporaryRedirect, "/home")
+}
diff --git a/assignment-2/main.go b/assignment-2/main.go
new file mode 100644
index 0000000..a1afdce
--- /dev/null
+++ b/assignment-2/main.go
@@ -0,0 +1,44 @@
+package main
+
+import (
+ "context"
+ "log"
+ "net/http"
+
+ "github.com/ibrahimker/golang-intermediate/assignment-2/driver"
+ "github.com/ibrahimker/golang-intermediate/assignment-2/handler"
+ "github.com/ibrahimker/golang-intermediate/assignment-2/repository"
+ "github.com/labstack/echo"
+)
+
+func main() {
+ e := echo.New()
+ e.Renderer = driver.NewRenderer("template/*.html", true)
+ ctx := context.Background()
+ store := driver.NewPostgresStore()
+ pgPool, err := driver.NewPostgresConn(ctx)
+ if err != nil {
+ log.Fatal(err)
+ }
+
+ userRepo := repository.NewUser(pgPool)
+
+ loginHandler := handler.NewLoginHandler(nil, store, userRepo)
+ registerHandler := handler.NewRegisterHandler(nil, store, userRepo)
+
+ e.GET("/", func(ctx echo.Context) error {
+ return ctx.Redirect(http.StatusTemporaryRedirect, "/home")
+ })
+
+ e.GET("/login", loginHandler.LoginHandler)
+ e.POST("/login", loginHandler.LoginHandler)
+
+ e.GET("/home", loginHandler.HomeHandler)
+ e.POST("/home", loginHandler.HomeHandler)
+
+ e.POST("/register", registerHandler.RegisterHandler)
+
+ e.POST("/logout", loginHandler.LogoutHandler)
+
+ e.Start(":8080")
+}
diff --git a/assignment-2/repository/common.go b/assignment-2/repository/common.go
new file mode 100644
index 0000000..26484ce
--- /dev/null
+++ b/assignment-2/repository/common.go
@@ -0,0 +1,18 @@
+package repository
+
+import (
+ "context"
+
+ "github.com/jackc/pgconn"
+ "github.com/jackc/pgx/v4"
+)
+
+// PgxPoolIface defines a little interface for pgxpool functionality.
+// Since in the real implementation we can use pgxpool.Pool,
+// this interface exists mostly for testing purpose.
+type PgxPoolIface interface {
+ Exec(ctx context.Context, sql string, arguments ...interface{}) (pgconn.CommandTag, error)
+ Query(ctx context.Context, sql string, args ...interface{}) (pgx.Rows, error)
+ QueryRow(ctx context.Context, sql string, args ...interface{}) pgx.Row
+ Ping(ctx context.Context) error
+}
diff --git a/assignment-2/repository/users.go b/assignment-2/repository/users.go
new file mode 100644
index 0000000..3d60d7a
--- /dev/null
+++ b/assignment-2/repository/users.go
@@ -0,0 +1,59 @@
+package repository
+
+import (
+ "context"
+
+ "github.com/ibrahimker/golang-intermediate/assignment-2/entity"
+ log "github.com/sirupsen/logrus"
+)
+
+// User is responsible to connect user entity with users table in PostgreSQL.
+type User struct {
+ pool PgxPoolIface
+}
+
+// NewUser creates an instance of User.
+func NewUser(pool PgxPoolIface) *User {
+ return &User{pool: pool}
+}
+
+// Insert inserts the user into the users table and return the user id.
+func (u *User) Insert(ctx context.Context, user *entity.User) error {
+ query := "INSERT INTO " +
+ "users (username, first_name, last_name, password) " +
+ "VALUES ($1, $2, $3, $4) " +
+ "RETURNING id"
+
+ row := u.pool.QueryRow(ctx, query,
+ user.Username,
+ user.FirstName,
+ user.LastName,
+ user.Password,
+ )
+
+ if err := row.Scan(&user.ID); err != nil {
+ log.Warn("Error when scan rows")
+ return err
+ }
+
+ return nil
+}
+
+// GetByUsernamePassword gets a user from PostgreSQL.
+// If there isn't any data, it returns error.
+func (u *User) GetByUsernamePassword(ctx context.Context, username, password string) (*entity.User, error) {
+ log.Info("Start GetByUsernamePassword", username, password)
+ query := "SELECT id,username,first_name,last_name,password " +
+ "FROM users WHERE username = $1 AND password = $2"
+
+ row := u.pool.QueryRow(ctx, query, username, password)
+
+ user := entity.User{}
+ err := row.Scan(&user.ID, &user.Username, &user.FirstName, &user.LastName, &user.Password)
+ if err != nil {
+ log.WithError(err).Warn("Error when scan rows")
+ return nil, err
+ }
+ log.Infof("Success retrieve user %+v\n", user)
+ return &user, nil
+}
diff --git a/assignment-2/template/home.html b/assignment-2/template/home.html
new file mode 100644
index 0000000..6cc869a
--- /dev/null
+++ b/assignment-2/template/home.html
@@ -0,0 +1,18 @@
+
+
+ Assignment 2 - Home
+
+
+
+
+
+
+
Halo, {{.User.Username}}
+
...
+
+
+