You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

191 lines
6.1 KiB

5 months ago
using UnityEngine;
4 months ago
using Amazon.S3.Model;
using UltraCombos.Upload;
4 months ago
using OscJack;
using System.Collections;
4 months ago
using TMPro;
4 months ago
using System;
4 months ago
[System.Serializable]
public class S3Tag
{
public string key;
public string value;
public static implicit operator Tag(S3Tag t) => new Tag { Key = t.key, Value = t.value };
}
5 months ago
public class SaveImage : MonoBehaviour
{
4 months ago
4 months ago
4 months ago
public Uploader uploader;
4 months ago
public RenderTexture renderTexture_postcard;
public RenderTexture renderTexture_share;
4 months ago
public string OutputFolder = "output";
4 months ago
public string FirebaseUrl = "https://firestore.googleapis.com/v1/projects/uc-24070-thegreattipsy/databases/(default)/documents/";
4 months ago
public TextMeshProUGUI TextNumber;
public TextMeshProUGUI TextSummary;
4 months ago
5 months ago
// Start is called before the first frame update
void Start()
{
4 months ago
uploader = GetComponent<Uploader>();
5 months ago
}
// Update is called once per frame
void Update()
{
4 months ago
5 months ago
}
4 months ago
5 months ago
4 months ago
public void Save(string input_text)
5 months ago
{
4 months ago
var parmas = input_text.Split('#');
if (parmas.Length < 2)
{
Debug.LogError("Input text must contain at least two parts separated by '#'.");
return;
}
string id = parmas[0].Trim();
4 months ago
string summary = parmas[1].Trim();
string password = parmas.Length > 2 ? parmas[2].Trim() : string.Empty;
4 months ago
TextNumber.text = id;
TextSummary.text = summary;
4 months ago
string filename = $"{password}_{id}.png"; // Use the ID as the filename
4 months ago
4 months ago
if (string.IsNullOrEmpty(filename))
4 months ago
{
Debug.LogError("Filename cannot be null or empty.");
return;
}
4 months ago
Debug.Log("Saving image to: " + filename + " with password: " + password);
4 months ago
4 months ago
string timestamp = System.DateTime.Now.ToString("yyyyMMdd");
4 months ago
if (!System.IO.Directory.Exists(OutputFolder + "/print/" + timestamp))
4 months ago
{
4 months ago
System.IO.Directory.CreateDirectory(OutputFolder + "/print/" + timestamp);
4 months ago
}
4 months ago
if (!System.IO.Directory.Exists(OutputFolder + "/share/" + timestamp))
{
System.IO.Directory.CreateDirectory(OutputFolder + "/share/" + timestamp);
}
4 months ago
4 months ago
SaveRenderTextureToPNG(renderTexture_postcard, System.IO.Path.Combine(OutputFolder + "/print/" + timestamp, filename));
Debug.Log("Image saved to " + filename);
4 months ago
4 months ago
SaveRenderTextureToPNG(renderTexture_share, System.IO.Path.Combine(OutputFolder + "/share/" + timestamp, filename));
4 months ago
Debug.Log("Image saved to " + filename);
5 months ago
// Optionally, you can also log the full path
4 months ago
string print_fullPath = System.IO.Path.Combine(OutputFolder + "/print/" + timestamp + "/", filename);
Debug.Log("print_path: " + print_fullPath);
upload(id, print_fullPath, "print");
4 months ago
4 months ago
string share_fullPath = System.IO.Path.Combine(OutputFolder + "/share/" + timestamp + "/", filename);
Debug.Log("share_path: " + share_fullPath);
upload(id, share_fullPath, "share");
5 months ago
}
void SaveRenderTextureToPNG(RenderTexture rt, string filePath)
{
RenderTexture currentRT = RenderTexture.active;
RenderTexture.active = rt;
Texture2D tex = new Texture2D(rt.width, rt.height, TextureFormat.RGBA32, false);
tex.ReadPixels(new Rect(0, 0, rt.width, rt.height), 0, 0);
tex.Apply();
byte[] bytes = tex.EncodeToPNG();
System.IO.File.WriteAllBytes(filePath, bytes);
4 months ago
Debug.Log("Saved RenderTexture to PNG at " + filePath + " with size: " + bytes.Length + " bytes");
4 months ago
5 months ago
RenderTexture.active = currentRT;
Destroy(tex);
}
4 months ago
4 months ago
void upload(string id, string filename, string type)
4 months ago
{
4 months ago
if (uploader != null)
{
uploader.Upload(filename, (response) =>
{
if (response.success)
{
4 months ago
Debug.Log("Upload successful: " + response.message);
4 months ago
4 months ago
if(type=="print") writeToFirebase("prints", id, response.message, type);
writeToFirebase("records", id, response.message, type);
4 months ago
}
else
{
Debug.LogError("Upload failed: " + response);
}
});
}
else
{
Debug.LogError("Uploader is not assigned.");
}
}
4 months ago
4 months ago
public void writeToFirebase(string collection, string id, string url, string type)
4 months ago
{
4 months ago
Debug.Log("Writing to Firebase: " + id);
StartCoroutine(WriteToFirebaseCoroutine(id, url));
4 months ago
4 months ago
IEnumerator WriteToFirebaseCoroutine(string id, string url)
4 months ago
{
4 months ago
// string[] tags=filename.Split('/');
// string id=tags[tags.Length - 1].Split('.')[0];
4 months ago
Debug.Log("Extracted ID: " + id);
4 months ago
string firebaseUrl = FirebaseUrl + collection + "/" + id;
4 months ago
string jsonData = "{ \"fields\": { " +
4 months ago
"\""+type+"_url\": { \"stringValue\": \"" + url + "\" }, " +
4 months ago
"\"timestamp\": { \"timestampValue\": \"" + System.DateTime.UtcNow.ToString("o") + "\" } " +
"} }";
using (var www = new UnityEngine.Networking.UnityWebRequest(firebaseUrl, "PATCH"))
{
byte[] bodyRaw = System.Text.Encoding.UTF8.GetBytes(jsonData);
www.uploadHandler = new UnityEngine.Networking.UploadHandlerRaw(bodyRaw);
www.downloadHandler = new UnityEngine.Networking.DownloadHandlerBuffer();
www.SetRequestHeader("Content-Type", "application/json");
yield return www.SendWebRequest();
#if UNITY_2020_1_OR_NEWER
if (www.result != UnityEngine.Networking.UnityWebRequest.Result.Success)
#else
if (www.isNetworkError || www.isHttpError)
#endif
{
Debug.LogError("Error writing to Firebase: " + www.error);
Debug.LogError("Response: " + www.downloadHandler.text);
}
else
{
Debug.Log("Successfully wrote to Firebase: " + www.downloadHandler.text);
}
}
}
}
4 months ago
4 months ago
}