nekomimi.pet / simplelink
A very performant and light (2mb in memory) link shortener and tracker. Written in Rust and React and uses Postgres/SQLite.
fork atom

Compare changes

Choose any two refs to compare.

options
unified split
Changed files
+756 -300
.github
workflows
docker-image.yml
main.yml
Cargo.lock
Cargo.toml
Dockerfile
README.md
docker-compose.yml
frontend
src
api
client.ts
components
EditModal.tsx
LinkList.tsx
StatisticsModal.tsx
types
api.ts
vite.config.ts
migrations
20250219000000_extend_short_code.sql
src
auth.rs
handlers.rs
main.rs
models.rs
+5 -15
.github/workflows/docker-image.yml 
···

       
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


     
···

       
59

       
 

       
            ${{ env.IMAGE_NAME }}


     

       
60

       
 

       
            ${{ env.REGISTRY }}/${{ github.repository }}


     

       
61

       
 

       



     

       
62

       
-

       
      - name: Build and push Docker image (amd64)


     

       
63

       
-

       
        uses: docker/build-push-action@v6


     

       
64

       
-

       
        with:


     

       
65

       
-

       
          context: .


     

       
66

       
-

       
          file: ./Dockerfile


     

       
67

       
-

       
          platforms: linux/amd64


     

       
68

       
-

       
          push: ${{ github.event_name != 'pull_request' }}


     

       
69

       
-

       
          tags: ${{ steps.meta.outputs.tags }}-amd64


     

       
70

       
-

       
          labels: ${{ steps.meta.outputs.labels }}


     

       
71

       
-

       



     

       
72

       
-

       
      - name: Build and push Docker image (arm64)


     

       
73

       
 

       
        uses: docker/build-push-action@v6


     

       
74

       
 

       
        with:


     

       
75

       
 

       
          context: .


     

       
76

       
 

       
          file: ./Dockerfile


     

       
77

       
-

       
          platforms: linux/arm64


     

       
78

       
 

       
          push: ${{ github.event_name != 'pull_request' }}


     

       
79

       
-

       
          tags: ${{ steps.meta.outputs.tags }}-arm64


     

       
80

       
 

       
          labels: ${{ steps.meta.outputs.labels }}


     
···

       
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


     
···

       
59

       
 

       
            ${{ env.IMAGE_NAME }}


     

       
60

       
 

       
            ${{ env.REGISTRY }}/${{ github.repository }}


     

       
61

       
 

       



     

       
62

       
+

       
      - name: Build and push Docker image


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
63

       
 

       
        uses: docker/build-push-action@v6


     

       
64

       
 

       
        with:


     

       
65

       
 

       
          context: .


     

       
66

       
 

       
          file: ./Dockerfile


     

       
67

       
+

       
          platforms: linux/amd64,linux/arm64


     

       
68

       
 

       
          push: ${{ github.event_name != 'pull_request' }}


     

       
69

       
+

       
          tags: ${{ steps.meta.outputs.tags }}


     

       
70

       
 

       
          labels: ${{ steps.meta.outputs.labels }}
-80
.github/workflows/main.yml 
···

       
1

       
-

       
name: Docker


     

       
2

       
-

       



     

       
3

       
-

       
on:


     

       
4

       
-

       
  schedule:


     

       
5

       
-

       
    - cron: "38 9 * * *"


     

       
6

       
-

       
  push:


     

       
7

       
-

       
    branches: ["main"]


     

       
8

       
-

       
    tags: ["v*.*.*"]


     

       
9

       
-

       
  pull_request:


     

       
10

       
-

       
    branches: ["main"]


     

       
11

       
-

       
  release:


     

       
12

       
-

       
    types: [published]


     

       
13

       
-

       



     

       
14

       
-

       
env:


     

       
15

       
-

       
  REGISTRY: ghcr.io


     

       
16

       
-

       
  IMAGE_NAME: ${{ github.repository }}


     

       
17

       
-

       



     

       
18

       
-

       
jobs:


     

       
19

       
-

       
  build:


     

       
20

       
-

       
    runs-on: macos-latest


     

       
21

       
-

       
    permissions:


     

       
22

       
-

       
      contents: read


     

       
23

       
-

       
      packages: write


     

       
24

       
-

       
      id-token: write


     

       
25

       
-

       



     

       
26

       
-

       
    steps:


     

       
27

       
-

       
      - name: Checkout repository


     

       
28

       
-

       
        uses: actions/checkout@v3


     

       
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


     

       
38

       
-

       



     

       
39

       
-

       
      - name: Log into registry ${{ env.REGISTRY }}


     

       
40

       
-

       
        if: github.event_name != 'pull_request'


     

       
41

       
-

       
        uses: docker/login-action@v3


     

       
42

       
-

       
        with:


     

       
43

       
-

       
          registry: ${{ env.REGISTRY }}


     

       
44

       
-

       
          username: ${{ github.actor }}


     

       
45

       
-

       
          password: ${{ secrets.GITHUB_TOKEN }}


     

       
46

       
-

       



     

       
47

       
-

       
      - name: Log in to Docker Hub


     

       
48

       
-

       
        if: github.event_name != 'pull_request'


     

       
49

       
-

       
        uses: docker/login-action@v3


     

       
50

       
-

       
        with:


     

       
51

       
-

       
          username: ${{ secrets.DOCKER_USERNAME }}


     

       
52

       
-

       
          password: ${{ secrets.DOCKER_PASSWORD }}


     

       
53

       
-

       



     

       
54

       
-

       
      - name: Extract metadata (tags, labels) for Docker


     

       
55

       
-

       
        id: meta


     

       
56

       
-

       
        uses: docker/metadata-action@v5


     

       
57

       
-

       
        with:


     

       
58

       
-

       
          images: |


     

       
59

       
-

       
            ${{ env.IMAGE_NAME }}


     

       
60

       
-

       
            ${{ env.REGISTRY }}/${{ github.repository }}


     

       
61

       
-

       



     

       
62

       
-

       
      - name: Build and push Docker image (amd64)


     

       
63

       
-

       
        uses: docker/build-push-action@v6


     

       
64

       
-

       
        with:


     

       
65

       
-

       
          context: .


     

       
66

       
-

       
          file: ./Dockerfile


     

       
67

       
-

       
          platforms: linux/amd64


     

       
68

       
-

       
          push: ${{ github.event_name != 'pull_request' }}


     

       
69

       
-

       
          tags: ${{ steps.meta.outputs.tags }}-amd64


     

       
70

       
-

       
          labels: ${{ steps.meta.outputs.labels }}


     

       
71

       
-

       



     

       
72

       
-

       
      - name: Build and push Docker image (arm64)


     

       
73

       
-

       
        uses: docker/build-push-action@v6


     

       
74

       
-

       
        with:


     

       
75

       
-

       
          context: .


     

       
76

       
-

       
          file: ./Dockerfile


     

       
77

       
-

       
          platforms: linux/arm64


     

       
78

       
-

       
          push: ${{ github.event_name != 'pull_request' }}


     

       
79

       
-

       
          tags: ${{ steps.meta.outputs.tags }}-arm64


     

       
80

       
-

       
          labels: ${{ steps.meta.outputs.labels }}


     
···
+9 -39
Cargo.lock 
···

       
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",


     

       

       

       
     

       
1448

       
 

       
]


     

       
1449

       
 

       



     

       
1450

       
 

       
[[package]]


     
···

       
1524

       
 

       
version = "1.20.2"


     

       
1525

       
 

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


     

       
1526

       
 

       
checksum = "1261fe7e33c73b354eab43b1273a57c8f967d0391e80353e51f764ac02cf6775"


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
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",


     

       

       

       
     

       
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]]


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
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

       
 

       
]


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
2741

       
 

       



     

       
2742

       
 

       
[[package]]


     

       
2743

       
 

       
name = "windows-core"
+1 -1
Cargo.toml 
···

       
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"
+2 -2
Dockerfile 
···

       
4

       
 

       
WORKDIR /usr/src/frontend


     

       
5

       
 

       



     

       
6

       
 

       
# Copy frontend files


     

       
7

       
-

       
COPY frontend/package*.json ./


     

       
8

       
 

       
RUN bun install


     

       
9

       
 

       



     

       
10

       
 

       
COPY frontend/ ./


     
···

       
57

       
 

       
# Copy static files


     

       
58

       
 

       
COPY --from=backend-builder /usr/src/app/static /app/static


     

       
59

       
 

       



     

       
60

       
-

       
# Expose the port (this is just documentation)


     

       
61

       
 

       
EXPOSE 8080


     

       
62

       
 

       



     

       
63

       
 

       
# Set default network configuration


     
···

       
4

       
 

       
WORKDIR /usr/src/frontend


     

       
5

       
 

       



     

       
6

       
 

       
# Copy frontend files


     

       
7

       
+

       
COPY frontend/package.json ./


     

       
8

       
 

       
RUN bun install


     

       
9

       
 

       



     

       
10

       
 

       
COPY frontend/ ./


     
···

       
57

       
 

       
# Copy static files


     

       
58

       
 

       
COPY --from=backend-builder /usr/src/app/static /app/static


     

       
59

       
 

       



     

       
60

       
+

       
# Expose the port


     

       
61

       
 

       
EXPOSE 8080


     

       
62

       
 

       



     

       
63

       
 

       
# Set default network configuration
+42 -8
README.md 
···

       
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

       
 

       



     

       
7

       
 

       
![StatsView](readme_img/statview.jpg)


     

       
8

       
 

       



     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
9

       
 

       
## Build


     

       
10

       
 

       



     

       
11

       
 

       
### From Source


     

       
12

       
 

       



     

       
13

       
 

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


     

       
14

       
 

       



     

       
15

       
-

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


     

       
16

       
-

       



     

       
17

       
 

       
```bash


     

       
18

       
-

       
#set api-domain to where you will be deploying the link shortener, eg: link.example.com, default is localhost:8080


     

       
19

       
 

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


     

       
20

       
 

       
./build.sh


     

       
21

       
 

       
cargo run


     

       
22

       
 

       
```


     

       
23

       
 

       



     

       
24

       
-

       
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.


     

       
25

       
-

       



     

       
26

       
-

       
Alternatively if you want a binary form


     

       
27

       
 

       



     

       
28

       
 

       
```bash


     

       
29

       
 

       
./build.sh --binary


     
···

       
34

       
 

       
### From Docker


     

       
35

       
 

       



     

       
36

       
 

       
```bash


     

       
37

       
-

       
docker build --build-arg -t simplelink .


     

       
38

       
 

       
docker run -p 8080:8080 \


     

       
39

       
 

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


     

       

       

       
     

       

       

       
     

       
40

       
 

       
    -v simplelink_data:/data \


     

       
41

       
 

       
    simplelink


     

       
42

       
 

       
```


     
···

       
44

       
 

       
### From Docker Compose


     

       
45

       
 

       



     

       
46

       
 

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


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     
···

       
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

       
 

       



     

       
7

       
 

       
![StatsView](readme_img/statview.jpg)


     

       
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

       
 

       



     

       
40

       
 

       
### From Source


     

       
41

       
 

       



     

       
42

       
 

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


     

       
43

       
 

       



     

       

       

       
     

       

       

       
     

       
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:


     

       

       

       
     

       

       

       
     

       
51

       
 

       



     

       
52

       
 

       
```bash


     

       
53

       
 

       
./build.sh --binary


     
···

       
58

       
 

       
### From Docker


     

       
59

       
 

       



     

       
60

       
 

       
```bash


     

       
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
+1 -3
docker-compose.yml 
···

       
19

       
 

       
      - shortener-network


     

       
20

       
 

       



     

       
21

       
 

       
  app:


     

       
22

       
-

       
    build:


     

       
23

       
-

       
      context: .


     

       
24

       
-

       
      dockerfile: Dockerfile


     

       
25

       
 

       
    container_name: shortener-app


     

       
26

       
 

       
    ports:


     

       
27

       
 

       
      - "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"
+6
frontend/src/api/client.ts 
···

       
58

       
 

       
	return response.data;


     

       
59

       
 

       
};


     

       
60

       
 

       



     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
61

       
 

       
export const deleteLink = async (id: number) => {


     

       
62

       
 

       
	await api.delete(`/links/${id}`);


     

       
63

       
 

       
};


     
···

       
58

       
 

       
	return response.data;


     

       
59

       
 

       
};


     

       
60

       
 

       



     

       
61

       
+

       
export const editLink = async (id: number, data: Partial<CreateLinkRequest>) => {


     

       
62

       
+

       
	const response = await api.patch<Link>(`/links/${id}`, data);


     

       
63

       
+

       
	return response.data;


     

       
64

       
+

       
};


     

       
65

       
+

       



     

       
66

       
+

       



     

       
67

       
 

       
export const deleteLink = async (id: number) => {


     

       
68

       
 

       
	await api.delete(`/links/${id}`);


     

       
69

       
 

       
};
+139
frontend/src/components/EditModal.tsx 
···

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     
···

       
1

       
+

       
// src/components/EditModal.tsx


     

       
2

       
+

       
import { useState } from 'react';


     

       
3

       
+

       
import { useForm } from 'react-hook-form';


     

       
4

       
+

       
import { zodResolver } from '@hookform/resolvers/zod';


     

       
5

       
+

       
import * as z from 'zod';


     

       
6

       
+

       
import { Link } from '../types/api';


     

       
7

       
+

       
import { editLink } from '../api/client';


     

       
8

       
+

       
import { useToast } from '@/hooks/use-toast';


     

       
9

       
+

       
import {


     

       
10

       
+

       
    Dialog,


     

       
11

       
+

       
    DialogContent,


     

       
12

       
+

       
    DialogHeader,


     

       
13

       
+

       
    DialogTitle,


     

       
14

       
+

       
    DialogFooter,


     

       
15

       
+

       
} from '@/components/ui/dialog';


     

       
16

       
+

       
import { Button } from '@/components/ui/button';


     

       
17

       
+

       
import { Input } from '@/components/ui/input';


     

       
18

       
+

       
import {


     

       
19

       
+

       
    Form,


     

       
20

       
+

       
    FormControl,


     

       
21

       
+

       
    FormField,


     

       
22

       
+

       
    FormItem,


     

       
23

       
+

       
    FormLabel,


     

       
24

       
+

       
    FormMessage,


     

       
25

       
+

       
} from '@/components/ui/form';


     

       
26

       
+

       



     

       
27

       
+

       
const formSchema = z.object({


     

       
28

       
+

       
    url: z


     

       
29

       
+

       
        .string()


     

       
30

       
+

       
        .min(1, 'URL is required')


     

       
31

       
+

       
        .url('Must be a valid URL')


     

       
32

       
+

       
        .refine((val) => val.startsWith('http://') || val.startsWith('https://'), {


     

       
33

       
+

       
            message: 'URL must start with http:// or https://',


     

       
34

       
+

       
        }),


     

       
35

       
+

       
    custom_code: z


     

       
36

       
+

       
        .string()


     

       
37

       
+

       
        .regex(/^[a-zA-Z0-9_-]{1,32}$/, {


     

       
38

       
+

       
            message:


     

       
39

       
+

       
                'Custom code must be 1-32 characters and contain only letters, numbers, underscores, and hyphens',


     

       
40

       
+

       
        })


     

       
41

       
+

       
        .optional(),


     

       
42

       
+

       
});


     

       
43

       
+

       



     

       
44

       
+

       
interface EditModalProps {


     

       
45

       
+

       
    isOpen: boolean;


     

       
46

       
+

       
    onClose: () => void;


     

       
47

       
+

       
    link: Link;


     

       
48

       
+

       
    onSuccess: () => void;


     

       
49

       
+

       
}


     

       
50

       
+

       



     

       
51

       
+

       
export function EditModal({ isOpen, onClose, link, onSuccess }: EditModalProps) {


     

       
52

       
+

       
    const [loading, setLoading] = useState(false);


     

       
53

       
+

       
    const { toast } = useToast();


     

       
54

       
+

       



     

       
55

       
+

       
    const form = useForm<z.infer<typeof formSchema>>({


     

       
56

       
+

       
        resolver: zodResolver(formSchema),


     

       
57

       
+

       
        defaultValues: {


     

       
58

       
+

       
            url: link.original_url,


     

       
59

       
+

       
            custom_code: link.short_code,


     

       
60

       
+

       
        },


     

       
61

       
+

       
    });


     

       
62

       
+

       



     

       
63

       
+

       
    const onSubmit = async (values: z.infer<typeof formSchema>) => {


     

       
64

       
+

       
        try {


     

       
65

       
+

       
            setLoading(true);


     

       
66

       
+

       
            await editLink(link.id, values);


     

       
67

       
+

       
            toast({


     

       
68

       
+

       
                description: 'Link updated successfully',


     

       
69

       
+

       
            });


     

       
70

       
+

       
            onSuccess();


     

       
71

       
+

       
            onClose();


     

       
72

       
+

       
        } catch (err: unknown) {


     

       
73

       
+

       
            const error = err as { response?: { data?: { error?: string } } };


     

       
74

       
+

       
            toast({


     

       
75

       
+

       
                variant: 'destructive',


     

       
76

       
+

       
                title: 'Error',


     

       
77

       
+

       
                description: error.response?.data?.error || 'Failed to update link',


     

       
78

       
+

       
            });


     

       
79

       
+

       
        } finally {


     

       
80

       
+

       
            setLoading(false);


     

       
81

       
+

       
        }


     

       
82

       
+

       
    };


     

       
83

       
+

       



     

       
84

       
+

       
    return (


     

       
85

       
+

       
        <Dialog open={isOpen} onOpenChange={onClose}>


     

       
86

       
+

       
            <DialogContent>


     

       
87

       
+

       
                <DialogHeader>


     

       
88

       
+

       
                    <DialogTitle>Edit Link</DialogTitle>


     

       
89

       
+

       
                </DialogHeader>


     

       
90

       
+

       



     

       
91

       
+

       
                <Form {...form}>


     

       
92

       
+

       
                    <form onSubmit={form.handleSubmit(onSubmit)} className="space-y-4">


     

       
93

       
+

       
                        <FormField


     

       
94

       
+

       
                            control={form.control}


     

       
95

       
+

       
                            name="url"


     

       
96

       
+

       
                            render={({ field }) => (


     

       
97

       
+

       
                                <FormItem>


     

       
98

       
+

       
                                    <FormLabel>Destination URL</FormLabel>


     

       
99

       
+

       
                                    <FormControl>


     

       
100

       
+

       
                                        <Input placeholder="https://example.com" {...field} />


     

       
101

       
+

       
                                    </FormControl>


     

       
102

       
+

       
                                    <FormMessage />


     

       
103

       
+

       
                                </FormItem>


     

       
104

       
+

       
                            )}


     

       
105

       
+

       
                        />


     

       
106

       
+

       



     

       
107

       
+

       
                        <FormField


     

       
108

       
+

       
                            control={form.control}


     

       
109

       
+

       
                            name="custom_code"


     

       
110

       
+

       
                            render={({ field }) => (


     

       
111

       
+

       
                                <FormItem>


     

       
112

       
+

       
                                    <FormLabel>Short Code</FormLabel>


     

       
113

       
+

       
                                    <FormControl>


     

       
114

       
+

       
                                        <Input placeholder="custom-code" {...field} />


     

       
115

       
+

       
                                    </FormControl>


     

       
116

       
+

       
                                    <FormMessage />


     

       
117

       
+

       
                                </FormItem>


     

       
118

       
+

       
                            )}


     

       
119

       
+

       
                        />


     

       
120

       
+

       



     

       
121

       
+

       
                        <DialogFooter>


     

       
122

       
+

       
                            <Button


     

       
123

       
+

       
                                type="button"


     

       
124

       
+

       
                                variant="outline"


     

       
125

       
+

       
                                onClick={onClose}


     

       
126

       
+

       
                                disabled={loading}


     

       
127

       
+

       
                            >


     

       
128

       
+

       
                                Cancel


     

       
129

       
+

       
                            </Button>


     

       
130

       
+

       
                            <Button type="submit" disabled={loading}>


     

       
131

       
+

       
                                {loading ? 'Saving...' : 'Save Changes'}


     

       
132

       
+

       
                            </Button>


     

       
133

       
+

       
                        </DialogFooter>


     

       
134

       
+

       
                    </form>


     

       
135

       
+

       
                </Form>


     

       
136

       
+

       
            </DialogContent>


     

       
137

       
+

       
        </Dialog>


     

       
138

       
+

       
    );


     

       
139

       
+

       
}
+44 -13
frontend/src/components/LinkList.tsx 
···

       
1

       
-

       
import { useEffect, useState } from 'react'


     

       
2

       
 

       
import { Link } from '../types/api'


     

       
3

       
 

       
import { getAllLinks, deleteLink } from '../api/client'


     

       
4

       
 

       
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card"


     
···

       
12

       
 

       
} from "@/components/ui/table"


     

       
13

       
 

       
import { Button } from "@/components/ui/button"


     

       
14

       
 

       
import { useToast } from "@/hooks/use-toast"


     

       
15

       
-

       
import { Copy, Trash2, BarChart2 } from "lucide-react"


     

       
16

       
 

       
import {


     

       
17

       
 

       
	Dialog,


     

       
18

       
 

       
	DialogContent,


     
···

       
23

       
 

       
} from "@/components/ui/dialog"


     

       
24

       
 

       



     

       
25

       
 

       
import { StatisticsModal } from "./StatisticsModal"


     

       

       

       
     

       
26

       
 

       



     

       
27

       
 

       
interface LinkListProps {


     

       
28

       
 

       
	refresh?: number;


     
···

       
39

       
 

       
		isOpen: false,


     

       
40

       
 

       
		linkId: null,


     

       
41

       
 

       
	});


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
42

       
 

       
	const { toast } = useToast()


     

       
43

       
 

       



     

       
44

       
-

       
	const fetchLinks = async () => {


     

       
45

       
 

       
		try {


     

       
46

       
 

       
			setLoading(true)


     

       
47

       
 

       
			const data = await getAllLinks()


     

       
48

       
 

       
			setLinks(data)


     

       
49

       
-

       
		} catch (err) {


     

       

       

       
     

       
50

       
 

       
			toast({


     

       
51

       
 

       
				title: "Error",


     

       
52

       
-

       
				description: "Failed to load links",


     

       
53

       
 

       
				variant: "destructive",


     

       
54

       
 

       
			})


     

       
55

       
 

       
		} finally {


     

       
56

       
 

       
			setLoading(false)


     

       
57

       
 

       
		}


     

       
58

       
-

       
	}


     

       
59

       
 

       



     

       
60

       
 

       
	useEffect(() => {


     

       
61

       
 

       
		fetchLinks()


     

       
62

       
-

       
	}, [refresh]) // Re-fetch when refresh counter changes


     

       
63

       
 

       



     

       
64

       
 

       
	const handleDelete = async () => {


     

       
65

       
 

       
		if (!deleteModal.linkId) return


     
···

       
71

       
 

       
			toast({


     

       
72

       
 

       
				description: "Link deleted successfully",


     

       
73

       
 

       
			})


     

       
74

       
-

       
		} catch (err) {


     

       

       

       
     

       
75

       
 

       
			toast({


     

       
76

       
 

       
				title: "Error",


     

       
77

       
-

       
				description: "Failed to delete link",


     

       
78

       
 

       
				variant: "destructive",


     

       
79

       
 

       
			})


     

       
80

       
 

       
		}


     
···

       
85

       
 

       
		const baseUrl = window.location.origin


     

       
86

       
 

       
		navigator.clipboard.writeText(`${baseUrl}/${shortCode}`)


     

       
87

       
 

       
		toast({


     

       
88

       
-

       
		  description: "Link copied to clipboard",


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
89

       
 

       
		})


     

       
90

       
 

       
	}


     

       
91

       
 

       



     
···

       
121

       
 

       
				</CardHeader>


     

       
122

       
 

       
				<CardContent>


     

       
123

       
 

       
					<div className="rounded-md border">


     

       

       

       
     

       
124

       
 

       
						<Table>


     

       
125

       
 

       
							<TableHeader>


     

       
126

       
 

       
								<TableRow>


     
···

       
128

       
 

       
									<TableHead className="hidden md:table-cell">Original URL</TableHead>


     

       
129

       
 

       
									<TableHead>Clicks</TableHead>


     

       
130

       
 

       
									<TableHead className="hidden md:table-cell">Created</TableHead>


     

       
131

       
-

       
									<TableHead>Actions</TableHead>


     

       
132

       
 

       
								</TableRow>


     

       
133

       
 

       
							</TableHeader>


     

       
134

       
 

       
							<TableBody>


     
···

       
142

       
 

       
										<TableCell className="hidden md:table-cell">


     

       
143

       
 

       
											{new Date(link.created_at).toLocaleDateString()}


     

       
144

       
 

       
										</TableCell>


     

       
145

       
-

       
										<TableCell>


     

       
146

       
-

       
											<div className="flex gap-2">


     

       
147

       
 

       
												<Button


     

       
148

       
 

       
													variant="ghost"


     

       
149

       
 

       
													size="icon"


     
···

       
165

       
 

       
												<Button


     

       
166

       
 

       
													variant="ghost"


     

       
167

       
 

       
													size="icon"


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
168

       
 

       
													className="h-8 w-8 text-destructive"


     

       
169

       
 

       
													onClick={() => setDeleteModal({ isOpen: true, linkId: link.id })}


     

       
170

       
 

       
												>


     
···

       
185

       
 

       
				onClose={() => setStatsModal({ isOpen: false, linkId: null })}


     

       
186

       
 

       
				linkId={statsModal.linkId!}


     

       
187

       
 

       
			/>


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
188

       
 

       
		</>


     

       
189

       
 

       
	)


     

       
190

       
 

       
}

     
···

       
1

       
+

       
import { useCallback, useEffect, useState } from 'react'


     

       
2

       
 

       
import { Link } from '../types/api'


     

       
3

       
 

       
import { getAllLinks, deleteLink } from '../api/client'


     

       
4

       
 

       
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card"


     
···

       
12

       
 

       
} from "@/components/ui/table"


     

       
13

       
 

       
import { Button } from "@/components/ui/button"


     

       
14

       
 

       
import { useToast } from "@/hooks/use-toast"


     

       
15

       
+

       
import { Copy, Trash2, BarChart2, Pencil } from "lucide-react"


     

       
16

       
 

       
import {


     

       
17

       
 

       
	Dialog,


     

       
18

       
 

       
	DialogContent,


     
···

       
23

       
 

       
} from "@/components/ui/dialog"


     

       
24

       
 

       



     

       
25

       
 

       
import { StatisticsModal } from "./StatisticsModal"


     

       
26

       
+

       
import { EditModal } from './EditModal'


     

       
27

       
 

       



     

       
28

       
 

       
interface LinkListProps {


     

       
29

       
 

       
	refresh?: number;


     
···

       
40

       
 

       
		isOpen: false,


     

       
41

       
 

       
		linkId: null,


     

       
42

       
 

       
	});


     

       
43

       
+

       
	const [editModal, setEditModal] = useState<{ isOpen: boolean; link: Link | null }>({


     

       
44

       
+

       
		isOpen: false,


     

       
45

       
+

       
		link: null,


     

       
46

       
+

       
	});


     

       
47

       
 

       
	const { toast } = useToast()


     

       
48

       
 

       



     

       
49

       
+

       
	const fetchLinks = useCallback(async () => {


     

       
50

       
 

       
		try {


     

       
51

       
 

       
			setLoading(true)


     

       
52

       
 

       
			const data = await getAllLinks()


     

       
53

       
 

       
			setLinks(data)


     

       
54

       
+

       
		} catch (err: unknown) {


     

       
55

       
+

       
			const errorMessage = err instanceof Error ? err.message : 'Unknown error occurred';


     

       
56

       
 

       
			toast({


     

       
57

       
 

       
				title: "Error",


     

       
58

       
+

       
				description: `Failed to load links: ${errorMessage}`,


     

       
59

       
 

       
				variant: "destructive",


     

       
60

       
 

       
			})


     

       
61

       
 

       
		} finally {


     

       
62

       
 

       
			setLoading(false)


     

       
63

       
 

       
		}


     

       
64

       
+

       
	}, [toast, setLinks, setLoading])


     

       
65

       
 

       



     

       
66

       
 

       
	useEffect(() => {


     

       
67

       
 

       
		fetchLinks()


     

       
68

       
+

       
	}, [fetchLinks, refresh]) // Re-fetch when refresh counter changes


     

       
69

       
 

       



     

       
70

       
 

       
	const handleDelete = async () => {


     

       
71

       
 

       
		if (!deleteModal.linkId) return


     
···

       
77

       
 

       
			toast({


     

       
78

       
 

       
				description: "Link deleted successfully",


     

       
79

       
 

       
			})


     

       
80

       
+

       
		} catch (err: unknown) {


     

       
81

       
+

       
			const errorMessage = err instanceof Error ? err.message : 'Unknown error occurred';


     

       
82

       
 

       
			toast({


     

       
83

       
 

       
				title: "Error",


     

       
84

       
+

       
				description: `Failed to delete link: ${errorMessage}`,


     

       
85

       
 

       
				variant: "destructive",


     

       
86

       
 

       
			})


     

       
87

       
 

       
		}


     
···

       
92

       
 

       
		const baseUrl = window.location.origin


     

       
93

       
 

       
		navigator.clipboard.writeText(`${baseUrl}/${shortCode}`)


     

       
94

       
 

       
		toast({


     

       
95

       
+

       
			description: (


     

       
96

       
+

       
				<>


     

       
97

       
+

       
					Link copied to clipboard


     

       
98

       
+

       
					<br />


     

       
99

       
+

       
					You can add ?source=TextHere to the end of the link to track the source of clicks


     

       
100

       
+

       
				</>


     

       
101

       
+

       
			),


     

       
102

       
 

       
		})


     

       
103

       
 

       
	}


     

       
104

       
 

       



     
···

       
134

       
 

       
				</CardHeader>


     

       
135

       
 

       
				<CardContent>


     

       
136

       
 

       
					<div className="rounded-md border">


     

       
137

       
+

       



     

       
138

       
 

       
						<Table>


     

       
139

       
 

       
							<TableHeader>


     

       
140

       
 

       
								<TableRow>


     
···

       
142

       
 

       
									<TableHead className="hidden md:table-cell">Original URL</TableHead>


     

       
143

       
 

       
									<TableHead>Clicks</TableHead>


     

       
144

       
 

       
									<TableHead className="hidden md:table-cell">Created</TableHead>


     

       
145

       
+

       
									<TableHead className="w-[1%] whitespace-nowrap pr-4">Actions</TableHead>


     

       
146

       
 

       
								</TableRow>


     

       
147

       
 

       
							</TableHeader>


     

       
148

       
 

       
							<TableBody>


     
···

       
156

       
 

       
										<TableCell className="hidden md:table-cell">


     

       
157

       
 

       
											{new Date(link.created_at).toLocaleDateString()}


     

       
158

       
 

       
										</TableCell>


     

       
159

       
+

       
										<TableCell className="p-2 pr-4">


     

       
160

       
+

       
											<div className="flex items-center gap-1">


     

       
161

       
 

       
												<Button


     

       
162

       
 

       
													variant="ghost"


     

       
163

       
 

       
													size="icon"


     
···

       
179

       
 

       
												<Button


     

       
180

       
 

       
													variant="ghost"


     

       
181

       
 

       
													size="icon"


     

       
182

       
+

       
													className="h-8 w-8"


     

       
183

       
+

       
													onClick={() => setEditModal({ isOpen: true, link })}


     

       
184

       
+

       
												>


     

       
185

       
+

       
													<Pencil className="h-4 w-4" />


     

       
186

       
+

       
													<span className="sr-only">Edit Link</span>


     

       
187

       
+

       
												</Button>


     

       
188

       
+

       
												<Button


     

       
189

       
+

       
													variant="ghost"


     

       
190

       
+

       
													size="icon"


     

       
191

       
 

       
													className="h-8 w-8 text-destructive"


     

       
192

       
 

       
													onClick={() => setDeleteModal({ isOpen: true, linkId: link.id })}


     

       
193

       
 

       
												>


     
···

       
208

       
 

       
				onClose={() => setStatsModal({ isOpen: false, linkId: null })}


     

       
209

       
 

       
				linkId={statsModal.linkId!}


     

       
210

       
 

       
			/>


     

       
211

       
+

       
			{editModal.link && (


     

       
212

       
+

       
				<EditModal


     

       
213

       
+

       
					isOpen={editModal.isOpen}


     

       
214

       
+

       
					onClose={() => setEditModal({ isOpen: false, link: null })}


     

       
215

       
+

       
					link={editModal.link}


     

       
216

       
+

       
					onSuccess={fetchLinks}


     

       
217

       
+

       
				/>


     

       
218

       
+

       
			)}


     

       
219

       
 

       
		</>


     

       
220

       
 

       
	)


     

       
221

       
 

       
}
+161 -105
frontend/src/components/StatisticsModal.tsx 
···

       
1

       
 

       
import { Dialog, DialogContent, DialogHeader, DialogTitle } from "@/components/ui/dialog";


     

       
2

       
 

       
import {


     

       
3

       
-

       
    LineChart,


     

       
4

       
-

       
    Line,


     

       
5

       
-

       
    XAxis,


     

       
6

       
-

       
    YAxis,


     

       
7

       
-

       
    CartesianGrid,


     

       
8

       
-

       
    Tooltip,


     

       
9

       
-

       
    ResponsiveContainer,


     

       
10

       
 

       
} from "recharts";


     

       
11

       
 

       
import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card";


     

       
12

       
-

       
import { toast } from "@/hooks/use-toast"


     

       
13

       
-

       
import { useState, useEffect } from "react";


     

       
14

       
 

       



     

       
15

       
-

       
import { getLinkClickStats, getLinkSourceStats } from '../api/client';


     

       
16

       
-

       
import { ClickStats, SourceStats } from '../types/api';


     

       
17

       
 

       



     

       
18

       
 

       
interface StatisticsModalProps {


     

       
19

       
-

       
    isOpen: boolean;


     

       
20

       
-

       
    onClose: () => void;


     

       
21

       
-

       
    linkId: number;


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
22

       
 

       
}


     

       
23

       
 

       



     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
24

       
 

       
export function StatisticsModal({ isOpen, onClose, linkId }: StatisticsModalProps) {


     

       
25

       
-

       
    const [clicksOverTime, setClicksOverTime] = useState<ClickStats[]>([]);


     

       
26

       
-

       
    const [sourcesData, setSourcesData] = useState<SourceStats[]>([]);


     

       
27

       
-

       
    const [loading, setLoading] = useState(true);


     

       
28

       
 

       



     

       
29

       
-

       
    useEffect(() => {


     

       
30

       
-

       
        if (isOpen && linkId) {


     

       
31

       
-

       
            const fetchData = async () => {


     

       
32

       
-

       
                try {


     

       
33

       
-

       
                    setLoading(true);


     

       
34

       
-

       
                    const [clicksData, sourcesData] = await Promise.all([


     

       
35

       
-

       
                        getLinkClickStats(linkId),


     

       
36

       
-

       
                        getLinkSourceStats(linkId),


     

       
37

       
-

       
                    ]);


     

       
38

       
-

       
                    setClicksOverTime(clicksData);


     

       
39

       
-

       
                    setSourcesData(sourcesData);


     

       
40

       
-

       
                } catch (error: any) {


     

       
41

       
-

       
                    console.error("Failed to fetch statistics:", error);


     

       
42

       
-

       
                    toast({


     

       
43

       
-

       
                        variant: "destructive",


     

       
44

       
-

       
                        title: "Error",


     

       
45

       
-

       
                        description: error.response?.data || "Failed to load statistics",


     

       
46

       
-

       
                    });


     

       
47

       
-

       
                } finally {


     

       
48

       
-

       
                    setLoading(false);


     

       
49

       
-

       
                }


     

       
50

       
-

       
            };


     

       
51

       
-

       
    


     

       
52

       
-

       
            fetchData();


     

       
53

       
-

       
        }


     

       
54

       
-

       
    }, [isOpen, linkId]);


     

       
55

       
 

       



     

       
56

       
-

       
    return (


     

       
57

       
-

       
        <Dialog open={isOpen} onOpenChange={onClose}>


     

       
58

       
-

       
            <DialogContent className="max-w-3xl">


     

       
59

       
-

       
                <DialogHeader>


     

       
60

       
-

       
                    <DialogTitle>Link Statistics</DialogTitle>


     

       
61

       
-

       
                </DialogHeader>


     

       
62

       
 

       



     

       
63

       
-

       
                {loading ? (


     

       
64

       
-

       
                    <div className="flex items-center justify-center h-64">Loading...</div>


     

       
65

       
-

       
                ) : (


     

       
66

       
-

       
                    <div className="grid gap-4">


     

       
67

       
-

       
                        <Card>


     

       
68

       
-

       
                            <CardHeader>


     

       
69

       
-

       
                                <CardTitle>Clicks Over Time</CardTitle>


     

       
70

       
-

       
                            </CardHeader>


     

       
71

       
-

       
                            <CardContent>


     

       
72

       
-

       
                                <div className="h-[300px]">


     

       
73

       
-

       
                                    <ResponsiveContainer width="100%" height="100%">


     

       
74

       
-

       
                                        <LineChart data={clicksOverTime}>


     

       
75

       
-

       
                                            <CartesianGrid strokeDasharray="3 3" />


     

       
76

       
-

       
                                            <XAxis dataKey="date" />


     

       
77

       
-

       
                                            <YAxis />


     

       
78

       
-

       
                                            <Tooltip />


     

       
79

       
-

       
                                            <Line


     

       
80

       
-

       
                                                type="monotone"


     

       
81

       
-

       
                                                dataKey="clicks"


     

       
82

       
-

       
                                                stroke="#8884d8"


     

       
83

       
-

       
                                                strokeWidth={2}


     

       
84

       
-

       
                                            />


     

       
85

       
-

       
                                        </LineChart>


     

       
86

       
-

       
                                    </ResponsiveContainer>


     

       
87

       
-

       
                                </div>


     

       
88

       
-

       
                            </CardContent>


     

       
89

       
-

       
                        </Card>


     

       
90

       
 

       



     

       
91

       
-

       
                        <Card>


     

       
92

       
-

       
                            <CardHeader>


     

       
93

       
-

       
                                <CardTitle>Top Sources</CardTitle>


     

       
94

       
-

       
                            </CardHeader>


     

       
95

       
-

       
                            <CardContent>


     

       
96

       
-

       
                                <ul className="space-y-2">


     

       
97

       
-

       
                                    {sourcesData.map((source, index) => (


     

       
98

       
-

       
                                        <li


     

       
99

       
-

       
                                            key={source.source}


     

       
100

       
-

       
                                            className="flex items-center justify-between py-2 border-b last:border-0"


     

       
101

       
-

       
                                        >


     

       
102

       
-

       
                                            <span className="text-sm">


     

       
103

       
-

       
                                                <span className="font-medium text-muted-foreground mr-2">


     

       
104

       
-

       
                                                    {index + 1}.


     

       
105

       
-

       
                                                </span>


     

       
106

       
-

       
                                                {source.source}


     

       
107

       
-

       
                                            </span>


     

       
108

       
-

       
                                            <span className="text-sm font-medium">


     

       
109

       
-

       
                                                {source.count} clicks


     

       
110

       
-

       
                                            </span>


     

       
111

       
-

       
                                        </li>


     

       
112

       
-

       
                                    ))}


     

       
113

       
-

       
                                </ul>


     

       
114

       
-

       
                            </CardContent>


     

       
115

       
-

       
                        </Card>


     

       
116

       
-

       
                    </div>


     

       
117

       
-

       
                )}


     

       
118

       
-

       
            </DialogContent>


     

       
119

       
-

       
        </Dialog>


     

       
120

       
 

       
    );


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
121

       
 

       
}


     
···

       
1

       
 

       
import { Dialog, DialogContent, DialogHeader, DialogTitle } from "@/components/ui/dialog";


     

       
2

       
 

       
import {


     

       
3

       
+

       
  LineChart,


     

       
4

       
+

       
  Line,


     

       
5

       
+

       
  XAxis,


     

       
6

       
+

       
  YAxis,


     

       
7

       
+

       
  CartesianGrid,


     

       
8

       
+

       
  Tooltip,


     

       
9

       
+

       
  ResponsiveContainer,


     

       
10

       
 

       
} from "recharts";


     

       
11

       
 

       
import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card";


     

       
12

       
+

       
import { toast } from "@/hooks/use-toast";


     

       
13

       
+

       
import { useState, useEffect, useMemo } from "react";


     

       
14

       
 

       



     

       
15

       
+

       
import { getLinkClickStats, getLinkSourceStats } from "../api/client";


     

       
16

       
+

       
import { ClickStats, SourceStats } from "../types/api";


     

       
17

       
 

       



     

       
18

       
 

       
interface StatisticsModalProps {


     

       
19

       
+

       
  isOpen: boolean;


     

       
20

       
+

       
  onClose: () => void;


     

       
21

       
+

       
  linkId: number;


     

       
22

       
+

       
}


     

       
23

       
+

       



     

       
24

       
+

       
interface EnhancedClickStats extends ClickStats {


     

       
25

       
+

       
  sources?: { source: string; count: number }[];


     

       
26

       
 

       
}


     

       
27

       
 

       



     

       
28

       
+

       
const CustomTooltip = ({


     

       
29

       
+

       
  active,


     

       
30

       
+

       
  payload,


     

       
31

       
+

       
  label,


     

       
32

       
+

       
}: {


     

       
33

       
+

       
  active?: boolean;


     

       
34

       
+

       
  payload?: { value: number; payload: EnhancedClickStats }[];


     

       
35

       
+

       
  label?: string;


     

       
36

       
+

       
}) => {


     

       
37

       
+

       
  if (active && payload && payload.length > 0) {


     

       
38

       
+

       
    const data = payload[0].payload;


     

       
39

       
+

       
    return (


     

       
40

       
+

       
      <div className="bg-background text-foreground p-4 rounded-lg shadow-lg border">


     

       
41

       
+

       
        <p className="font-medium">{label}</p>


     

       
42

       
+

       
        <p className="text-sm">Clicks: {data.clicks}</p>


     

       
43

       
+

       
        {data.sources && data.sources.length > 0 && (


     

       
44

       
+

       
          <div className="mt-2">


     

       
45

       
+

       
            <p className="font-medium text-sm">Sources:</p>


     

       
46

       
+

       
            <ul className="text-sm">


     

       
47

       
+

       
              {data.sources.map((source: { source: string; count: number }) => (


     

       
48

       
+

       
                <li key={source.source}>


     

       
49

       
+

       
                  {source.source}: {source.count}


     

       
50

       
+

       
                </li>


     

       
51

       
+

       
              ))}


     

       
52

       
+

       
            </ul>


     

       
53

       
+

       
          </div>


     

       
54

       
+

       
        )}


     

       
55

       
+

       
      </div>


     

       
56

       
+

       
    );


     

       
57

       
+

       
  }


     

       
58

       
+

       
  return null;


     

       
59

       
+

       
};


     

       
60

       
+

       



     

       
61

       
 

       
export function StatisticsModal({ isOpen, onClose, linkId }: StatisticsModalProps) {


     

       
62

       
+

       
  const [clicksOverTime, setClicksOverTime] = useState<EnhancedClickStats[]>([]);


     

       
63

       
+

       
  const [sourcesData, setSourcesData] = useState<SourceStats[]>([]);


     

       
64

       
+

       
  const [loading, setLoading] = useState(true);


     

       
65

       
 

       



     

       
66

       
+

       
  useEffect(() => {


     

       
67

       
+

       
    if (isOpen && linkId) {


     

       
68

       
+

       
      const fetchData = async () => {


     

       
69

       
+

       
        try {


     

       
70

       
+

       
          setLoading(true);


     

       
71

       
+

       
          const [clicksData, sourcesData] = await Promise.all([


     

       
72

       
+

       
            getLinkClickStats(linkId),


     

       
73

       
+

       
            getLinkSourceStats(linkId),


     

       
74

       
+

       
          ]);


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
75

       
 

       



     

       
76

       
+

       
          // Enhance clicks data with source information


     

       
77

       
+

       
          const enhancedClicksData = clicksData.map((clickData) => ({


     

       
78

       
+

       
            ...clickData,


     

       
79

       
+

       
            sources: sourcesData.filter((source) => source.date === clickData.date),


     

       
80

       
+

       
          }));


     

       

       

       
     

       
81

       
 

       



     

       
82

       
+

       
          setClicksOverTime(enhancedClicksData);


     

       
83

       
+

       
          setSourcesData(sourcesData);


     

       
84

       
+

       
        } catch (error: unknown) {


     

       
85

       
+

       
          console.error("Failed to fetch statistics:", error);


     

       
86

       
+

       
          toast({


     

       
87

       
+

       
            variant: "destructive",


     

       
88

       
+

       
            title: "Error",


     

       
89

       
+

       
            description: error instanceof Error ? error.message : "Failed to load statistics",


     

       
90

       
+

       
          });


     

       
91

       
+

       
        } finally {


     

       
92

       
+

       
          setLoading(false);


     

       
93

       
+

       
        }


     

       
94

       
+

       
      };


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
95

       
 

       



     

       
96

       
+

       
      fetchData();


     

       
97

       
+

       
    }


     

       
98

       
+

       
  }, [isOpen, linkId]);


     

       
99

       
+

       



     

       
100

       
+

       
  const aggregatedSources = useMemo(() => {


     

       
101

       
+

       
    const sourceMap = sourcesData.reduce<Record<string, number>>(


     

       
102

       
+

       
      (acc, { source, count }) => ({


     

       
103

       
+

       
        ...acc,


     

       
104

       
+

       
        [source]: (acc[source] || 0) + count


     

       
105

       
+

       
      }),


     

       
106

       
+

       
      {}


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
107

       
 

       
    );


     

       
108

       
+

       



     

       
109

       
+

       
    return Object.entries(sourceMap)


     

       
110

       
+

       
      .map(([source, count]) => ({ source, count }))


     

       
111

       
+

       
      .sort((a, b) => b.count - a.count);


     

       
112

       
+

       
  }, [sourcesData]);


     

       
113

       
+

       



     

       
114

       
+

       
  return (


     

       
115

       
+

       
    <Dialog open={isOpen} onOpenChange={onClose}>


     

       
116

       
+

       
      <DialogContent className="max-w-3xl">


     

       
117

       
+

       
        <DialogHeader>


     

       
118

       
+

       
          <DialogTitle>Link Statistics</DialogTitle>


     

       
119

       
+

       
        </DialogHeader>


     

       
120

       
+

       



     

       
121

       
+

       
        {loading ? (


     

       
122

       
+

       
          <div className="flex items-center justify-center h-64">Loading...</div>


     

       
123

       
+

       
        ) : (


     

       
124

       
+

       
          <div className="grid gap-4">


     

       
125

       
+

       
            <Card>


     

       
126

       
+

       
              <CardHeader>


     

       
127

       
+

       
                <CardTitle>Clicks Over Time</CardTitle>


     

       
128

       
+

       
              </CardHeader>


     

       
129

       
+

       
              <CardContent>


     

       
130

       
+

       
                <div className="h-[300px]">


     

       
131

       
+

       
                  <ResponsiveContainer width="100%" height="100%">


     

       
132

       
+

       
                    <LineChart data={clicksOverTime}>


     

       
133

       
+

       
                      <CartesianGrid strokeDasharray="3 3" />


     

       
134

       
+

       
                      <XAxis dataKey="date" />


     

       
135

       
+

       
                      <YAxis />


     

       
136

       
+

       
                      <Tooltip content={<CustomTooltip />} />


     

       
137

       
+

       
                      <Line


     

       
138

       
+

       
                        type="monotone"


     

       
139

       
+

       
                        dataKey="clicks"


     

       
140

       
+

       
                        stroke="#8884d8"


     

       
141

       
+

       
                        strokeWidth={2}


     

       
142

       
+

       
                      />


     

       
143

       
+

       
                    </LineChart>


     

       
144

       
+

       
                  </ResponsiveContainer>


     

       
145

       
+

       
                </div>


     

       
146

       
+

       
              </CardContent>


     

       
147

       
+

       
            </Card>


     

       
148

       
+

       



     

       
149

       
+

       
            <Card>


     

       
150

       
+

       
              <CardHeader>


     

       
151

       
+

       
                <CardTitle>Top Sources</CardTitle>


     

       
152

       
+

       
              </CardHeader>


     

       
153

       
+

       
              <CardContent>


     

       
154

       
+

       
                <ul className="space-y-2">


     

       
155

       
+

       
                  {aggregatedSources.map((source, index) => (


     

       
156

       
+

       
                    <li


     

       
157

       
+

       
                      key={source.source}


     

       
158

       
+

       
                      className="flex items-center justify-between py-2 border-b last:border-0"


     

       
159

       
+

       
                    >


     

       
160

       
+

       
                      <span className="text-sm">


     

       
161

       
+

       
                        <span className="font-medium text-muted-foreground mr-2">


     

       
162

       
+

       
                          {index + 1}.


     

       
163

       
+

       
                        </span>


     

       
164

       
+

       
                        {source.source}


     

       
165

       
+

       
                      </span>


     

       
166

       
+

       
                      <span className="text-sm font-medium">{source.count} clicks</span>


     

       
167

       
+

       
                    </li>


     

       
168

       
+

       
                  ))}


     

       
169

       
+

       
                </ul>


     

       
170

       
+

       
              </CardContent>


     

       
171

       
+

       
            </Card>


     

       
172

       
+

       
          </div>


     

       
173

       
+

       
        )}


     

       
174

       
+

       
      </DialogContent>


     

       
175

       
+

       
    </Dialog>


     

       
176

       
+

       
  );


     

       
177

       
 

       
}
+1
frontend/src/types/api.ts 
···

       
32

       
 

       
}


     

       
33

       
 

       



     

       
34

       
 

       
export interface SourceStats {


     

       

       

       
     

       
35

       
 

       
	source: string;


     

       
36

       
 

       
	count: number;


     

       
37

       
 

       
}


     
···

       
32

       
 

       
}


     

       
33

       
 

       



     

       
34

       
 

       
export interface SourceStats {


     

       
35

       
+

       
	date: string;


     

       
36

       
 

       
	source: string;


     

       
37

       
 

       
	count: number;


     

       
38

       
 

       
}
+28 -15
frontend/vite.config.ts 
···

       
3

       
 

       
import tailwindcss from '@tailwindcss/vite'


     

       
4

       
 

       
import path from "path"


     

       
5

       
 

       



     

       
6

       
-

       
export default defineConfig(() => ({


     

       
7

       
-

       
	plugins: [react(), tailwindcss()],


     

       
8

       
-

       
	/*server: {


     

       
9

       
-

       
		proxy: {


     

       
10

       
-

       
			'/api': {


     

       
11

       
-

       
				target: process.env.VITE_API_URL || 'http://localhost:8080',


     

       
12

       
-

       
				changeOrigin: true,


     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
13

       
 

       
			},


     

       
14

       
-

       
		},


     

       
15

       
-

       
	},*/


     

       
16

       
-

       
	resolve: {


     

       
17

       
-

       
		alias: {


     

       
18

       
-

       
			"@": path.resolve(__dirname, "./src"),


     

       
19

       
-

       
		},


     

       
20

       
-

       
	},


     

       
21

       
-

       
}))


     
···

       
3

       
 

       
import tailwindcss from '@tailwindcss/vite'


     

       
4

       
 

       
import path from "path"


     

       
5

       
 

       



     

       
6

       
+

       
export default defineConfig(({ command }) => {


     

       
7

       
+

       
	if (command === 'serve') { //command == 'dev'


     

       
8

       
+

       
		return {


     

       
9

       
+

       
			server: {


     

       
10

       
+

       
				proxy: {


     

       
11

       
+

       
					'/api': {


     

       
12

       
+

       
						target: process.env.VITE_API_URL || 'http://localhost:8080',


     

       
13

       
+

       
						changeOrigin: true,


     

       
14

       
+

       
					},


     

       
15

       
+

       
				},


     

       
16

       
+

       
			},


     

       
17

       
+

       
			plugins: [react(), tailwindcss()],


     

       
18

       
+

       
			resolve: {


     

       
19

       
+

       
				alias: {


     

       
20

       
+

       
					"@": path.resolve(__dirname, "./src"),


     

       
21

       
+

       
				},


     

       
22

       
+

       
			},


     

       
23

       
+

       
		}


     

       
24

       
+

       
	} else { //command === 'build'


     

       
25

       
+

       
		return {


     

       
26

       
+

       
			plugins: [react(), tailwindcss()],


     

       
27

       
+

       
			resolve: {


     

       
28

       
+

       
				alias: {


     

       
29

       
+

       
					"@": path.resolve(__dirname, "./src"),


     

       
30

       
+

       
				},


     

       
31

       
 

       
			},


     

       
32

       
+

       
		}


     

       
33

       
+

       
	}


     

       
34

       
+

       
})
+3
migrations/20250219000000_extend_short_code.sql 
···

       

       

       
     

       

       

       
     

       

       

       
     
···

       
1

       
+

       
-- PostgreSQL migration


     

       
2

       
+

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


     

       
3

       
+
+8 -7
src/auth.rs 
···

       

       

       
     

       
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()


     

       

       

       
     

       
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

       
-

       
}

     

       

       

       
     
···

       
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};


     

       

       

       
     

       
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

       
 

       
        }


     

       

       

       
     

       
39

       
 

       
        ready(Err(AppError::Unauthorized))


     

       
40

       
 

       
    }


     

       
41

       
+

       
}


     

       
42

       
+
+145 -10
src/handlers.rs 
···

       
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

       
 

       
    }


     
···

       
457

       
 

       
    }))


     

       
458

       
 

       
}


     

       
459

       
 

       



     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
460

       
 

       
pub async fn delete_link(


     

       
461

       
 

       
    state: web::Data<AppState>,


     

       
462

       
 

       
    user: AuthenticatedUser,


     

       
463

       
 

       
    path: web::Path<i32>,


     

       
464

       
 

       
) -> Result<impl Responder, AppError> {


     

       
465

       
-

       
    let link_id = path.into_inner();


     

       
466

       
 

       



     

       
467

       
 

       
    match &state.db {


     

       
468

       
 

       
        DatabasePool::Postgres(pool) => {


     
···

       
570

       
 

       
                WHERE link_id = $1


     

       
571

       
 

       
                GROUP BY DATE(created_at)


     

       
572

       
 

       
                ORDER BY DATE(created_at) ASC


     

       
573

       
-

       
                LIMIT 30


     

       
574

       
 

       
                "#,


     

       
575

       
 

       
            )


     

       
576

       
 

       
            .bind(link_id)


     
···

       
587

       
 

       
                WHERE link_id = ?


     

       
588

       
 

       
                GROUP BY DATE(created_at)


     

       
589

       
 

       
                ORDER BY DATE(created_at) ASC


     

       
590

       
-

       
                LIMIT 30


     

       
591

       
 

       
                "#,


     

       
592

       
 

       
            )


     

       
593

       
 

       
            .bind(link_id)


     
···

       
643

       
 

       
            sqlx::query_as::<_, SourceStats>(


     

       
644

       
 

       
                r#"


     

       
645

       
 

       
                SELECT 


     

       

       

       
     

       
646

       
 

       
                    query_source as source,


     

       
647

       
 

       
                    COUNT(*)::bigint as count


     

       
648

       
 

       
                FROM clicks


     

       
649

       
 

       
                WHERE link_id = $1


     

       
650

       
 

       
                    AND query_source IS NOT NULL


     

       
651

       
 

       
                    AND query_source != ''


     

       
652

       
-

       
                GROUP BY query_source


     

       
653

       
-

       
                ORDER BY COUNT(*) DESC


     

       
654

       
-

       
                LIMIT 10


     

       
655

       
 

       
                "#,


     

       
656

       
 

       
            )


     

       
657

       
 

       
            .bind(link_id)


     
···

       
662

       
 

       
            sqlx::query_as::<_, SourceStats>(


     

       
663

       
 

       
                r#"


     

       
664

       
 

       
                SELECT 


     

       

       

       
     

       
665

       
 

       
                    query_source as source,


     

       
666

       
 

       
                    COUNT(*) as count


     

       
667

       
 

       
                FROM clicks


     

       
668

       
 

       
                WHERE link_id = ?


     

       
669

       
 

       
                    AND query_source IS NOT NULL


     

       
670

       
 

       
                    AND query_source != ''


     

       
671

       
-

       
                GROUP BY query_source


     

       
672

       
-

       
                ORDER BY COUNT(*) DESC


     

       
673

       
-

       
                LIMIT 10


     

       
674

       
 

       
                "#,


     

       
675

       
 

       
            )


     

       
676

       
 

       
            .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

       
 

       
    }


     
···

       
457

       
 

       
    }))


     

       
458

       
 

       
}


     

       
459

       
 

       



     

       
460

       
+

       
pub async fn edit_link(


     

       
461

       
+

       
    state: web::Data<AppState>,


     

       
462

       
+

       
    user: AuthenticatedUser,


     

       
463

       
+

       
    path: web::Path<i32>,


     

       
464

       
+

       
    payload: web::Json<CreateLink>,


     

       
465

       
+

       
) -> Result<impl Responder, AppError> {


     

       
466

       
+

       
    let link_id: i32 = path.into_inner();


     

       
467

       
+

       



     

       
468

       
+

       
    // Validate the new URL if provided


     

       
469

       
+

       
    validate_url(&payload.url)?;


     

       
470

       
+

       



     

       
471

       
+

       
    // Validate custom code if provided


     

       
472

       
+

       
    if let Some(ref custom_code) = payload.custom_code {


     

       
473

       
+

       
        validate_custom_code(custom_code)?;


     

       
474

       
+

       



     

       
475

       
+

       
        // Check if the custom code is already taken by another link


     

       
476

       
+

       
        let existing_link = match &state.db {


     

       
477

       
+

       
            DatabasePool::Postgres(pool) => {


     

       
478

       
+

       
                sqlx::query_as::<_, Link>("SELECT * FROM links WHERE short_code = $1 AND id != $2")


     

       
479

       
+

       
                    .bind(custom_code)


     

       
480

       
+

       
                    .bind(link_id)


     

       
481

       
+

       
                    .fetch_optional(pool)


     

       
482

       
+

       
                    .await?


     

       
483

       
+

       
            }


     

       
484

       
+

       
            DatabasePool::Sqlite(pool) => {


     

       
485

       
+

       
                sqlx::query_as::<_, Link>("SELECT * FROM links WHERE short_code = ?1 AND id != ?2")


     

       
486

       
+

       
                    .bind(custom_code)


     

       
487

       
+

       
                    .bind(link_id)


     

       
488

       
+

       
                    .fetch_optional(pool)


     

       
489

       
+

       
                    .await?


     

       
490

       
+

       
            }


     

       
491

       
+

       
        };


     

       
492

       
+

       



     

       
493

       
+

       
        if existing_link.is_some() {


     

       
494

       
+

       
            return Err(AppError::InvalidInput(


     

       
495

       
+

       
                "Custom code already taken".to_string(),


     

       
496

       
+

       
            ));


     

       
497

       
+

       
        }


     

       
498

       
+

       
    }


     

       
499

       
+

       



     

       
500

       
+

       
    // Update the link


     

       
501

       
+

       
    let updated_link = match &state.db {


     

       
502

       
+

       
        DatabasePool::Postgres(pool) => {


     

       
503

       
+

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


     

       
504

       
+

       



     

       
505

       
+

       
            // First verify the link belongs to the user


     

       
506

       
+

       
            let link =


     

       
507

       
+

       
                sqlx::query_as::<_, Link>("SELECT * FROM links WHERE id = $1 AND user_id = $2")


     

       
508

       
+

       
                    .bind(link_id)


     

       
509

       
+

       
                    .bind(user.user_id)


     

       
510

       
+

       
                    .fetch_optional(&mut *tx)


     

       
511

       
+

       
                    .await?;


     

       
512

       
+

       



     

       
513

       
+

       
            if link.is_none() {


     

       
514

       
+

       
                return Err(AppError::NotFound);


     

       
515

       
+

       
            }


     

       
516

       
+

       



     

       
517

       
+

       
            // Update the link


     

       
518

       
+

       
            let updated = sqlx::query_as::<_, Link>(


     

       
519

       
+

       
                r#"


     

       
520

       
+

       
                UPDATE links 


     

       
521

       
+

       
                SET 


     

       
522

       
+

       
                    original_url = $1,


     

       
523

       
+

       
                    short_code = COALESCE($2, short_code)


     

       
524

       
+

       
                WHERE id = $3 AND user_id = $4


     

       
525

       
+

       
                RETURNING *


     

       
526

       
+

       
                "#,


     

       
527

       
+

       
            )


     

       
528

       
+

       
            .bind(&payload.url)


     

       
529

       
+

       
            .bind(&payload.custom_code)


     

       
530

       
+

       
            .bind(link_id)


     

       
531

       
+

       
            .bind(user.user_id)


     

       
532

       
+

       
            .fetch_one(&mut *tx)


     

       
533

       
+

       
            .await?;


     

       
534

       
+

       



     

       
535

       
+

       
            // If source is provided, add a click record


     

       
536

       
+

       
            if let Some(ref source) = payload.source {


     

       
537

       
+

       
                sqlx::query("INSERT INTO clicks (link_id, source) VALUES ($1, $2)")


     

       
538

       
+

       
                    .bind(link_id)


     

       
539

       
+

       
                    .bind(source)


     

       
540

       
+

       
                    .execute(&mut *tx)


     

       
541

       
+

       
                    .await?;


     

       
542

       
+

       
            }


     

       
543

       
+

       



     

       
544

       
+

       
            tx.commit().await?;


     

       
545

       
+

       
            updated


     

       
546

       
+

       
        }


     

       
547

       
+

       
        DatabasePool::Sqlite(pool) => {


     

       
548

       
+

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


     

       
549

       
+

       



     

       
550

       
+

       
            // First verify the link belongs to the user


     

       
551

       
+

       
            let link =


     

       
552

       
+

       
                sqlx::query_as::<_, Link>("SELECT * FROM links WHERE id = ?1 AND user_id = ?2")


     

       
553

       
+

       
                    .bind(link_id)


     

       
554

       
+

       
                    .bind(user.user_id)


     

       
555

       
+

       
                    .fetch_optional(&mut *tx)


     

       
556

       
+

       
                    .await?;


     

       
557

       
+

       



     

       
558

       
+

       
            if link.is_none() {


     

       
559

       
+

       
                return Err(AppError::NotFound);


     

       
560

       
+

       
            }


     

       
561

       
+

       



     

       
562

       
+

       
            // Update the link


     

       
563

       
+

       
            let updated = sqlx::query_as::<_, Link>(


     

       
564

       
+

       
                r#"


     

       
565

       
+

       
                UPDATE links 


     

       
566

       
+

       
                SET 


     

       
567

       
+

       
                    original_url = ?1,


     

       
568

       
+

       
                    short_code = COALESCE(?2, short_code)


     

       
569

       
+

       
                WHERE id = ?3 AND user_id = ?4


     

       
570

       
+

       
                RETURNING *


     

       
571

       
+

       
                "#,


     

       
572

       
+

       
            )


     

       
573

       
+

       
            .bind(&payload.url)


     

       
574

       
+

       
            .bind(&payload.custom_code)


     

       
575

       
+

       
            .bind(link_id)


     

       
576

       
+

       
            .bind(user.user_id)


     

       
577

       
+

       
            .fetch_one(&mut *tx)


     

       
578

       
+

       
            .await?;


     

       
579

       
+

       



     

       
580

       
+

       
            // If source is provided, add a click record


     

       
581

       
+

       
            if let Some(ref source) = payload.source {


     

       
582

       
+

       
                sqlx::query("INSERT INTO clicks (link_id, source) VALUES (?1, ?2)")


     

       
583

       
+

       
                    .bind(link_id)


     

       
584

       
+

       
                    .bind(source)


     

       
585

       
+

       
                    .execute(&mut *tx)


     

       
586

       
+

       
                    .await?;


     

       
587

       
+

       
            }


     

       
588

       
+

       



     

       
589

       
+

       
            tx.commit().await?;


     

       
590

       
+

       
            updated


     

       
591

       
+

       
        }


     

       
592

       
+

       
    };


     

       
593

       
+

       



     

       
594

       
+

       
    Ok(HttpResponse::Ok().json(updated_link))


     

       
595

       
+

       
}


     

       
596

       
+

       



     

       
597

       
 

       
pub async fn delete_link(


     

       
598

       
 

       
    state: web::Data<AppState>,


     

       
599

       
 

       
    user: AuthenticatedUser,


     

       
600

       
 

       
    path: web::Path<i32>,


     

       
601

       
 

       
) -> Result<impl Responder, AppError> {


     

       
602

       
+

       
    let link_id: i32 = path.into_inner();


     

       
603

       
 

       



     

       
604

       
 

       
    match &state.db {


     

       
605

       
 

       
        DatabasePool::Postgres(pool) => {


     
···

       
707

       
 

       
                WHERE link_id = $1


     

       
708

       
 

       
                GROUP BY DATE(created_at)


     

       
709

       
 

       
                ORDER BY DATE(created_at) ASC


     

       

       

       
     

       
710

       
 

       
                "#,


     

       
711

       
 

       
            )


     

       
712

       
 

       
            .bind(link_id)


     
···

       
723

       
 

       
                WHERE link_id = ?


     

       
724

       
 

       
                GROUP BY DATE(created_at)


     

       
725

       
 

       
                ORDER BY DATE(created_at) ASC


     

       

       

       
     

       
726

       
 

       
                "#,


     

       
727

       
 

       
            )


     

       
728

       
 

       
            .bind(link_id)


     
···

       
778

       
 

       
            sqlx::query_as::<_, SourceStats>(


     

       
779

       
 

       
                r#"


     

       
780

       
 

       
                SELECT 


     

       
781

       
+

       
                    DATE(created_at)::text as date,


     

       
782

       
 

       
                    query_source as source,


     

       
783

       
 

       
                    COUNT(*)::bigint as count


     

       
784

       
 

       
                FROM clicks


     

       
785

       
 

       
                WHERE link_id = $1


     

       
786

       
 

       
                    AND query_source IS NOT NULL


     

       
787

       
 

       
                    AND query_source != ''


     

       
788

       
+

       
                GROUP BY DATE(created_at), query_source


     

       
789

       
+

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


     

       

       

       
     

       
790

       
 

       
                "#,


     

       
791

       
 

       
            )


     

       
792

       
 

       
            .bind(link_id)


     
···

       
797

       
 

       
            sqlx::query_as::<_, SourceStats>(


     

       
798

       
 

       
                r#"


     

       
799

       
 

       
                SELECT 


     

       
800

       
+

       
                    DATE(created_at) as date,


     

       
801

       
 

       
                    query_source as source,


     

       
802

       
 

       
                    COUNT(*) as count


     

       
803

       
 

       
                FROM clicks


     

       
804

       
 

       
                WHERE link_id = ?


     

       
805

       
 

       
                    AND query_source IS NOT NULL


     

       
806

       
 

       
                    AND query_source != ''


     

       
807

       
+

       
                GROUP BY DATE(created_at), query_source


     

       
808

       
+

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


     

       

       

       
     

       
809

       
 

       
                "#,


     

       
810

       
 

       
            )


     

       
811

       
 

       
            .bind(link_id)
+159 -1
src/main.rs 
···

       
1

       
 

       
use actix_cors::Cors;


     

       
2

       
 

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


     

       
3

       
 

       
use anyhow::Result;


     

       

       

       
     

       
4

       
 

       
use rust_embed::RustEmbed;


     

       
5

       
 

       
use simplelink::check_and_generate_admin_token;


     

       

       

       
     

       
6

       
 

       
use simplelink::{create_db_pool, run_migrations};


     

       
7

       
 

       
use simplelink::{handlers, AppState};


     

       
8

       
-

       
use tracing::info;


     

       

       

       
     

       
9

       
 

       



     

       

       

       
     

       

       

       
     

       
10

       
 

       
#[derive(RustEmbed)]


     

       
11

       
 

       
#[folder = "static/"]


     

       
12

       
 

       
struct Asset;


     
···

       
23

       
 

       
    }


     

       
24

       
 

       
}


     

       
25

       
 

       



     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
26

       
 

       
#[actix_web::main]


     

       
27

       
 

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


     

       
28

       
 

       
    // Load environment variables from .env file


     
···

       
35

       
 

       
    let pool = create_db_pool().await?;


     

       
36

       
 

       
    run_migrations(&pool).await?;


     

       
37

       
 

       



     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       

       

       
     

       
38

       
 

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


     

       
39

       
 

       



     

       
40

       
 

       
    let state = AppState {


     
···

       
70

       
 

       
                        "/links/{id}/sources",


     

       
71

       
 

       
                        web::get().to(handlers::get_link_sources),


     

       
72

       
 

       
                    )


     

       

       

       
     

       
73

       
 

       
                    .route("/auth/register", web::post().to(handlers::register))


     

       
74

       
 

       
                    .route("/auth/login", web::post().to(handlers::login))


     

       
75

       
 

       
                    .route(


     
···

       
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


     
···

       
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

       
 

       



     

       
197

       
 

       
    let state = AppState {


     
···

       
227

       
 

       
                        "/links/{id}/sources",


     

       
228

       
 

       
                        web::get().to(handlers::get_link_sources),


     

       
229

       
 

       
                    )


     

       
230

       
+

       
                    .route("/links/{id}", web::patch().to(handlers::edit_link))


     

       
231

       
 

       
                    .route("/auth/register", web::post().to(handlers::register))


     

       
232

       
 

       
                    .route("/auth/login", web::post().to(handlers::login))


     

       
233

       
 

       
                    .route(
+2 -1
src/models.rs 
···

       
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

       
 

       
    }


     
···

       
150

       
 

       



     

       
151

       
 

       
#[derive(sqlx::FromRow, Serialize)]


     

       
152

       
 

       
pub struct SourceStats {


     

       

       

       
     

       
153

       
 

       
    pub source: String,


     

       
154

       
 

       
    pub count: i64,


     

       
155

       
 

       
}


     
···

       
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

       
 

       
    }


     
···

       
150

       
 

       



     

       
151

       
 

       
#[derive(sqlx::FromRow, Serialize)]


     

       
152

       
 

       
pub struct SourceStats {


     

       
153

       
+

       
    pub date: String,


     

       
154

       
 

       
    pub source: String,


     

       
155

       
 

       
    pub count: i64,


     

       
156

       
 

       
}