···

       
29
       
 
       


     

       
30
       
 
       
      - name: Install cosign

     

       
31
       
 
       
        if: github.event_name != 'pull_request'

     

       
32
       
-
       
        uses: sigstore/cosign-installer@v3.7.0

     

       
33
       
 
       
        with:

     

       
34
       
-
       
          cosign-release: "v2.4.1"

     

       
35
       
 
       


     

       
36
       
 
       
      - name: Setup Docker buildx

     

       
37
       
 
       
        uses: docker/setup-buildx-action@v3

     

···

       
29
       
 
       


     

       
30
       
 
       
      - name: Install cosign

     

       
31
       
 
       
        if: github.event_name != 'pull_request'

     

       
32
       
+
       
        uses: sigstore/cosign-installer@v3.8.1

     

       
33
       
 
       
        with:

     

       
34
       
+
       
          cosign-release: "v2.4.3"

     

       
35
       
 
       


     

       
36
       
 
       
      - name: Setup Docker buildx

     

       
37
       
 
       
        uses: docker/setup-buildx-action@v3

     

···

       
1440
       
 
       


     

       
1441
       
 
       
[[package]]

     

       
1442
       
 
       
name = "nu-ansi-term"

     

       
1443
       
-
       
version = "0.46.0"

     

       
1444
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
1445
       
-
       
checksum = "77a8165726e8236064dbb45459242600304b42a5ea24ee2948e18e023bf7ba84"

     

       
1446
       
 
       
dependencies = [

     

       
1447
       
-
       
 "overload",

     

       
1448
       
-
       
 "winapi",

     

       
1449
       
 
       
]

     

       
1450
       
 
       


     

       
1451
       
 
       
[[package]]

     
···

       
1525
       
 
       
version = "1.20.2"

     

       
1526
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
1527
       
 
       
checksum = "1261fe7e33c73b354eab43b1273a57c8f967d0391e80353e51f764ac02cf6775"

     

       
1528
       
-
       


     

       
1529
       
-
       
[[package]]

     

       
1530
       
-
       
name = "overload"

     

       
1531
       
-
       
version = "0.1.1"

     

       
1532
       
-
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
1533
       
-
       
checksum = "b15813163c1d831bf4a13c3610c05c0d03b39feb07f7e09fa234dac9b15aaf39"

     

       
1534
       
 
       


     

       
1535
       
 
       
[[package]]

     

       
1536
       
 
       
name = "parking"

     
···

       
1751
       
 
       


     

       
1752
       
 
       
[[package]]

     

       
1753
       
 
       
name = "ring"

     

       
1754
       
-
       
version = "0.17.8"

     

       
1755
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
1756
       
-
       
checksum = "c17fa4cb658e3583423e915b9f3acc01cceaee1860e33d59ebae66adc3a2dc0d"

     

       
1757
       
 
       
dependencies = [

     

       
1758
       
 
       
 "cc",

     

       
1759
       
 
       
 "cfg-if",

     

       
1760
       
 
       
 "getrandom",

     

       
1761
       
 
       
 "libc",

     

       
1762
       
-
       
 "spin",

     

       
1763
       
 
       
 "untrusted",

     

       
1764
       
 
       
 "windows-sys 0.52.0",

     

       
1765
       
 
       
]

     
···

       
2432
       
 
       


     

       
2433
       
 
       
[[package]]

     

       
2434
       
 
       
name = "tokio"

     

       
2435
       
-
       
version = "1.43.0"

     

       
2436
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
2437
       
-
       
checksum = "3d61fa4ffa3de412bfea335c6ecff681de2b609ba3c77ef3e00e521813a9ed9e"

     

       
2438
       
 
       
dependencies = [

     

       
2439
       
 
       
 "backtrace",

     

       
2440
       
 
       
 "bytes",

     
···

       
2529
       
 
       


     

       
2530
       
 
       
[[package]]

     

       
2531
       
 
       
name = "tracing-subscriber"

     

       
2532
       
-
       
version = "0.3.19"

     

       
2533
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
2534
       
-
       
checksum = "e8189decb5ac0fa7bc8b96b7cb9b2701d60d48805aca84a238004d665fcc4008"

     

       
2535
       
 
       
dependencies = [

     

       
2536
       
 
       
 "nu-ansi-term",

     

       
2537
       
 
       
 "sharded-slab",

     
···

       
2739
       
 
       
]

     

       
2740
       
 
       


     

       
2741
       
 
       
[[package]]

     

       
2742
       
-
       
name = "winapi"

     

       
2743
       
-
       
version = "0.3.9"

     

       
2744
       
-
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
2745
       
-
       
checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419"

     

       
2746
       
-
       
dependencies = [

     

       
2747
       
-
       
 "winapi-i686-pc-windows-gnu",

     

       
2748
       
-
       
 "winapi-x86_64-pc-windows-gnu",

     

       
2749
       
-
       
]

     

       
2750
       
-
       


     

       
2751
       
-
       
[[package]]

     

       
2752
       
-
       
name = "winapi-i686-pc-windows-gnu"

     

       
2753
       
-
       
version = "0.4.0"

     

       
2754
       
-
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
2755
       
-
       
checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"

     

       
2756
       
-
       


     

       
2757
       
-
       
[[package]]

     

       
2758
       
 
       
name = "winapi-util"

     

       
2759
       
 
       
version = "0.1.9"

     

       
2760
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     
···

       
2762
       
 
       
dependencies = [

     

       
2763
       
 
       
 "windows-sys 0.59.0",

     

       
2764
       
 
       
]

     

       
2765
       
-
       


     

       
2766
       
-
       
[[package]]

     

       
2767
       
-
       
name = "winapi-x86_64-pc-windows-gnu"

     

       
2768
       
-
       
version = "0.4.0"

     

       
2769
       
-
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
2770
       
-
       
checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"

     

       
2771
       
 
       


     

       
2772
       
 
       
[[package]]

     

       
2773
       
 
       
name = "windows-core"

     

···

       
1440
       
 
       


     

       
1441
       
 
       
[[package]]

     

       
1442
       
 
       
name = "nu-ansi-term"

     

       
1443
       
+
       
version = "0.50.1"

     

       
1444
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
1445
       
+
       
checksum = "d4a28e057d01f97e61255210fcff094d74ed0466038633e95017f5beb68e4399"

     

       
1446
       
 
       
dependencies = [

     

       
1447
       
+
       
 "windows-sys 0.52.0",

     

       
0
       
 
       

     

       
1448
       
 
       
]

     

       
1449
       
 
       


     

       
1450
       
 
       
[[package]]

     
···

       
1524
       
 
       
version = "1.20.2"

     

       
1525
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
1526
       
 
       
checksum = "1261fe7e33c73b354eab43b1273a57c8f967d0391e80353e51f764ac02cf6775"

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
1527
       
 
       


     

       
1528
       
 
       
[[package]]

     

       
1529
       
 
       
name = "parking"

     
···

       
1744
       
 
       


     

       
1745
       
 
       
[[package]]

     

       
1746
       
 
       
name = "ring"

     

       
1747
       
+
       
version = "0.17.13"

     

       
1748
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
1749
       
+
       
checksum = "70ac5d832aa16abd7d1def883a8545280c20a60f523a370aa3a9617c2b8550ee"

     

       
1750
       
 
       
dependencies = [

     

       
1751
       
 
       
 "cc",

     

       
1752
       
 
       
 "cfg-if",

     

       
1753
       
 
       
 "getrandom",

     

       
1754
       
 
       
 "libc",

     

       
0
       
 
       

     

       
1755
       
 
       
 "untrusted",

     

       
1756
       
 
       
 "windows-sys 0.52.0",

     

       
1757
       
 
       
]

     
···

       
2424
       
 
       


     

       
2425
       
 
       
[[package]]

     

       
2426
       
 
       
name = "tokio"

     

       
2427
       
+
       
version = "1.43.1"

     

       
2428
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
2429
       
+
       
checksum = "492a604e2fd7f814268a378409e6c92b5525d747d10db9a229723f55a417958c"

     

       
2430
       
 
       
dependencies = [

     

       
2431
       
 
       
 "backtrace",

     

       
2432
       
 
       
 "bytes",

     
···

       
2521
       
 
       


     

       
2522
       
 
       
[[package]]

     

       
2523
       
 
       
name = "tracing-subscriber"

     

       
2524
       
+
       
version = "0.3.20"

     

       
2525
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     

       
2526
       
+
       
checksum = "2054a14f5307d601f88daf0553e1cbf472acc4f2c51afab632431cdcd72124d5"

     

       
2527
       
 
       
dependencies = [

     

       
2528
       
 
       
 "nu-ansi-term",

     

       
2529
       
 
       
 "sharded-slab",

     
···

       
2731
       
 
       
]

     

       
2732
       
 
       


     

       
2733
       
 
       
[[package]]

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
2734
       
 
       
name = "winapi-util"

     

       
2735
       
 
       
version = "0.1.9"

     

       
2736
       
 
       
source = "registry+https://github.com/rust-lang/crates.io-index"

     
···

       
2738
       
 
       
dependencies = [

     

       
2739
       
 
       
 "windows-sys 0.59.0",

     

       
2740
       
 
       
]

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
2741
       
 
       


     

       
2742
       
 
       
[[package]]

     

       
2743
       
 
       
name = "windows-core"

     

···

       
13
       
 
       
actix-web = "4.4"

     

       
14
       
 
       
actix-files = "0.6"

     

       
15
       
 
       
actix-cors = "0.6"

     

       
16
       
-
       
tokio = { version = "1.36", features = ["rt-multi-thread", "macros"] }

     

       
17
       
 
       
sqlx = { version = "0.8", features = ["runtime-tokio", "postgres", "sqlite", "chrono"] }

     

       
18
       
 
       
serde = { version = "1.0", features = ["derive"] }

     

       
19
       
 
       
serde_json = "1.0"

     

···

       
13
       
 
       
actix-web = "4.4"

     

       
14
       
 
       
actix-files = "0.6"

     

       
15
       
 
       
actix-cors = "0.6"

     

       
16
       
+
       
tokio = { version = "1.43", features = ["rt-multi-thread", "macros"] }

     

       
17
       
 
       
sqlx = { version = "0.8", features = ["runtime-tokio", "postgres", "sqlite", "chrono"] }

     

       
18
       
 
       
serde = { version = "1.0", features = ["derive"] }

     

       
19
       
 
       
serde_json = "1.0"

     

···

       
4
       
 
       
WORKDIR /usr/src/frontend

     

       
5
       
 
       


     

       
6
       
 
       
# Copy frontend files

     

       
7
       
-
       
COPY frontend/package*.json ./

     

       
8
       
 
       
RUN bun install

     

       
9
       
 
       


     

       
10
       
 
       
COPY frontend/ ./

     

···

       
4
       
 
       
WORKDIR /usr/src/frontend

     

       
5
       
 
       


     

       
6
       
 
       
# Copy frontend files

     

       
7
       
+
       
COPY frontend/package.json ./

     

       
8
       
 
       
RUN bun install

     

       
9
       
 
       


     

       
10
       
 
       
COPY frontend/ ./

     

···

       
1
       
 
       
# SimpleLink

     

       
2
       
 
       


     

       
3
       
-
       
A very performant and light (6mb in memory) link shortener and tracker. Written in Rust and React and uses Postgres.

     

       
4
       
 
       


     

       
5
       
 
       
![MainView](readme_img/mainview.jpg)

     

       
6
       
 
       


     
···

       
8
       
 
       


     

       
9
       
 
       
## How to Run

     

       
10
       
 
       


     

       
11
       
-
       
### From Docker:

     

       
12
       
 
       


     

       
13
       
-
       
```Bash

     

       
14
       
 
       
docker run -p 8080:8080 \

     

       
15
       
 
       
    -e JWT_SECRET=change-me-in-production \

     

       
0
       
 
       

     

       
0
       
 
       

     

       
16
       
 
       
    -v simplelink_data:/data \

     

       
17
       
-
       
    ghcr.io/waveringana/simplelink:v2

     

       
18
       
 
       
```

     

       
19
       
 
       


     

       
20
       
-
       
Find the admin-setup-token pasted into the terminal output, or in admin-setup-token.txt in the container's root.

     

       
21
       
 
       


     

       
22
       
-
       
This is needed to register with the frontend. (TODO, register admin account with ENV)

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
23
       
 
       


     

       
24
       
-
       
### From Docker Compose:

     

       
0
       
 
       

     

       
0
       
 
       

     

       
25
       
 
       


     

       
26
       
-
       
Edit the docker-compose.yml file. It comes included with a postgressql db for use

     

       
27
       
 
       


     

       
28
       
 
       
## Build

     

       
29
       
 
       


     
···

       
31
       
 
       


     

       
32
       
 
       
First configure .env.example and save it to .env

     

       
33
       
 
       


     

       
34
       
-
       
If DATABASE_URL is set, it will connect to a Postgres DB. If blank, it will use an sqlite db in /data

     

       
35
       
-
       


     

       
36
       
 
       
```bash

     

       
37
       
 
       
git clone https://github.com/waveringana/simplelink && cd simplelink

     

       
38
       
 
       
./build.sh

     

       
39
       
 
       
cargo run

     

       
40
       
 
       
```

     

       
41
       
 
       


     

       
42
       
-
       
On an empty database, an admin-setup-token.txt is created as well as pasted into the terminal output. This is needed to make the admin account.

     

       
43
       
-
       


     

       
44
       
-
       
Alternatively if you want a binary form

     

       
45
       
 
       


     

       
46
       
 
       
```bash

     

       
47
       
 
       
./build.sh --binary

     
···

       
55
       
 
       
docker build -t simplelink .

     

       
56
       
 
       
docker run -p 8080:8080 \

     

       
57
       
 
       
    -e JWT_SECRET=change-me-in-production \

     

       
0
       
 
       

     

       
0
       
 
       

     

       
58
       
 
       
    -v simplelink_data:/data \

     

       
59
       
 
       
    simplelink

     

       
60
       
 
       
```

     
···

       
62
       
 
       
### From Docker Compose

     

       
63
       
 
       


     

       
64
       
 
       
Adjust the included docker-compose.yml to your liking; it includes a postgres config as well.

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

···

       
1
       
 
       
# SimpleLink

     

       
2
       
 
       


     

       
3
       
+
       
A very performant and light (2MB in memory) link shortener and tracker. Written in Rust and React and uses Postgres or SQLite.

     

       
4
       
 
       


     

       
5
       
 
       
![MainView](readme_img/mainview.jpg)

     

       
6
       
 
       


     
···

       
8
       
 
       


     

       
9
       
 
       
## How to Run

     

       
10
       
 
       


     

       
11
       
+
       
### From Docker

     

       
12
       
 
       


     

       
13
       
+
       
```bash

     

       
14
       
 
       
docker run -p 8080:8080 \

     

       
15
       
 
       
    -e JWT_SECRET=change-me-in-production \

     

       
16
       
+
       
    -e SIMPLELINK_USER=admin@example.com \

     

       
17
       
+
       
    -e SIMPLELINK_PASS=your-secure-password \

     

       
18
       
 
       
    -v simplelink_data:/data \

     

       
19
       
+
       
    ghcr.io/waveringana/simplelink:v2.2

     

       
20
       
 
       
```

     

       
21
       
 
       


     

       
22
       
+
       
### Environment Variables

     

       
23
       
 
       


     

       
24
       
+
       
- `JWT_SECRET`: Required. Used for JWT token generation

     

       
25
       
+
       
- `SIMPLELINK_USER`: Optional. If set along with SIMPLELINK_PASS, creates an admin user on first run

     

       
26
       
+
       
- `SIMPLELINK_PASS`: Optional. Admin user password

     

       
27
       
+
       
- `DATABASE_URL`: Optional. Postgres connection string. If not set, uses SQLite

     

       
28
       
+
       
- `INITIAL_LINKS`: Optional. Semicolon-separated list of initial links in format "url,code;url2,code2"

     

       
29
       
+
       
- `SERVER_HOST`: Optional. Default: "127.0.0.1"

     

       
30
       
+
       
- `SERVER_PORT`: Optional. Default: "8080"

     

       
31
       
 
       


     

       
32
       
+
       
If `SIMPLELINK_USER` and `SIMPLELINK_PASS` are not passed, an admin-setup-token is pasted to the console and as a text file in the project root.

     

       
33
       
+
       


     

       
34
       
+
       
### From Docker Compose

     

       
35
       
 
       


     

       
36
       
+
       
Edit the docker-compose.yml file. It comes included with a PostgreSQL db configuration.

     

       
37
       
 
       


     

       
38
       
 
       
## Build

     

       
39
       
 
       


     
···

       
41
       
 
       


     

       
42
       
 
       
First configure .env.example and save it to .env

     

       
43
       
 
       


     

       
0
       
 
       

     

       
0
       
 
       

     

       
44
       
 
       
```bash

     

       
45
       
 
       
git clone https://github.com/waveringana/simplelink && cd simplelink

     

       
46
       
 
       
./build.sh

     

       
47
       
 
       
cargo run

     

       
48
       
 
       
```

     

       
49
       
 
       


     

       
50
       
+
       
Alternatively for a binary build:

     

       
0
       
 
       

     

       
0
       
 
       

     

       
51
       
 
       


     

       
52
       
 
       
```bash

     

       
53
       
 
       
./build.sh --binary

     
···

       
61
       
 
       
docker build -t simplelink .

     

       
62
       
 
       
docker run -p 8080:8080 \

     

       
63
       
 
       
    -e JWT_SECRET=change-me-in-production \

     

       
64
       
+
       
    -e SIMPLELINK_USER=admin@example.com \

     

       
65
       
+
       
    -e SIMPLELINK_PASS=your-secure-password \

     

       
66
       
 
       
    -v simplelink_data:/data \

     

       
67
       
 
       
    simplelink

     

       
68
       
 
       
```

     
···

       
70
       
 
       
### From Docker Compose

     

       
71
       
 
       


     

       
72
       
 
       
Adjust the included docker-compose.yml to your liking; it includes a postgres config as well.

     

       
73
       
+
       


     

       
74
       
+
       
## Features

     

       
75
       
+
       


     

       
76
       
+
       
- Support for both PostgreSQL and SQLite databases

     

       
77
       
+
       
- Initial links can be configured via environment variables

     

       
78
       
+
       
- Admin user can be created on first run via environment variables

     

       
79
       
+
       
- Link click tracking and statistics

     

       
80
       
+
       
- Lightweight and performant

     

···

       
19
       
 
       
      - shortener-network

     

       
20
       
 
       


     

       
21
       
 
       
  app:

     

       
22
       
-
       
    image: ghcr.io/waveringana/simplelink:v2

     

       
23
       
 
       
    container_name: shortener-app

     

       
24
       
 
       
    ports:

     

       
25
       
 
       
      - "8080:8080"

     

···

       
19
       
 
       
      - shortener-network

     

       
20
       
 
       


     

       
21
       
 
       
  app:

     

       
22
       
+
       
    image: ghcr.io/waveringana/simplelink:v2.2

     

       
23
       
 
       
    container_name: shortener-app

     

       
24
       
 
       
    ports:

     

       
25
       
 
       
      - "8080:8080"

     

···

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

···

       
1
       
+
       
-- PostgreSQL migration

     

       
2
       
+
       
ALTER TABLE links ALTER COLUMN short_code TYPE VARCHAR(32);

     

       
3
       
+
       


     

···

       
0
       
 
       

     

       
1
       
 
       
use actix_web::{dev::Payload, FromRequest, HttpRequest};

     

       
2
       
 
       
use jsonwebtoken::{decode, DecodingKey, Validation};

     

       
3
       
 
       
use std::future::{ready, Ready};

     

       
4
       
-
       
use crate::{error::AppError, models::Claims};

     

       
5
       
 
       


     

       
6
       
 
       
pub struct AuthenticatedUser {

     

       
7
       
 
       
    pub user_id: i32,

     
···

       
12
       
 
       
    type Future = Ready<Result<Self, Self::Error>>;

     

       
13
       
 
       


     

       
14
       
 
       
    fn from_request(req: &HttpRequest, _: &mut Payload) -> Self::Future {

     

       
15
       
-
       
        let auth_header = req.headers()

     

       
0
       
 
       

     

       
16
       
 
       
            .get("Authorization")

     

       
17
       
 
       
            .and_then(|h| h.to_str().ok());

     

       
18
       
 
       


     

       
19
       
 
       
        if let Some(auth_header) = auth_header {

     

       
20
       
 
       
            if auth_header.starts_with("Bearer ") {

     

       
21
       
 
       
                let token = &auth_header[7..];

     

       
22
       
-
       
                let secret = std::env::var("JWT_SECRET").unwrap_or_else(|_| "default_secret".to_string());

     

       
23
       
-
       
                

     

       
24
       
 
       
                match decode::<Claims>(

     

       
25
       
 
       
                    token,

     

       
26
       
 
       
                    &DecodingKey::from_secret(secret.as_bytes()),

     

       
27
       
-
       
                    &Validation::default()

     

       
28
       
 
       
                ) {

     

       
29
       
 
       
                    Ok(token_data) => {

     

       
30
       
 
       
                        return ready(Ok(AuthenticatedUser {

     
···

       
35
       
 
       
                }

     

       
36
       
 
       
            }

     

       
37
       
 
       
        }

     

       
38
       
-
       
        

     

       
39
       
 
       
        ready(Err(AppError::Unauthorized))

     

       
40
       
 
       
    }

     

       
41
       
-
       
}
     

       
0
       
 
       

     

···

       
1
       
+
       
use crate::{error::AppError, models::Claims};

     

       
2
       
 
       
use actix_web::{dev::Payload, FromRequest, HttpRequest};

     

       
3
       
 
       
use jsonwebtoken::{decode, DecodingKey, Validation};

     

       
4
       
 
       
use std::future::{ready, Ready};

     

       
0
       
 
       

     

       
5
       
 
       


     

       
6
       
 
       
pub struct AuthenticatedUser {

     

       
7
       
 
       
    pub user_id: i32,

     
···

       
12
       
 
       
    type Future = Ready<Result<Self, Self::Error>>;

     

       
13
       
 
       


     

       
14
       
 
       
    fn from_request(req: &HttpRequest, _: &mut Payload) -> Self::Future {

     

       
15
       
+
       
        let auth_header = req

     

       
16
       
+
       
            .headers()

     

       
17
       
 
       
            .get("Authorization")

     

       
18
       
 
       
            .and_then(|h| h.to_str().ok());

     

       
19
       
 
       


     

       
20
       
 
       
        if let Some(auth_header) = auth_header {

     

       
21
       
 
       
            if auth_header.starts_with("Bearer ") {

     

       
22
       
 
       
                let token = &auth_header[7..];

     

       
23
       
+
       
                let secret =

     

       
24
       
+
       
                    std::env::var("JWT_SECRET").unwrap_or_else(|_| "default_secret".to_string());

     

       
25
       
 
       
                match decode::<Claims>(

     

       
26
       
 
       
                    token,

     

       
27
       
 
       
                    &DecodingKey::from_secret(secret.as_bytes()),

     

       
28
       
+
       
                    &Validation::default(),

     

       
29
       
 
       
                ) {

     

       
30
       
 
       
                    Ok(token_data) => {

     

       
31
       
 
       
                        return ready(Ok(AuthenticatedUser {

     
···

       
36
       
 
       
                }

     

       
37
       
 
       
            }

     

       
38
       
 
       
        }

     

       
0
       
 
       

     

       
39
       
 
       
        ready(Err(AppError::Unauthorized))

     

       
40
       
 
       
    }

     

       
41
       
+
       
}

     

       
42
       
+
       


     

···

       
131
       
 
       
    Ok(())

     

       
132
       
 
       
}

     

       
133
       
 
       


     

       
134
       
-
       
fn validate_url(url: &String) -> Result<(), AppError> {

     

       
135
       
 
       
    if url.is_empty() {

     

       
136
       
 
       
        return Err(AppError::InvalidInput("URL cannot be empty".to_string()));

     

       
137
       
 
       
    }

     
···

       
707
       
 
       
                WHERE link_id = $1

     

       
708
       
 
       
                GROUP BY DATE(created_at)

     

       
709
       
 
       
                ORDER BY DATE(created_at) ASC

     

       
710
       
-
       
                LIMIT 30

     

       
711
       
 
       
                "#,

     

       
712
       
 
       
            )

     

       
713
       
 
       
            .bind(link_id)

     
···

       
724
       
 
       
                WHERE link_id = ?

     

       
725
       
 
       
                GROUP BY DATE(created_at)

     

       
726
       
 
       
                ORDER BY DATE(created_at) ASC

     

       
727
       
-
       
                LIMIT 30

     

       
728
       
 
       
                "#,

     

       
729
       
 
       
            )

     

       
730
       
 
       
            .bind(link_id)

     
···

       
789
       
 
       
                    AND query_source != ''

     

       
790
       
 
       
                GROUP BY DATE(created_at), query_source

     

       
791
       
 
       
                ORDER BY DATE(created_at) ASC, COUNT(*) DESC

     

       
792
       
-
       
                LIMIT 300

     

       
793
       
 
       
                "#,

     

       
794
       
 
       
            )

     

       
795
       
 
       
            .bind(link_id)

     
···

       
809
       
 
       
                    AND query_source != ''

     

       
810
       
 
       
                GROUP BY DATE(created_at), query_source

     

       
811
       
 
       
                ORDER BY DATE(created_at) ASC, COUNT(*) DESC

     

       
812
       
-
       
                LIMIT 300

     

       
813
       
 
       
                "#,

     

       
814
       
 
       
            )

     

       
815
       
 
       
            .bind(link_id)

     

···

       
131
       
 
       
    Ok(())

     

       
132
       
 
       
}

     

       
133
       
 
       


     

       
134
       
+
       
fn validate_url(url: &str) -> Result<(), AppError> {

     

       
135
       
 
       
    if url.is_empty() {

     

       
136
       
 
       
        return Err(AppError::InvalidInput("URL cannot be empty".to_string()));

     

       
137
       
 
       
    }

     
···

       
707
       
 
       
                WHERE link_id = $1

     

       
708
       
 
       
                GROUP BY DATE(created_at)

     

       
709
       
 
       
                ORDER BY DATE(created_at) ASC

     

       
0
       
 
       

     

       
710
       
 
       
                "#,

     

       
711
       
 
       
            )

     

       
712
       
 
       
            .bind(link_id)

     
···

       
723
       
 
       
                WHERE link_id = ?

     

       
724
       
 
       
                GROUP BY DATE(created_at)

     

       
725
       
 
       
                ORDER BY DATE(created_at) ASC

     

       
0
       
 
       

     

       
726
       
 
       
                "#,

     

       
727
       
 
       
            )

     

       
728
       
 
       
            .bind(link_id)

     
···

       
787
       
 
       
                    AND query_source != ''

     

       
788
       
 
       
                GROUP BY DATE(created_at), query_source

     

       
789
       
 
       
                ORDER BY DATE(created_at) ASC, COUNT(*) DESC

     

       
0
       
 
       

     

       
790
       
 
       
                "#,

     

       
791
       
 
       
            )

     

       
792
       
 
       
            .bind(link_id)

     
···

       
806
       
 
       
                    AND query_source != ''

     

       
807
       
 
       
                GROUP BY DATE(created_at), query_source

     

       
808
       
 
       
                ORDER BY DATE(created_at) ASC, COUNT(*) DESC

     

       
0
       
 
       

     

       
809
       
 
       
                "#,

     

       
810
       
 
       
            )

     

       
811
       
 
       
            .bind(link_id)

     

···

       
1
       
 
       
use actix_cors::Cors;

     

       
2
       
 
       
use actix_web::{web, App, HttpResponse, HttpServer};

     

       
3
       
 
       
use anyhow::Result;

     

       
0
       
 
       

     

       
4
       
 
       
use rust_embed::RustEmbed;

     

       
5
       
 
       
use simplelink::check_and_generate_admin_token;

     

       
0
       
 
       

     

       
6
       
 
       
use simplelink::{create_db_pool, run_migrations};

     

       
7
       
 
       
use simplelink::{handlers, AppState};

     

       
8
       
-
       
use tracing::info;

     

       
0
       
 
       

     

       
9
       
 
       


     

       
0
       
 
       

     

       
0
       
 
       

     

       
10
       
 
       
#[derive(RustEmbed)]

     

       
11
       
 
       
#[folder = "static/"]

     

       
12
       
 
       
struct Asset;

     
···

       
23
       
 
       
    }

     

       
24
       
 
       
}

     

       
25
       
 
       


     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
26
       
 
       
#[actix_web::main]

     

       
27
       
 
       
async fn main() -> Result<()> {

     

       
28
       
 
       
    // Load environment variables from .env file

     
···

       
34
       
 
       
    // Create database connection pool

     

       
35
       
 
       
    let pool = create_db_pool().await?;

     

       
36
       
 
       
    run_migrations(&pool).await?;

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
37
       
 
       


     

       
38
       
 
       
    let admin_token = check_and_generate_admin_token(&pool).await?;

     

       
39
       
 
       


     

···

       
1
       
 
       
use actix_cors::Cors;

     

       
2
       
 
       
use actix_web::{web, App, HttpResponse, HttpServer};

     

       
3
       
 
       
use anyhow::Result;

     

       
4
       
+
       
use clap::Parser;

     

       
5
       
 
       
use rust_embed::RustEmbed;

     

       
6
       
 
       
use simplelink::check_and_generate_admin_token;

     

       
7
       
+
       
use simplelink::models::DatabasePool;

     

       
8
       
 
       
use simplelink::{create_db_pool, run_migrations};

     

       
9
       
 
       
use simplelink::{handlers, AppState};

     

       
10
       
+
       
use sqlx::{Postgres, Sqlite};

     

       
11
       
+
       
use tracing::{error, info};

     

       
12
       
 
       


     

       
13
       
+
       
#[derive(Parser, Debug)]

     

       
14
       
+
       
#[command(author, version, about, long_about = None)]

     

       
15
       
 
       
#[derive(RustEmbed)]

     

       
16
       
 
       
#[folder = "static/"]

     

       
17
       
 
       
struct Asset;

     
···

       
28
       
 
       
    }

     

       
29
       
 
       
}

     

       
30
       
 
       


     

       
31
       
+
       
async fn create_initial_links(pool: &DatabasePool) -> Result<()> {

     

       
32
       
+
       
    if let Ok(links) = std::env::var("INITIAL_LINKS") {

     

       
33
       
+
       
        for link_entry in links.split(';') {

     

       
34
       
+
       
            let parts: Vec<&str> = link_entry.split(',').collect();

     

       
35
       
+
       
            if parts.len() >= 2 {

     

       
36
       
+
       
                let url = parts[0];

     

       
37
       
+
       
                let code = parts[1];

     

       
38
       
+
       


     

       
39
       
+
       
                match pool {

     

       
40
       
+
       
                    DatabasePool::Postgres(pool) => {

     

       
41
       
+
       
                        sqlx::query(

     

       
42
       
+
       
                            "INSERT INTO links (original_url, short_code, user_id)  

     

       
43
       
+
       
                             VALUES ($1, $2, $3) 

     

       
44
       
+
       
                             ON CONFLICT (short_code) 

     

       
45
       
+
       
                             DO UPDATE SET short_code = EXCLUDED.short_code

     

       
46
       
+
       
                             WHERE links.original_url = EXCLUDED.original_url",

     

       
47
       
+
       
                        )

     

       
48
       
+
       
                        .bind(url)

     

       
49
       
+
       
                        .bind(code)

     

       
50
       
+
       
                        .bind(1)

     

       
51
       
+
       
                        .execute(pool)

     

       
52
       
+
       
                        .await?;

     

       
53
       
+
       
                    }

     

       
54
       
+
       
                    DatabasePool::Sqlite(pool) => {

     

       
55
       
+
       
                        // First check if the exact combination exists

     

       
56
       
+
       
                        let exists = sqlx::query_scalar::<_, bool>(

     

       
57
       
+
       
                            "SELECT EXISTS(

     

       
58
       
+
       
                                SELECT 1 FROM links 

     

       
59
       
+
       
                                WHERE original_url = ?1 

     

       
60
       
+
       
                                AND short_code = ?2

     

       
61
       
+
       
                            )",

     

       
62
       
+
       
                        )

     

       
63
       
+
       
                        .bind(url)

     

       
64
       
+
       
                        .bind(code)

     

       
65
       
+
       
                        .fetch_one(pool)

     

       
66
       
+
       
                        .await?;

     

       
67
       
+
       


     

       
68
       
+
       
                        // Only insert if the exact combination doesn't exist

     

       
69
       
+
       
                        if !exists {

     

       
70
       
+
       
                            sqlx::query(

     

       
71
       
+
       
                                "INSERT INTO links (original_url, short_code, user_id) 

     

       
72
       
+
       
                                 VALUES (?1, ?2, ?3)",

     

       
73
       
+
       
                            )

     

       
74
       
+
       
                            .bind(url)

     

       
75
       
+
       
                            .bind(code)

     

       
76
       
+
       
                            .bind(1)

     

       
77
       
+
       
                            .execute(pool)

     

       
78
       
+
       
                            .await?;

     

       
79
       
+
       
                            info!("Created initial link: {} -> {} for user_id: 1", code, url);

     

       
80
       
+
       
                        } else {

     

       
81
       
+
       
                            info!("Skipped existing link: {} -> {} for user_id: 1", code, url);

     

       
82
       
+
       
                        }

     

       
83
       
+
       
                    }

     

       
84
       
+
       
                }

     

       
85
       
+
       
            }

     

       
86
       
+
       
        }

     

       
87
       
+
       
    }

     

       
88
       
+
       
    Ok(())

     

       
89
       
+
       
}

     

       
90
       
+
       


     

       
91
       
+
       
async fn create_admin_user(pool: &DatabasePool, email: &str, password: &str) -> Result<()> {

     

       
92
       
+
       
    use argon2::{

     

       
93
       
+
       
        password_hash::{rand_core::OsRng, SaltString},

     

       
94
       
+
       
        Argon2, PasswordHasher,

     

       
95
       
+
       
    };

     

       
96
       
+
       


     

       
97
       
+
       
    let salt = SaltString::generate(&mut OsRng);

     

       
98
       
+
       
    let argon2 = Argon2::default();

     

       
99
       
+
       
    let password_hash = argon2

     

       
100
       
+
       
        .hash_password(password.as_bytes(), &salt)

     

       
101
       
+
       
        .map_err(|e| anyhow::anyhow!("Password hashing error: {}", e))?

     

       
102
       
+
       
        .to_string();

     

       
103
       
+
       


     

       
104
       
+
       
    match pool {

     

       
105
       
+
       
        DatabasePool::Postgres(pool) => {

     

       
106
       
+
       
            sqlx::query(

     

       
107
       
+
       
                "INSERT INTO users (email, password_hash) 

     

       
108
       
+
       
                 VALUES ($1, $2) 

     

       
109
       
+
       
                 ON CONFLICT (email) DO NOTHING",

     

       
110
       
+
       
            )

     

       
111
       
+
       
            .bind(email)

     

       
112
       
+
       
            .bind(&password_hash)

     

       
113
       
+
       
            .execute(pool)

     

       
114
       
+
       
            .await?;

     

       
115
       
+
       
        }

     

       
116
       
+
       
        DatabasePool::Sqlite(pool) => {

     

       
117
       
+
       
            sqlx::query(

     

       
118
       
+
       
                "INSERT OR IGNORE INTO users (email, password_hash) 

     

       
119
       
+
       
                 VALUES (?1, ?2)",

     

       
120
       
+
       
            )

     

       
121
       
+
       
            .bind(email)

     

       
122
       
+
       
            .bind(&password_hash)

     

       
123
       
+
       
            .execute(pool)

     

       
124
       
+
       
            .await?;

     

       
125
       
+
       
        }

     

       
126
       
+
       
    }

     

       
127
       
+
       
    info!("Created admin user: {}", email);

     

       
128
       
+
       
    Ok(())

     

       
129
       
+
       
}

     

       
130
       
+
       


     

       
131
       
 
       
#[actix_web::main]

     

       
132
       
 
       
async fn main() -> Result<()> {

     

       
133
       
 
       
    // Load environment variables from .env file

     
···

       
139
       
 
       
    // Create database connection pool

     

       
140
       
 
       
    let pool = create_db_pool().await?;

     

       
141
       
 
       
    run_migrations(&pool).await?;

     

       
142
       
+
       


     

       
143
       
+
       
    // First check if admin credentials are provided in environment variables

     

       
144
       
+
       
    let admin_credentials = match (

     

       
145
       
+
       
        std::env::var("SIMPLELINK_USER"),

     

       
146
       
+
       
        std::env::var("SIMPLELINK_PASS"),

     

       
147
       
+
       
    ) {

     

       
148
       
+
       
        (Ok(user), Ok(pass)) => Some((user, pass)),

     

       
149
       
+
       
        _ => None,

     

       
150
       
+
       
    };

     

       
151
       
+
       


     

       
152
       
+
       
    if let Some((email, password)) = admin_credentials {

     

       
153
       
+
       
        // Now check for existing users

     

       
154
       
+
       
        let user_count = match &pool {

     

       
155
       
+
       
            DatabasePool::Postgres(pool) => {

     

       
156
       
+
       
                let mut tx = pool.begin().await?;

     

       
157
       
+
       
                let count =

     

       
158
       
+
       
                    sqlx::query_as::<Postgres, (i64,)>("SELECT COUNT(*)::bigint FROM users")

     

       
159
       
+
       
                        .fetch_one(&mut *tx)

     

       
160
       
+
       
                        .await?

     

       
161
       
+
       
                        .0;

     

       
162
       
+
       
                tx.commit().await?;

     

       
163
       
+
       
                count

     

       
164
       
+
       
            }

     

       
165
       
+
       
            DatabasePool::Sqlite(pool) => {

     

       
166
       
+
       
                let mut tx = pool.begin().await?;

     

       
167
       
+
       
                let count = sqlx::query_as::<Sqlite, (i64,)>("SELECT COUNT(*) FROM users")

     

       
168
       
+
       
                    .fetch_one(&mut *tx)

     

       
169
       
+
       
                    .await?

     

       
170
       
+
       
                    .0;

     

       
171
       
+
       
                tx.commit().await?;

     

       
172
       
+
       
                count

     

       
173
       
+
       
            }

     

       
174
       
+
       
        };

     

       
175
       
+
       


     

       
176
       
+
       
        if user_count == 0 {

     

       
177
       
+
       
            info!("No users found, creating admin user: {}", email);

     

       
178
       
+
       
            match create_admin_user(&pool, &email, &password).await {

     

       
179
       
+
       
                Ok(_) => info!("Successfully created admin user"),

     

       
180
       
+
       
                Err(e) => {

     

       
181
       
+
       
                    error!("Failed to create admin user: {}", e);

     

       
182
       
+
       
                    return Err(anyhow::anyhow!("Failed to create admin user: {}", e));

     

       
183
       
+
       
                }

     

       
184
       
+
       
            }

     

       
185
       
+
       
        }

     

       
186
       
+
       
    } else {

     

       
187
       
+
       
        info!(

     

       
188
       
+
       
            "No admin credentials provided in environment variables, skipping admin user creation"

     

       
189
       
+
       
        );

     

       
190
       
+
       
    }

     

       
191
       
+
       


     

       
192
       
+
       
    // Create initial links from environment variables

     

       
193
       
+
       
    create_initial_links(&pool).await?;

     

       
194
       
 
       


     

       
195
       
 
       
    let admin_token = check_and_generate_admin_token(&pool).await?;

     

       
196
       
 
       


     

···

       
87
       
 
       
            .duration_since(UNIX_EPOCH)

     

       
88
       
 
       
            .unwrap()

     

       
89
       
 
       
            .as_secs() as usize

     

       
90
       
-
       
            + 24 * 60 * 60; // 24 hours from now

     

       
91
       
 
       


     

       
92
       
 
       
        Self { sub: user_id, exp }

     

       
93
       
 
       
    }

     

···

       
87
       
 
       
            .duration_since(UNIX_EPOCH)

     

       
88
       
 
       
            .unwrap()

     

       
89
       
 
       
            .as_secs() as usize

     

       
90
       
+
       
            + 14 * 24 * 60 * 60; // 2 weeks from now

     

       
91
       
 
       


     

       
92
       
 
       
        Self { sub: user_id, exp }

     

       
93
       
 
       
    }