HiRERNABackend.cs

Go to the documentation of this file.

/*
 * Adapted from https://github.com/eamonwoortman/django-unity3d-example
 */

//using Newtonsoft.Json;
//using Newtonsoft.Json.Linq;
using System;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;
using SimpleJSON;
using System.Text;
using System.IO;
using System.Collections;
using UI;

public struct Session {
    public int id;
    public string name;
    public string slug;
}

public partial class BackendManager {
    private List<Session> sessions = new List<Session>();
    public static string progress = "";

    public Session[] getSessions()
    {
        return sessions.ToArray();
    }
    
    public void populateSessions(JSONNode responseData)
    {
        sessions.Clear();
        char[] charsToTrim = { '"' };
        for (int i = 0; i < responseData.Count; i++)
        {
            Session s = new Session();
            s.id = responseData[i]["id"].AsInt;
            s.name = responseData[i]["name"].ToString().Trim(charsToTrim);
            s.slug = responseData[i]["slug"].ToString().Trim(charsToTrim);
            Debug.Log (s.slug);
            sessions.Add(s);
        }
    }

    public IEnumerator downloadSessionFile(string sessionSlug)
    {
        Debug.Log ("Downloading molecule");
        Dictionary<string, string> headers = new Dictionary<string, string>();

        headers.Add("Authorization", "Token " + authenticationToken);

        WWW www = new WWW(hostUrl + "sessions/" + sessionSlug + "/", null, headers);


        while (!www.isDone) {
            progress = "downloaded " + (www.progress*100).ToString() + "%...";
            yield return null;
        }
        
        string fullPath = Application.persistentDataPath + "/" + sessionSlug + ".pdb";
        Debug.Log (fullPath);
        File.WriteAllBytes (fullPath, www.bytes);

        Debug.Log ("Downloaded");
        
        progress = "downloaded, opening...";

        //clear before fetching again
        Molecule3D.clearScene();

        UIData.loadHireRNA = true;
        GameObject.Find ("LoadBox").GetComponent<Molecule3D>().gUIDisplay.OpenFileCallback(fullPath);

        progress = "Done.";
    }
    
    public delegate void LoggedIn();
    public delegate void Connecting();
    public delegate void LoginFailed(string errorMsg);
    public LoggedIn OnLoggedIn;
    public Connecting OnConnection;
    public LoginFailed OnLoginFailed;

    /*
     * Session list retrieval
     */
    public delegate void SessionSuccess();
    public delegate void RetrievingSession();
    public delegate void SessionFailed(string errorMsg);
    public SessionSuccess OnSessionSuccess;
    public RetrievingSession OnSession;
    public SessionFailed OnSessionFailed;

    public delegate void SubmissionSuccess();
    public delegate void Submitting();
    public delegate void SubmissionFailed(string errorMsg);
    public SubmissionSuccess OnSubmissionSuccess;
    public Submitting OnSubmission;
    public SubmissionFailed OnSubmissionFailed;

    private string authenticationToken = "";
    public bool logged()
    {
        return authenticationToken != "";
    }
    
    public void Login(string username, string password) {
        Dictionary<string, object> fields = new Dictionary<string, object>();
        fields.Add("username", username);
        fields.Add("password", password);
        if (OnConnection != null) {
            OnConnection();
        }
        PerformRequest("getauthtoken/", fields, OnLoginResponse);
    }

    private void OnLoginResponse(ResponseType responseType, JSONNode responseData, string callee) {
        if (responseType == ResponseType.Success) {
            authenticationToken = responseData["token"];
            if (OnLoggedIn != null) {
                OnLoggedIn();
            }
        } else {
            if (OnLoginFailed != null) {
                OnLoginFailed("Connection failed");
            }
        }
        //        } else {
//            JToken fieldToken = responseData["non_field_errors"];
//            if (fieldToken == null || !fieldToken.HasValues) {
//                if (OnLoginFailed != null) {
//                    OnLoginFailed("Login failed: unknown error.");
//                }
//            } else {
//                string errors = "";
//                JToken[] fieldValidationErrors = fieldToken.Values().ToArray();
//                foreach (JToken validationError in fieldValidationErrors) {
//                    errors += validationError.Value<string>();
//                }
//                if (OnLoginFailed != null) {
//                    OnLoginFailed("Login failed: " + errors);
//                }
//            }
//        }
    }

    public void Logout()
    {
        PerformRequest("logout/", null, OnLogoutResponse, authenticationToken);
    }

    private void OnLogoutResponse(ResponseType responseType, JSONNode responseData, string callee)
    {
        authenticationToken = "";
        sessions.Clear();
    }

    // Take a snapshot of the current atom positions
    public void SubmitPDB(string sessionSlug)
    {
        if (OnSubmission != null) {
            OnSubmission();
        }

        string filename = HiRERNAContestOldGUI.username + "-" + sessionSlug + "-" + DateTime.Now.ToString ("yyyyMMdd-Hmmss") +".pdb";

        string snapshot = SnapshotManager.generateSnapshot();

        WWWForm form = new WWWForm();
        form.AddBinaryData("filename", Encoding.ASCII.GetBytes(snapshot), filename);

        PerformFormRequest("sessions/" + sessionSlug + "/submit/", form, filename, snapshot, OnSubmittedPDB, authenticationToken);
    }
    
    private void OnSubmittedPDB(ResponseType responseType, JSONNode responseData, string callee)
    {
        if (responseType == ResponseType.Success) {
            Debug.Log ("Success");
            if (OnSubmissionSuccess != null) {
                OnSubmissionSuccess();
            }
        } else {
            Debug.Log ("Failure");
            if (OnSubmissionFailed != null) {
                OnSubmissionFailed("Submission refused");
            }
        }
    }

    public void RetrieveSessions()
    {
        if (OnSession != null) {
            OnSession();
        }

        PerformRequest("sessions/", null, OnRetrievedSessions, authenticationToken);
    }

    private void OnRetrievedSessions(ResponseType responseType, JSONNode responseData, string callee)
    {
        if (responseType == ResponseType.Success) {
            populateSessions(responseData);
            if (OnSessionSuccess != null) {
                OnSessionSuccess();
            }
        } else {
            Debug.Log ("Failure");
            if (OnSessionFailed != null) {
                OnSessionFailed("Session list retrieval refused");
            }
        }
    }
}