fix: typo

feat: accept post_short param (default true)

.drone.yml

@@ -0,0 +1,26 @@
+kind: pipeline
+type: kubernetes
+name: default
+metadata:
+  namespace: git
+concurrency:
+  limit: 1
+volumes:
+  - name: cache
+    claim:
+      name: storage2
+steps:
+  - name: build
+    image: docker
+    privileged: true
+    volumes:
+      - name: cache
+        path: /cache
+    environment:
+      TOKEN_DOCKER:
+        from_secret: docker_hub
+    commands:
+      - dockerd --data-root /cache/dockerd &
+      - docker login -u voidic -p $TOKEN_DOCKER
+      - docker buildx build --push -t voidic/tiktok-dvm:latest .
+      - kill $(cat /var/run/docker.pid)

Cargo.toml

@@ -1,5 +1,5 @@
 [package]
-name = "rust_dvm"
+name = "tiktok_dvm"
 version = "0.1.0"
 edition = "2021"
 

Dockerfile

@@ -0,0 +1,17 @@
+ARG IMAGE=rust:bookworm
+ARG FEATURES
+
+FROM $IMAGE AS build
+WORKDIR /app/src
+COPY src src
+COPY Cargo.lock Cargo.lock
+COPY Cargo.toml Cargo.toml
+RUN cargo install --path . --root /app/build --features "${FEATURES}"
+
+FROM $IMAGE AS runner
+LABEL org.opencontainers.image.source="https://git.v0l.io/Kieran/tiktok-dvm"
+LABEL org.opencontainers.image.licenses="MIT"
+LABEL org.opencontainers.image.authors="Kieran"
+WORKDIR /app
+COPY --from=build /app/build .
+ENTRYPOINT ["./bin/tiktok_dvm"]

src/dvm/tiktok.rs

@@ -6,15 +6,13 @@ use nostr_sdk::prelude::DataVendingMachineStatus;
 use nostr_sdk::{serde_json, Client, EventBuilder, Kind, Tag};
 use serde::{Deserialize, Serialize};
 use std::env::temp_dir;
-use std::fmt::format;
 use std::fs::create_dir_all;
 use std::future::Future;
-use std::path::{Path, PathBuf};
+use std::path::Path;
 use std::pin::Pin;
 use std::time::Duration;
+use yt_dlp::executor::Executor;
 use yt_dlp::fetcher::deps::Libraries;
-use yt_dlp::utils::executor::Executor;
-use yt_dlp::Youtube;
 
 /// Basic DVM which clones videos from TikTok and posts them as kind 22 shorts
 pub struct TikTokDvm {
@@ -49,7 +47,12 @@ impl DVMHandler for TikTokDvm {
                 bail!("Only URL inputs are accepted")
             };
 
-            if input.host_str() != Some("www.tiktok.com") {
+            if !input
+                .host_str()
+                .as_ref()
+                .map(|h| h.ends_with("tiktok.com"))
+                .unwrap_or(false)
+            {
                 bail!("Only tiktok urls are accepted");
             }
 
@@ -94,17 +97,32 @@ impl DVMHandler for TikTokDvm {
                 imeta.push(format!("{} {}", k, v));
             }
 
-            let ev = EventBuilder::new(Kind::Custom(22), "")
-                .tag(Tag::parse(imeta)?)
-                .tags([
-                    Tag::parse(["title", &info.full_title])?,
-                    Tag::parse(["r", input.as_str()])?,
-                ]);
+            let should_post = if let Some(ps) = request.params.get("post_short") {
+                ps.to_ascii_lowercase() == "true"
+            } else {
+                true
+            };
 
-            let event_posted = client.send_event_builder(ev).await?;
-            let job_data = NoviaVideoData {
-                event_id: event_posted.val.to_hex(),
-                video: res.sha256,
+            let job_data = if should_post {
+                let ev = EventBuilder::new(Kind::Custom(22), &info.full_title)
+                    .tag(Tag::parse(imeta)?)
+                    .tags([
+                        Tag::parse(["title", &info.full_title])?,
+                        Tag::parse(["r", input.as_str()])?,
+                    ]);
+
+                let event_posted = client.send_event_builder(ev).await?;
+                NoviaVideoData {
+                    event_id: Some(event_posted.val.to_hex()),
+                    video: Some(res.sha256),
+                    url: None,
+                }
+            } else {
+                NoviaVideoData {
+                    event_id: None,
+                    video: Some(res.sha256),
+                    url: Some(res.url),
+                }
             };
 
             let status = build_status_for_job(
@@ -164,6 +182,10 @@ struct TikTokVideo {
 
 #[derive(Deserialize, Serialize)]
 struct NoviaVideoData {
-    event_id: String,
-    video: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    event_id: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    video: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    url: Option<String>,
 }