initial commit

This commit is contained in:
Daniel Hjartland
2024-11-26 18:08:49 +01:00
commit 75be79c6cf
26 changed files with 1885 additions and 0 deletions

12
.env.example Normal file
View File

@@ -0,0 +1,12 @@
DB_PATH="./tubely.db"
JWT_SECRET="JKFNDKAJSDKFASFNJWIROIOTNKNFDSKNFD"
PLATFORM="dev"
FILEPATH_ROOT="./app"
ASSETS_ROOT="./assets"
S3_BUCKET="tubely-123456789"
S3_REGION="us-east-2"
S3_CF_DISTRO="TEST"
PORT="8091"
# aws credentials should be set in ~/.aws/credentials
# using the `aws configure` command, the SDK will automatically
# read them from there

5
.gitignore vendored Normal file
View File

@@ -0,0 +1,5 @@
out
*.db
.env
assets/
samples/

65
README.md Normal file
View File

@@ -0,0 +1,65 @@
# learn-file-storage-s3-golang-starter (Tubely)
This repo contains the starter code for the Tubely application - the #1 tool for engagement bait - for the "Learn File Servers and CDNs with S3 and CloudFront" [course](https://www.boot.dev/courses/learn-file-servers-s3-cloudfront-golang) on [boot.dev](https://www.boot.dev)
## Quickstart
*This is to be used as a *reference\* in case you need it, you should follow the instructions in the course rather than trying to do everything here.
## 1. Install dependencies
- [Go](https://golang.org/doc/install)
- `go mod download` to download all dependencies
- [FFMPEG](https://ffmpeg.org/download.html) - both `ffmpeg` and `ffprobe` are required to be in your `PATH`.
```bash
# linux
sudo apt update
sudo apt install ffmpeg
# mac
brew update
brew install ffmpeg
```
- [SQLite 3](https://www.sqlite.org/download.html) only required for you to manually inspect the database.
```bash
# linux
sudo apt update
sudo apt install sqlite3
# mac
brew update
brew install sqlite3
```
- [AWS CLI](https://docs.aws.amazon.com/cli/latest/userguide/getting-started-install.html)
## 2. Download sample images and videos
```bash
./samplesdownload.sh
# samples/ dir will be created
# with sample images and videos
```
## 3. Configure environment variables
Copy the `.env.example` file to `.env` and fill in the values.
```bash
cp .env.example .env
```
You'll need to update values in the `.env` file to match your configuration, but _you won't need to do anything here until the course tells you to_.
## 3. Run the server
```bash
go run .
```
- You should see a new database file `tubely.db` created in the root directory.
- You should see a new `assets` directory created in the root directory, this is where the images will be stored.
- You should see a link in your console to open the local web page.

301
app/app.js Normal file
View File

@@ -0,0 +1,301 @@
document.addEventListener('DOMContentLoaded', async () => {
const token = localStorage.getItem('token');
if (token) {
document.getElementById('auth-section').style.display = 'none';
document.getElementById('video-section').style.display = 'block';
await getVideos();
} else {
document.getElementById('auth-section').style.display = 'block';
document.getElementById('video-section').style.display = 'none';
}
});
document.getElementById('video-draft-form').addEventListener('submit', async (event) => {
event.preventDefault();
await createVideoDraft();
});
document.getElementById('login-form').addEventListener('submit', async (event) => {
event.preventDefault();
await login();
});
async function createVideoDraft() {
const title = document.getElementById('video-title').value;
const description = document.getElementById('video-description').value;
try {
const res = await fetch('/api/videos', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
Authorization: `Bearer ${localStorage.getItem('token')}`,
},
body: JSON.stringify({ title, description }),
});
const data = await res.json();
if (!res.ok) {
throw new Error(`Failed to create video draft: ${data.error}`);
}
const videoID = data.id;
if (videoID) {
await getVideos();
await videoStateHandler(videoID);
}
} catch (error) {
alert(`Error: ${error.message}`);
}
}
async function login() {
const email = document.getElementById('email').value;
const password = document.getElementById('password').value;
try {
const res = await fetch('/api/login', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
body: JSON.stringify({ email, password }),
});
const data = await res.json();
if (!res.ok) {
throw new Error(`Failed to login: ${data.error}`);
}
if (data.token) {
localStorage.setItem('token', data.token);
document.getElementById('auth-section').style.display = 'none';
document.getElementById('video-section').style.display = 'block';
await getVideos();
} else {
alert('Login failed. Please check your credentials.');
}
} catch (error) {
alert(`Error: ${error.message}`);
}
}
async function signup() {
const email = document.getElementById('email').value;
const password = document.getElementById('password').value;
try {
const res = await fetch('/api/users', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
body: JSON.stringify({ email, password }),
});
if (!res.ok) {
const data = await res.json();
throw new Error(`Failed to create user: ${data.error}`);
}
console.log('User created!');
await login();
} catch (error) {
alert(`Error: ${error.message}`);
}
}
function logout() {
localStorage.removeItem('token');
document.getElementById('auth-section').style.display = 'block';
document.getElementById('video-section').style.display = 'none';
}
function setUploadButtonState(uploading, selector) {
const uploadBtn = document.getElementById(selector);
if (uploading) {
uploadBtn.textContent = 'Uploading...';
uploadBtn.disabled = true;
return;
}
uploadBtn.textContent = 'Upload';
uploadBtn.disabled = false;
}
async function uploadThumbnail(videoID) {
const thumbnailFile = document.getElementById('thumbnail').files[0];
if (!thumbnailFile) return;
const formData = new FormData();
formData.append('thumbnail', thumbnailFile);
uploadBtnSelector = 'upload-thumbnail-btn';
setUploadButtonState(true, uploadBtnSelector);
try {
const res = await fetch(`/api/thumbnail_upload/${videoID}`, {
method: 'POST',
headers: {
Authorization: `Bearer ${localStorage.getItem('token')}`,
},
body: formData,
});
if (!res.ok) {
const data = await res.json();
throw new Error(`Failed to upload thumbnail. Error: ${data.error}`);
}
await res.json();
console.log('Thumbnail uploaded!');
await getVideo(videoID);
} catch (error) {
alert(`Error: ${error.message}`);
}
setUploadButtonState(false, uploadBtnSelector);
}
async function uploadVideoFile(videoID) {
const videoFile = document.getElementById('video-file').files[0];
if (!videoFile) return;
const formData = new FormData();
formData.append('video', videoFile);
uploadBtnSelector = 'upload-video-btn';
setUploadButtonState(true, uploadBtnSelector);
try {
const res = await fetch(`/api/video_upload/${videoID}`, {
method: 'POST',
headers: {
Authorization: `Bearer ${localStorage.getItem('token')}`,
},
body: formData,
});
if (!res.ok) {
const data = await res.json();
throw new Error(`Failed to upload video file. Error: ${data.error}`);
}
console.log('Video uploaded!');
await getVideo(videoID);
} catch (error) {
alert(`Error: ${error.message}`);
}
setUploadButtonState(false, uploadBtnSelector);
}
const videoStateHandler = createVideoStateHandler();
async function getVideos() {
try {
const res = await fetch('/api/videos', {
method: 'GET',
headers: {
Authorization: `Bearer ${localStorage.getItem('token')}`,
},
});
if (!res.ok) {
const data = await res.json();
throw new Error(`Failed to get videos. Error: ${data.error}`);
}
const videos = await res.json();
const videoList = document.getElementById('video-list');
videoList.innerHTML = '';
for (const video of videos) {
const listItem = document.createElement('li');
listItem.textContent = video.title;
listItem.onclick = () => videoStateHandler(video.id);
videoList.appendChild(listItem);
}
} catch (error) {
alert(`Error: ${error.message}`);
}
}
function createVideoStateHandler() {
let currentVideoID = null;
return async function handleVideoClick(videoID) {
if (currentVideoID !== videoID) {
currentVideoID = videoID;
// Reset file input values
document.getElementById('thumbnail').value = '';
document.getElementById('video-file').value = '';
await getVideo(videoID);
}
};
}
async function getVideo(videoID) {
try {
const res = await fetch(`/api/videos/${videoID}`, {
method: 'GET',
headers: {
Authorization: `Bearer ${localStorage.getItem('token')}`,
},
});
if (!res.ok) {
throw new Error('Failed to get video.');
}
const video = await res.json();
viewVideo(video);
} catch (error) {
alert(`Error: ${error.message}`);
}
}
let currentVideo = null;
function viewVideo(video) {
currentVideo = video;
document.getElementById('video-display').style.display = 'block';
document.getElementById('video-title-display').textContent = video.title;
document.getElementById('video-description-display').textContent = video.description;
const thumbnailImg = document.getElementById('thumbnail-image');
if (!video.thumbnail_url) {
thumbnailImg.style.display = 'none';
} else {
thumbnailImg.style.display = 'block';
thumbnailImg.src = video.thumbnail_url;
}
const videoPlayer = document.getElementById('video-player');
if (videoPlayer) {
if (!video.video_url) {
videoPlayer.style.display = 'none';
} else {
videoPlayer.style.display = 'block';
videoPlayer.src = video.video_url;
videoPlayer.load();
}
}
}
async function deleteVideo() {
if (!currentVideo) {
alert('No video selected for deletion.');
return;
}
try {
const res = await fetch(`/api/videos/${currentVideo.id}`, {
method: 'DELETE',
headers: {
Authorization: `Bearer ${localStorage.getItem('token')}`,
},
});
if (!res.ok) {
throw new Error('Failed to delete video.');
}
alert('Video deleted successfully.');
document.getElementById('video-display').style.display = 'none';
await getVideos();
} catch (error) {
alert(`Error: ${error.message}`);
}
}

105
app/index.html Normal file
View File

@@ -0,0 +1,105 @@
<!doctype html>
<html lang="en">
<head>
<meta charset="UTF-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>Tubely</title>
<link rel="stylesheet" href="styles.css" />
<script src="app.js" defer></script>
</head>
<body>
<div class="nav-bar">
<h1>
Tubely
<span class="subtitle">The #1 tool for engagement bait</span>
</h1>
<button onclick="logout()">Logout</button>
</div>
<div id="auth-section">
<h2>Login</h2>
<form id="login-form">
<input
class="input-area"
type="email"
id="email"
placeholder="Email"
required
/>
<input
class="input-area"
type="password"
id="password"
placeholder="Password"
required
/>
<div class="button-container">
<button type="submit">Login</button>
<button onclick="signup()" type="button">Signup</button>
</div>
</form>
</div>
<div id="video-section" style="display: none">
<h2>Create Draft</h2>
<form id="video-draft-form">
<input
class="input-area"
type="text"
id="video-title"
placeholder="Video Title"
required
/>
<textarea
class="input-area"
id="video-description"
placeholder="Video Description"
required
></textarea>
<div class="button-container">
<button type="submit">Create Draft</button>
</div>
</form>
<h2>All Videos</h2>
<ul id="video-list"></ul>
<div id="video-display" style="display: none">
<h2>Current Video: <span id="video-title-display"></span></h2>
<p id="video-description-display"></p>
<div class="button-container mb-4">
<button onclick="deleteVideo()">Delete Video</button>
</div>
<div id="video-upload-forms">
<form
id="thumbnail-upload-form"
onsubmit="event.preventDefault(); uploadThumbnail(currentVideo?.id)"
>
<h3>Update Thumbnail</h3>
<input
type="file"
id="thumbnail"
accept="image/*,application/pdf"
required
/>
<button type="submit" id="upload-thumbnail-btn">Upload</button>
<img id="thumbnail-image" style="display: block" />
</form>
<div id="video-container">
<form
id="video-file-upload-form"
onsubmit="event.preventDefault(); uploadVideoFile(currentVideo?.id)"
>
<h3>Update Video File</h3>
<input type="file" id="video-file" accept="video/*" required />
<button type="submit" id="upload-video-btn">Upload</button>
</form>
<video id="video-player" controls style="display: block"></video>
</div>
</div>
</div>
</div>
</body>
</html>

183
app/styles.css Normal file
View File

@@ -0,0 +1,183 @@
:root {
--bg-color: #1e1e1e;
--fg-color: #f5f5f5;
--subtle-color: #888;
--primary-color: #bb86fc;
--input-bg: #2a2a2a;
--input-border: #444;
--button-bg: #7e57c2;
--button-hover: #5c4db1;
}
body {
margin: 0;
font-family: Arial, sans-serif;
background-color: var(--bg-color);
color: var(--fg-color);
line-height: 1.6;
}
.nav-bar {
display: flex;
justify-content: space-between;
align-items: center;
background-color: var(--bg-color);
padding: 10px 24px;
box-shadow: 0 2px 5px rgba(0, 0, 0, 0.2);
}
.nav-bar h1 {
margin: 0;
color: var(--primary-color);
display: flex;
align-items: center;
}
.nav-bar button {
background-color: var(--button-bg);
color: white;
border: none;
border-radius: 5px;
padding: 10px 20px;
cursor: pointer;
transition: background-color 0.3s ease;
}
.nav-bar button:hover {
background-color: var(--button-hover);
}
h1 {
color: var(--primary-color);
margin-bottom: 16px;
margin-left: 24px;
}
.subtitle {
color: var(--subtle-color);
font-size: 0.5em;
margin-left: 10px;
}
h2 {
color: var(--primary-color);
margin-bottom: 16px;
}
.input-area {
width: 100%;
padding: 10px;
border: 1px solid var(--input-border);
border-radius: 5px;
background-color: var(--input-bg);
color: var(--fg-color);
box-sizing: border-box;
}
button {
padding: 10px 20px;
border: none;
border-radius: 5px;
background-color: var(--button-bg);
color: #fff;
cursor: pointer;
transition: background-color 0.3s ease;
}
button:hover {
background-color: var(--button-hover);
}
.button-container {
display: flex;
justify-content: center;
gap: 10px;
}
#video-display {
border-top: 2px solid #333;
padding-top: 20px;
margin-top: 20px;
}
#video-list {
list-style: none;
padding: 0;
margin-top: 10px;
}
#video-list li {
padding: 10px;
margin-top: 5px;
background-color: #1e1e1e;
border-radius: 5px;
cursor: pointer;
transition: background-color 0.3s ease;
}
#video-list .active {
background-color: var(--primary-color);
color: #000;
}
#video-list li:hover {
background-color: #333;
}
#thumbnail-image,
#video-player {
max-width: 300px;
margin-left: 10px;
vertical-align: middle;
}
#auth-section {
max-width: 600px;
margin: 0 auto;
padding: 20px;
}
#video-section {
max-width: 1080px;
margin: 0 auto;
padding: 20px;
}
form {
border-radius: 5px;
display: flex;
flex-direction: column;
gap: 10px; /* Space between form elements */
}
#video-upload-forms {
display: flex;
gap: 20px; /* Space between the two forms */
}
#thumbnail-upload-form,
#video-container {
background-color: #1a1a1a;
padding: 20px;
border-radius: 5px;
flex: 1;
}
#download-button,
#video-player {
margin-top: 0.5rem;
width: 100%;
}
#video-upload-forms form {
flex: 1;
}
.mb-4 {
margin-bottom: 16px;
}
button[disabled] {
background-color: var(--subtle-color);
cursor: not-allowed;
}

12
assets.go Normal file
View File

@@ -0,0 +1,12 @@
package main
import (
"os"
)
func (cfg apiConfig) ensureAssetsDir() error {
if _, err := os.Stat(cfg.assetsRoot); os.IsNotExist(err) {
return os.Mkdir(cfg.assetsRoot, 0755)
}
return nil
}

10
cache.go Normal file
View File

@@ -0,0 +1,10 @@
package main
import "net/http"
func cacheMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Cache-Control", "max-age=3600")
next.ServeHTTP(w, r)
})
}

15
go.mod Normal file
View File

@@ -0,0 +1,15 @@
module github.com/bootdotdev/learn-file-storage-s3-golang-starter
go 1.23.0
require (
github.com/golang-jwt/jwt/v5 v5.0.0-rc.1
golang.org/x/crypto v0.7.0
)
require (
github.com/google/uuid v1.6.0
github.com/joho/godotenv v1.5.1
github.com/lib/pq v1.10.9
github.com/mattn/go-sqlite3 v1.14.24
)

12
go.sum Normal file
View File

@@ -0,0 +1,12 @@
github.com/golang-jwt/jwt/v5 v5.0.0-rc.1 h1:tDQ1LjKga657layZ4JLsRdxgvupebc0xuPwRNuTfUgs=
github.com/golang-jwt/jwt/v5 v5.0.0-rc.1/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
github.com/mattn/go-sqlite3 v1.14.24 h1:tpSp2G2KyMnnQu99ngJ47EIkWVmliIizyZBfPrBWDRM=
github.com/mattn/go-sqlite3 v1.14.24/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
golang.org/x/crypto v0.7.0 h1:AvwMYaRytfdeVt3u6mLaxYtErKYjxA2OXjJ1HHq6t3A=
golang.org/x/crypto v0.7.0/go.mod h1:pYwdfH91IfpZVANVyUOhSIPZaFoJGxTFbZhFTx+dXZU=

32
handler_get_thumbnail.go Normal file
View File

@@ -0,0 +1,32 @@
package main
import (
"fmt"
"net/http"
"github.com/google/uuid"
)
func (cfg *apiConfig) handlerThumbnailGet(w http.ResponseWriter, r *http.Request) {
videoIDString := r.PathValue("videoID")
videoID, err := uuid.Parse(videoIDString)
if err != nil {
respondWithError(w, http.StatusBadRequest, "Invalid video ID", err)
return
}
tn, ok := videoThumbnails[videoID]
if !ok {
respondWithError(w, http.StatusNotFound, "Thumbnail not found", nil)
return
}
w.Header().Set("Content-Type", tn.mediaType)
w.Header().Set("Content-Length", fmt.Sprintf("%d", len(tn.data)))
_, err = w.Write(tn.data)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Error writing response", err)
return
}
}

74
handler_login.go Normal file
View File

@@ -0,0 +1,74 @@
package main
import (
"encoding/json"
"net/http"
"time"
"github.com/bootdotdev/learn-file-storage-s3-golang-starter/internal/auth"
"github.com/bootdotdev/learn-file-storage-s3-golang-starter/internal/database"
)
func (cfg *apiConfig) handlerLogin(w http.ResponseWriter, r *http.Request) {
type parameters struct {
Password string `json:"password"`
Email string `json:"email"`
}
type response struct {
database.User
Token string `json:"token"`
RefreshToken string `json:"refresh_token"`
}
decoder := json.NewDecoder(r.Body)
params := parameters{}
err := decoder.Decode(&params)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't decode parameters", err)
return
}
user, err := cfg.db.GetUserByEmail(params.Email)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Incorrect email or password", err)
return
}
err = auth.CheckPasswordHash(params.Password, user.Password)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Incorrect email or password", err)
return
}
accessToken, err := auth.MakeJWT(
user.ID,
cfg.jwtSecret,
time.Hour*24*30,
)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't create access JWT", err)
return
}
refreshToken, err := auth.MakeRefreshToken()
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't create refresh token", err)
return
}
_, err = cfg.db.CreateRefreshToken(database.CreateRefreshTokenParams{
UserID: user.ID,
Token: refreshToken,
ExpiresAt: time.Now().UTC().Add(time.Hour * 24 * 60),
})
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't save refresh token", err)
return
}
respondWithJSON(w, http.StatusOK, response{
User: user,
Token: accessToken,
RefreshToken: refreshToken,
})
}

56
handler_refresh.go Normal file
View File

@@ -0,0 +1,56 @@
package main
import (
"net/http"
"time"
"github.com/bootdotdev/learn-file-storage-s3-golang-starter/internal/auth"
)
func (cfg *apiConfig) handlerRefresh(w http.ResponseWriter, r *http.Request) {
type response struct {
Token string `json:"token"`
}
refreshToken, err := auth.GetBearerToken(r.Header)
if err != nil {
respondWithError(w, http.StatusBadRequest, "Couldn't find token", err)
return
}
user, err := cfg.db.GetUserByRefreshToken(refreshToken)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't get user for refresh token", err)
return
}
accessToken, err := auth.MakeJWT(
user.ID,
cfg.jwtSecret,
time.Hour,
)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't validate token", err)
return
}
respondWithJSON(w, http.StatusOK, response{
Token: accessToken,
})
}
func (cfg *apiConfig) handlerRevoke(w http.ResponseWriter, r *http.Request) {
refreshToken, err := auth.GetBearerToken(r.Header)
if err != nil {
respondWithError(w, http.StatusBadRequest, "Couldn't find token", err)
return
}
err = cfg.db.RevokeRefreshToken(refreshToken)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't revoke session", err)
return
}
w.WriteHeader(http.StatusNoContent)
}

View File

@@ -0,0 +1,37 @@
package main
import (
"fmt"
"net/http"
"github.com/bootdotdev/learn-file-storage-s3-golang-starter/internal/auth"
"github.com/google/uuid"
)
func (cfg *apiConfig) handlerUploadThumbnail(w http.ResponseWriter, r *http.Request) {
videoIDString := r.PathValue("videoID")
videoID, err := uuid.Parse(videoIDString)
if err != nil {
respondWithError(w, http.StatusBadRequest, "Invalid ID", err)
return
}
token, err := auth.GetBearerToken(r.Header)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't find JWT", err)
return
}
userID, err := auth.ValidateJWT(token, cfg.jwtSecret)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't validate JWT", err)
return
}
fmt.Println("uploading thumbnail for video", videoID, "by user", userID)
// TODO: implement the upload here
respondWithJSON(w, http.StatusOK, struct{}{})
}

7
handler_upload_video.go Normal file
View File

@@ -0,0 +1,7 @@
package main
import (
"net/http"
)
func (cfg *apiConfig) handlerUploadVideo(w http.ResponseWriter, r *http.Request) {}

46
handler_users.go Normal file
View File

@@ -0,0 +1,46 @@
package main
import (
"encoding/json"
"net/http"
"github.com/bootdotdev/learn-file-storage-s3-golang-starter/internal/auth"
"github.com/bootdotdev/learn-file-storage-s3-golang-starter/internal/database"
)
func (cfg *apiConfig) handlerUsersCreate(w http.ResponseWriter, r *http.Request) {
type parameters struct {
Password string `json:"password"`
Email string `json:"email"`
}
decoder := json.NewDecoder(r.Body)
params := parameters{}
err := decoder.Decode(&params)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't decode parameters", err)
return
}
if params.Password == "" || params.Email == "" {
respondWithError(w, http.StatusBadRequest, "Email and password are required", nil)
return
}
hashedPassword, err := auth.HashPassword(params.Password)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't hash password", err)
return
}
user, err := cfg.db.CreateUser(database.CreateUserParams{
Email: params.Email,
Password: hashedPassword,
})
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't create user", err)
return
}
respondWithJSON(w, http.StatusCreated, user)
}

120
handler_video_meta.go Normal file
View File

@@ -0,0 +1,120 @@
package main
import (
"encoding/json"
"net/http"
"github.com/bootdotdev/learn-file-storage-s3-golang-starter/internal/auth"
"github.com/bootdotdev/learn-file-storage-s3-golang-starter/internal/database"
"github.com/google/uuid"
)
func (cfg *apiConfig) handlerVideoMetaCreate(w http.ResponseWriter, r *http.Request) {
type parameters struct {
database.CreateVideoParams
}
token, err := auth.GetBearerToken(r.Header)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't find JWT", err)
return
}
userID, err := auth.ValidateJWT(token, cfg.jwtSecret)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't validate JWT", err)
return
}
decoder := json.NewDecoder(r.Body)
params := parameters{}
err = decoder.Decode(&params)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't decode parameters", err)
return
}
params.UserID = userID
video, err := cfg.db.CreateVideo(params.CreateVideoParams)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't create video", err)
return
}
respondWithJSON(w, http.StatusCreated, video)
}
func (cfg *apiConfig) handlerVideoMetaDelete(w http.ResponseWriter, r *http.Request) {
videoIDString := r.PathValue("videoID")
videoID, err := uuid.Parse(videoIDString)
if err != nil {
respondWithError(w, http.StatusBadRequest, "Invalid ID", err)
return
}
token, err := auth.GetBearerToken(r.Header)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't find JWT", err)
return
}
userID, err := auth.ValidateJWT(token, cfg.jwtSecret)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't validate JWT", err)
return
}
video, err := cfg.db.GetVideo(videoID)
if err != nil {
respondWithError(w, http.StatusNotFound, "Couldn't get video", err)
return
}
if video.UserID != userID {
respondWithError(w, http.StatusForbidden, "You can't delete this video", err)
return
}
err = cfg.db.DeleteVideo(videoID)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't delete video", err)
return
}
w.WriteHeader(http.StatusNoContent)
}
func (cfg *apiConfig) handlerVideoGet(w http.ResponseWriter, r *http.Request) {
videoIDString := r.PathValue("videoID")
videoID, err := uuid.Parse(videoIDString)
if err != nil {
respondWithError(w, http.StatusBadRequest, "Invalid video ID", err)
return
}
video, err := cfg.db.GetVideo(videoID)
if err != nil {
respondWithError(w, http.StatusNotFound, "Couldn't get video", err)
return
}
respondWithJSON(w, http.StatusOK, video)
}
func (cfg *apiConfig) handlerVideosRetrieve(w http.ResponseWriter, r *http.Request) {
token, err := auth.GetBearerToken(r.Header)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't find JWT", err)
return
}
userID, err := auth.ValidateJWT(token, cfg.jwtSecret)
if err != nil {
respondWithError(w, http.StatusUnauthorized, "Couldn't validate JWT", err)
return
}
videos, err := cfg.db.GetVideos(userID)
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't retrieve videos", err)
return
}
respondWithJSON(w, http.StatusOK, videos)
}

116
internal/auth/auth.go Normal file
View File

@@ -0,0 +1,116 @@
package auth
import (
"crypto/rand"
"encoding/hex"
"errors"
"fmt"
"net/http"
"strings"
"time"
"github.com/golang-jwt/jwt/v5"
"github.com/google/uuid"
"golang.org/x/crypto/bcrypt"
)
type TokenType string
const (
TokenTypeAccess TokenType = "tubely-access"
)
var ErrNoAuthHeaderIncluded = errors.New("no auth header included in request")
func HashPassword(password string) (string, error) {
dat, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
if err != nil {
return "", err
}
return string(dat), nil
}
func CheckPasswordHash(password, hash string) error {
return bcrypt.CompareHashAndPassword([]byte(hash), []byte(password))
}
func MakeJWT(
userID uuid.UUID,
tokenSecret string,
expiresIn time.Duration,
) (string, error) {
signingKey := []byte(tokenSecret)
token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.RegisteredClaims{
Issuer: string(TokenTypeAccess),
IssuedAt: jwt.NewNumericDate(time.Now().UTC()),
ExpiresAt: jwt.NewNumericDate(time.Now().UTC().Add(expiresIn)),
Subject: userID.String(),
})
return token.SignedString(signingKey)
}
func ValidateJWT(tokenString, tokenSecret string) (uuid.UUID, error) {
claimsStruct := jwt.RegisteredClaims{}
token, err := jwt.ParseWithClaims(
tokenString,
&claimsStruct,
func(token *jwt.Token) (interface{}, error) { return []byte(tokenSecret), nil },
)
if err != nil {
return uuid.Nil, err
}
userIDString, err := token.Claims.GetSubject()
if err != nil {
return uuid.Nil, err
}
issuer, err := token.Claims.GetIssuer()
if err != nil {
return uuid.Nil, err
}
if issuer != string(TokenTypeAccess) {
return uuid.Nil, errors.New("invalid issuer")
}
id, err := uuid.Parse(userIDString)
if err != nil {
return uuid.Nil, fmt.Errorf("invalid user ID: %w", err)
}
return id, nil
}
func GetBearerToken(headers http.Header) (string, error) {
authHeader := headers.Get("Authorization")
if authHeader == "" {
return "", ErrNoAuthHeaderIncluded
}
splitAuth := strings.Split(authHeader, " ")
if len(splitAuth) < 2 || splitAuth[0] != "Bearer" {
return "", errors.New("malformed authorization header")
}
return splitAuth[1], nil
}
func MakeRefreshToken() (string, error) {
token := make([]byte, 32)
_, err := rand.Read(token)
if err != nil {
return "", err
}
return hex.EncodeToString(token), nil
}
func GetAPIKey(headers http.Header) (string, error) {
authHeader := headers.Get("Authorization")
if authHeader == "" {
return "", ErrNoAuthHeaderIncluded
}
splitAuth := strings.Split(authHeader, " ")
if len(splitAuth) < 2 || splitAuth[0] != "ApiKey" {
return "", errors.New("malformed authorization header")
}
return splitAuth[1], nil
}

View File

@@ -0,0 +1,89 @@
package database
import (
"database/sql"
"fmt"
_ "github.com/mattn/go-sqlite3"
)
type Client struct {
db *sql.DB
}
func NewClient(pathToDB string) (Client, error) {
db, err := sql.Open("sqlite3", pathToDB)
if err != nil {
return Client{}, err
}
c := Client{db}
err = c.autoMigrate()
if err != nil {
return Client{}, err
}
return c, nil
}
func (c *Client) autoMigrate() error {
userTable := `
CREATE TABLE IF NOT EXISTS users (
id TEXT PRIMARY KEY,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
password TEXT NOT NULL,
email TEXT UNIQUE NOT NULL
);
`
_, err := c.db.Exec(userTable)
if err != nil {
return err
}
refreshTokenTable := `
CREATE TABLE IF NOT EXISTS refresh_tokens (
token TEXT PRIMARY KEY,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
revoked_at TIMESTAMP,
user_id TEXT NOT NULL,
expires_at TIMESTAMP NOT NULL,
FOREIGN KEY(user_id) REFERENCES users(id)
);
`
_, err = c.db.Exec(refreshTokenTable)
if err != nil {
return err
}
videoTable := `
CREATE TABLE IF NOT EXISTS videos (
id TEXT PRIMARY KEY,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
title TEXT NOT NULL,
description TEXT,
thumbnail_url TEXT,
video_url TEXT TEXT,
user_id INTEGER,
FOREIGN KEY(user_id) REFERENCES users(id)
);
`
_, err = c.db.Exec(videoTable)
if err != nil {
return err
}
return nil
}
func (c Client) Reset() error {
if _, err := c.db.Exec("DELETE FROM refresh_tokens"); err != nil {
return fmt.Errorf("failed to reset table refresh_tokens: %w", err)
}
if _, err := c.db.Exec("DELETE FROM users"); err != nil {
return fmt.Errorf("failed to reset table users: %w", err)
}
if _, err := c.db.Exec("DELETE FROM videos"); err != nil {
return fmt.Errorf("failed to reset table videos: %w", err)
}
return nil
}

View File

@@ -0,0 +1,83 @@
package database
import (
"database/sql"
"time"
"github.com/google/uuid"
)
type RefreshToken struct {
CreateRefreshTokenParams
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
RevokedAt *time.Time `json:"revoked_at"`
}
type CreateRefreshTokenParams struct {
Token string `json:"token"`
UserID uuid.UUID `json:"user_id"`
ExpiresAt time.Time `json:"expires_at"`
}
func (c Client) CreateRefreshToken(params CreateRefreshTokenParams) (RefreshToken, error) {
query := `
INSERT INTO refresh_tokens (
token,
created_at,
updated_at,
user_id,
expires_at
) VALUES (?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP, ?, ?)
`
_, err := c.db.Exec(query, params.Token, params.UserID.String(), params.ExpiresAt)
if err != nil {
return RefreshToken{}, err
}
return c.GetRefreshToken(params.Token)
}
func (c Client) RevokeRefreshToken(token string) error {
query := `
UPDATE refresh_tokens
SET revoked_at = CURRENT_TIMESTAMP
WHERE token = ?
`
_, err := c.db.Exec(query, token)
return err
}
func (c Client) GetRefreshToken(token string) (RefreshToken, error) {
query := `
SELECT token, created_at, updated_at, user_id, expires_at, revoked_at
FROM refresh_tokens
WHERE token = ?
`
var rt RefreshToken
var userID string
err := c.db.QueryRow(query, token).
Scan(&rt.Token, &rt.CreatedAt, &rt.UpdatedAt, &userID, &rt.ExpiresAt, &rt.RevokedAt)
if err != nil {
if err == sql.ErrNoRows {
return RefreshToken{}, nil
}
return RefreshToken{}, err
}
rt.UserID, err = uuid.Parse(userID)
if err != nil {
return RefreshToken{}, err
}
return rt, nil
}
func (c Client) DeleteRefreshToken(token string) error {
query := `
DELETE FROM refresh_tokens
WHERE token = ?
`
_, err := c.db.Exec(query, token)
return err
}

147
internal/database/users.go Normal file
View File

@@ -0,0 +1,147 @@
package database
import (
"database/sql"
"errors"
"time"
"github.com/google/uuid"
)
type User struct {
ID uuid.UUID `json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
CreateUserParams
}
type CreateUserParams struct {
Email string `json:"email"`
Password string `json:"password"`
}
func (c Client) GetUsers() ([]User, error) {
query := `
SELECT
id,
email
FROM users
`
rows, err := c.db.Query(query)
if err != nil {
return nil, err
}
defer rows.Close()
users := []User{}
for rows.Next() {
var user User
var id string
if err := rows.Scan(&id, &user.Email); err != nil {
return nil, err
}
user.ID, err = uuid.Parse(id)
if err != nil {
return nil, err
}
users = append(users, user)
}
return users, nil
}
func (c Client) GetUserByEmail(email string) (User, error) {
query := `
SELECT id, created_at, updated_at, email, password
FROM users
WHERE email = ?
`
var user User
var id string
err := c.db.QueryRow(query, email).Scan(&id, &user.CreatedAt, &user.UpdatedAt, &user.Email, &user.Password)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return User{}, nil
}
return User{}, err
}
user.ID, err = uuid.Parse(id)
if err != nil {
return User{}, err
}
return user, nil
}
func (c Client) GetUserByRefreshToken(token string) (*User, error) {
query := `
SELECT u.id, u.email, u.created_at, u.updated_at, u.password
FROM users u
JOIN refresh_tokens rt ON u.id = rt.user_id
WHERE rt.token = ?
`
var user User
var id string
err := c.db.QueryRow(query, token).Scan(&id, &user.Email, &user.CreatedAt, &user.UpdatedAt, &user.Password)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return nil, nil
}
return nil, err
}
user.ID, err = uuid.Parse(id)
if err != nil {
return nil, err
}
return &user, nil
}
func (c Client) CreateUser(params CreateUserParams) (*User, error) {
id := uuid.New()
query := `
INSERT INTO users
(id, created_at, updated_at, email, password)
VALUES
(?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP, ?, ?)
`
_, err := c.db.Exec(query, id.String(), params.Email, params.Password)
if err != nil {
return nil, err
}
return c.GetUser(id)
}
func (c Client) GetUser(id uuid.UUID) (*User, error) {
query := `
SELECT id, created_at, updated_at, email, password
FROM users
WHERE id = ?
`
var user User
var idStr string
err := c.db.QueryRow(query, id.String()).Scan(&idStr, &user.CreatedAt, &user.UpdatedAt, &user.Email, &user.Password)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return nil, nil
}
return nil, err
}
user.ID, err = uuid.Parse(idStr)
if err != nil {
return nil, err
}
return &user, nil
}
func (c Client) DeleteUser(id uuid.UUID) error {
query := `
DELETE FROM users
WHERE id = ?
`
_, err := c.db.Exec(query, id.String())
return err
}

155
internal/database/videos.go Normal file
View File

@@ -0,0 +1,155 @@
package database
import (
"database/sql"
"errors"
"time"
"github.com/google/uuid"
)
type Video struct {
ID uuid.UUID `json:"id"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
ThumbnailURL *string `json:"thumbnail_url"`
VideoURL *string `json:"video_url"`
CreateVideoParams
}
type CreateVideoParams struct {
Title string `json:"title"`
Description string `json:"description"`
UserID uuid.UUID `json:"user_id"`
}
func (c Client) GetVideos(userID uuid.UUID) ([]Video, error) {
query := `
SELECT
id,
created_at,
updated_at,
title,
description,
thumbnail_url,
video_url,
user_id
FROM videos
WHERE user_id = ?
ORDER BY created_at DESC
`
rows, err := c.db.Query(query, userID)
if err != nil {
return nil, err
}
defer rows.Close()
videos := []Video{}
for rows.Next() {
var video Video
if err := rows.Scan(
&video.ID,
&video.CreatedAt,
&video.UpdatedAt,
&video.Title,
&video.Description,
&video.ThumbnailURL,
&video.VideoURL,
&video.UserID,
); err != nil {
return nil, err
}
videos = append(videos, video)
}
return videos, nil
}
func (c Client) CreateVideo(params CreateVideoParams) (Video, error) {
id := uuid.New()
query := `
INSERT INTO videos (
id,
created_at,
updated_at,
title,
description,
user_id
) VALUES (?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP, ?, ?, ?)
`
_, err := c.db.Exec(query, id, params.Title, params.Description, params.UserID)
if err != nil {
return Video{}, err
}
return c.GetVideo(id)
}
func (c Client) GetVideo(id uuid.UUID) (Video, error) {
query := `
SELECT
id,
created_at,
updated_at,
title,
description,
thumbnail_url,
video_url,
user_id
FROM videos
WHERE id = ?
`
var video Video
err := c.db.QueryRow(query, id).Scan(
&video.ID,
&video.CreatedAt,
&video.UpdatedAt,
&video.Title,
&video.Description,
&video.ThumbnailURL,
&video.VideoURL,
&video.UserID)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return Video{}, nil
}
return Video{}, err
}
return video, nil
}
func (c Client) UpdateVideo(video Video) error {
query := `
UPDATE videos
SET
title = ?,
description = ?,
thumbnail_url = ?,
video_url = ?,
user_id = ?
WHERE id = ?
`
_, err := c.db.Exec(
query,
video.Title,
video.Description,
&video.ThumbnailURL,
&video.VideoURL,
video.UserID,
video.ID,
)
return err
}
func (c Client) DeleteVideo(id uuid.UUID) error {
query := `
DELETE FROM videos
WHERE id = ?
`
_, err := c.db.Exec(query, id)
return err
}

34
json.go Normal file
View File

@@ -0,0 +1,34 @@
package main
import (
"encoding/json"
"log"
"net/http"
)
func respondWithError(w http.ResponseWriter, code int, msg string, err error) {
if err != nil {
log.Println(err)
}
if code > 499 {
log.Printf("Responding with 5XX error: %s", msg)
}
type errorResponse struct {
Error string `json:"error"`
}
respondWithJSON(w, code, errorResponse{
Error: msg,
})
}
func respondWithJSON(w http.ResponseWriter, code int, payload interface{}) {
w.Header().Set("Content-Type", "application/json")
dat, err := json.Marshal(payload)
if err != nil {
log.Printf("Error marshalling JSON: %s", err)
w.WriteHeader(500)
return
}
w.WriteHeader(code)
w.Write(dat)
}

134
main.go Normal file
View File

@@ -0,0 +1,134 @@
package main
import (
"log"
"net/http"
"os"
"github.com/bootdotdev/learn-file-storage-s3-golang-starter/internal/database"
"github.com/google/uuid"
"github.com/joho/godotenv"
_ "github.com/lib/pq"
)
type apiConfig struct {
db database.Client
jwtSecret string
platform string
filepathRoot string
assetsRoot string
s3Bucket string
s3Region string
s3CfDistribution string
port string
}
type thumbnail struct {
data []byte
mediaType string
}
var videoThumbnails = map[uuid.UUID]thumbnail{}
func main() {
godotenv.Load(".env")
pathToDB := os.Getenv("DB_PATH")
if pathToDB == "" {
log.Fatal("DB_URL must be set")
}
db, err := database.NewClient(pathToDB)
if err != nil {
log.Fatalf("Couldn't connect to database: %v", err)
}
jwtSecret := os.Getenv("JWT_SECRET")
if jwtSecret == "" {
log.Fatal("JWT_SECRET environment variable is not set")
}
platform := os.Getenv("PLATFORM")
if platform == "" {
log.Fatal("PLATFORM environment variable is not set")
}
filepathRoot := os.Getenv("FILEPATH_ROOT")
if filepathRoot == "" {
log.Fatal("FILEPATH_ROOT environment variable is not set")
}
assetsRoot := os.Getenv("ASSETS_ROOT")
if assetsRoot == "" {
log.Fatal("ASSETS_ROOT environment variable is not set")
}
s3Bucket := os.Getenv("S3_BUCKET")
if s3Bucket == "" {
log.Fatal("S3_BUCKET environment variable is not set")
}
s3Region := os.Getenv("S3_REGION")
if s3Region == "" {
log.Fatal("S3_REGION environment variable is not set")
}
s3CfDistribution := os.Getenv("S3_CF_DISTRO")
if s3CfDistribution == "" {
log.Fatal("S3_CF_DISTRO environment variable is not set")
}
port := os.Getenv("PORT")
if port == "" {
log.Fatal("PORT environment variable is not set")
}
cfg := apiConfig{
db: db,
jwtSecret: jwtSecret,
platform: platform,
filepathRoot: filepathRoot,
assetsRoot: assetsRoot,
s3Bucket: s3Bucket,
s3Region: s3Region,
s3CfDistribution: s3CfDistribution,
port: port,
}
err = cfg.ensureAssetsDir()
if err != nil {
log.Fatalf("Couldn't create assets directory: %v", err)
}
mux := http.NewServeMux()
appHandler := http.StripPrefix("/app", http.FileServer(http.Dir(filepathRoot)))
mux.Handle("/app/", appHandler)
assetsHandler := http.StripPrefix("/assets", http.FileServer(http.Dir(assetsRoot)))
mux.Handle("/assets/", cacheMiddleware(assetsHandler))
mux.HandleFunc("POST /api/login", cfg.handlerLogin)
mux.HandleFunc("POST /api/refresh", cfg.handlerRefresh)
mux.HandleFunc("POST /api/revoke", cfg.handlerRevoke)
mux.HandleFunc("POST /api/users", cfg.handlerUsersCreate)
mux.HandleFunc("POST /api/videos", cfg.handlerVideoMetaCreate)
mux.HandleFunc("POST /api/thumbnail_upload/{videoID}", cfg.handlerUploadThumbnail)
mux.HandleFunc("POST /api/video_upload/{videoID}", cfg.handlerUploadVideo)
mux.HandleFunc("GET /api/videos", cfg.handlerVideosRetrieve)
mux.HandleFunc("GET /api/videos/{videoID}", cfg.handlerVideoGet)
mux.HandleFunc("GET /api/thumbnails/{videoID}", cfg.handlerThumbnailGet)
mux.HandleFunc("DELETE /api/videos/{videoID}", cfg.handlerVideoMetaDelete)
mux.HandleFunc("POST /admin/reset", cfg.handlerReset)
srv := &http.Server{
Addr: ":" + port,
Handler: mux,
}
log.Printf("Serving on: http://localhost:%s/app/\n", port)
log.Fatal(srv.ListenAndServe())
}

19
reset.go Normal file
View File

@@ -0,0 +1,19 @@
package main
import "net/http"
func (cfg *apiConfig) handlerReset(w http.ResponseWriter, r *http.Request) {
if cfg.platform != "dev" {
w.WriteHeader(http.StatusForbidden)
w.Write([]byte("Reset is only allowed in dev environment."))
return
}
err := cfg.db.Reset()
if err != nil {
respondWithError(w, http.StatusInternalServerError, "Couldn't reset database", err)
return
}
w.WriteHeader(http.StatusOK)
w.Write([]byte("Database reset to initial state"))
}

16
samplesdownload.sh Executable file
View File

@@ -0,0 +1,16 @@
#!/bin/bash
mkdir -p samples
image_urls=(
"https://storage.googleapis.com/qvault-webapp-dynamic-assets/course_assets/boots-image-horizontal.png"
"https://storage.googleapis.com/qvault-webapp-dynamic-assets/course_assets/boots-image-vertical.png"
"https://storage.googleapis.com/qvault-webapp-dynamic-assets/course_assets/boots-video-horizontal.mp4"
"https://storage.googleapis.com/qvault-webapp-dynamic-assets/course_assets/boots-video-vertical.mp4"
"https://storage.googleapis.com/qvault-webapp-dynamic-assets/course_assets/is-bootdev-for-you.pdf"
)
for url in "${image_urls[@]}"; do
file_name=$(basename "$url")
curl -sSfL -o "samples/$file_name" "$url"
done